Sha256: 3039a34f84770c02a7ed86e10c54d159cbba53789c6996faf64ef4f92a3460e2

Contents?: true

Size: 1.8 KB

Versions: 4

Compression:

Stored size: 1.8 KB

Contents

# frozen_string_literal: true

class Freddy
  module Consumers
    class RespondToConsumer
      def self.consume(*attrs, &block)
        new(*attrs).consume(&block)
      end

      def initialize(thread_pool:, destination:, channel:, handler_adapter_factory:)
        @consume_thread_pool = thread_pool
        @destination = destination
        @channel = channel
        @handler_adapter_factory = handler_adapter_factory
      end

      def consume
        consumer = consume_from_destination do |delivery|
          adapter = @handler_adapter_factory.for(delivery)

          msg_handler = MessageHandler.new(adapter, delivery)
          yield(delivery.payload, msg_handler)
        end

        ResponderHandler.new(consumer, @consume_thread_pool)
      end

      private

      def consume_from_destination(&block)
        @channel.queue(@destination).subscribe(manual_ack: true) do |delivery|
          process_message(delivery, &block)
        end
      end

      def process_message(delivery)
        @consume_thread_pool.process do
          begin
            scope = delivery.build_trace("freddy:respond:#{@destination}",
                                         tags: {
                                           'peer.address' => "#{@destination}:#{delivery.payload[:type]}",
                                           'component' => 'freddy',
                                           'span.kind' => 'server', # RPC
                                           'message_bus.destination' => @destination,
                                           'message_bus.correlation_id' => delivery.correlation_id
                                         })

            yield(delivery)
          ensure
            @channel.acknowledge(delivery.tag, false)
            scope.close
          end
        end
      end
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version Path
freddy-1.7.0 lib/freddy/consumers/respond_to_consumer.rb
freddy-1.6.0 lib/freddy/consumers/respond_to_consumer.rb
freddy-1.5.0 lib/freddy/consumers/respond_to_consumer.rb
freddy-1.4.2 lib/freddy/consumers/respond_to_consumer.rb