Contents

# frozen_string_literal: true

require 'bunny'

class Freddy
  module Adapters
    class BunnyAdapter
      def self.connect(config)
        bunny = Bunny.new(config)
        bunny.start
        new(bunny)
      end

      def initialize(bunny)
        @bunny = bunny
      end

      def create_channel(prefetch: nil)
        bunny_channel = @bunny.create_channel
        bunny_channel.prefetch(prefetch) if prefetch
        Channel.new(bunny_channel)
      end

      def close
        @bunny.close
      end

      class Channel
        extend Forwardable

        NO_ROUTE = 312

        def initialize(channel)
          @channel = channel
        end

        def_delegators :@channel, :topic, :default_exchange, :consumers, :acknowledge, :reject

        def queue(*args)
          Queue.new(@channel.queue(*args))
        end

        def on_no_route
          default_exchange.on_return do |return_info, properties, _content|
            yield(properties[:correlation_id]) if return_info[:reply_code] == NO_ROUTE
          end
        end
      end

      class Queue
        def initialize(queue)
          @queue = queue
        end

        def bind(*args)
          @queue.bind(*args)
          self
        end

        def name
          @queue.name
        end

        def message_count
          @queue.message_count
        end

        def subscribe(manual_ack: false)
          @queue.subscribe(manual_ack: manual_ack) do |info, properties, payload|
            parsed_payload = Payload.parse(payload, properties[:content_encoding])
            delivery = Delivery.new(
              parsed_payload, properties, info.routing_key, info.delivery_tag, info.exchange
            )
            yield(delivery)
          end
        end
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
freddy-2.2.0	lib/freddy/adapters/bunny_adapter.rb
freddy-2.1.0	lib/freddy/adapters/bunny_adapter.rb