Contents

# frozen_string_literal: true

require_relative "logger"

module RightSpeed
  module Listener
    def self.setup(worker_type:, host:, port:, backlog: nil)
      case worker_type
      when :roundrobin
        RoundRobinListener.new(host, port, backlog)
      when :fair
        FairListener.new(host, port, backlog)
      else
        SimpleListener.new(host, port, backlog)
      end
    end

    class SimpleListener
      attr_reader :sock

      def initialize(host, port, backlog)
        @host = host
        @port = port
        @backlog = backlog
        @sock = nil
      end

      def run
        @running = true
        @sock = TCPServer.open(@host, @port)
        @sock.listen(@backlog) if @backlog
        @sock
      end

      def wait
        # do nothing
      end

      def stop
        @running = false
        if @sock
          @sock.close rescue nil
        end
      end
    end

    class RoundRobinListener < SimpleListener
      attr_reader :ractor

      def run(processor)
        @running = true
        @ractor = Ractor.new(@host, @port, @backlog, processor) do |host, port, backlog, processor|
          logger = RightSpeed.logger
          sock = TCPServer.open(host, port)
          sock.listen(backlog) if backlog
          logger.info { "listening #{host}:#{port}" }
          while conn = sock.accept
            processor.process(conn)
          end
        end
      end

      def wait
        @ractor.take
      end

      def stop
        @running = false
        @ractor = nil # TODO: terminate the Ractor if possible
      end
    end

    class FairListener < RoundRobinListener
      def wait
        # nothing to wait - @ractor.take consumes accepted connections unexpectedly
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
right_speed-0.2.0	lib/right_speed/listener.rb