Contents

require 'json'
require 'multi_json'
require 'amqp'

module Rodent
  class Listener
    attr_reader :type
    attr_accessor :params, :status, :headers, :body

    def initialize(type, &block)
      @type = type
      @source = block
    end

    def bind(error_handler)
      AMQP::Channel.new do |channel|
        channel.prefetch(100)
        queue = channel.queue(@type, exclusive: true, auto_delete: true)
        queue.bind(channel.direct('rodent.requests'), routing_key: @type)
        queue.subscribe(ack: true) do |metadata, payload|
          begin
            self.body = call(MultiJson.load(payload))
          rescue Exception => e
            self.status, self.headers, self.body = error_handler.call(e) if error_handler
          end
          channel.default_exchange.publish(MultiJson.dump(response), routing_key: metadata.reply_to, correlation_id: metadata.message_id)
          metadata.ack
        end
      end
    end

    def call(params = {})
      self.params = params
      self.status = 200
      self.headers = {}
      unless respond_to?(method_name)
        define_singleton_method(method_name, @source)
      end
      MultiJson.dump(self.send(method_name))
    end

    def method_name
      ('rodent_' + @type).gsub('.', '_').to_sym
    end

    def response
      {status: status, headers: headers, body: body}
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
rodent-0.0.7	lib/rodent/listener.rb
rodent-0.0.6	lib/rodent/listener.rb