rabbitmq-tutorials/ruby/rpc_client.rb
Jakub Stastny aka botanicus 2d91ef20b5 Ruby: RPC examples.
2011-05-03 11:19:53 +02:00

67 lines
1.4 KiB
Ruby
Executable File

#!/usr/bin/env ruby
# encoding: utf-8
# Note: This is just proof of concept. For
# real-world usage, you are strongly advised
# to use https://github.com/ruby-amqp/rpc
# or some other RPC library.
require "amqp"
class FibonacciRpcClient
def initialize
subscribe_to_callback_queue
end
def connection
@connection ||= AMQP.connect(:host => "localhost")
end
def channel
@channel ||= AMQP::Channel.new(self.connection)
end
def callback_queue
@callback_queue ||= self.channel.queue("", :exclusive => true)
end
def requests
@requests ||= Hash.new
end
def call(n, &block)
corr_id = rand(10_000_000).to_s
self.requests[corr_id] = nil
self.callback_queue.append_callback(:declare) do
AMQP::Exchange.default.publish(n.to_s, :routing_key => "rpc_queue", :reply_to => self.callback_queue.name, :correlation_id => corr_id)
EM.add_periodic_timer(0.1) do
# p self.requests
if result = self.requests[corr_id]
block.call(result.to_i)
EM.stop
end
end
end
end
private
def subscribe_to_callback_queue
self.callback_queue.subscribe do |header, body|
corr_id = header.correlation_id
unless self.requests[corr_id]
self.requests[corr_id] = body
end
end
end
end
EM.run do
fibonacci_rpc = FibonacciRpcClient.new()
puts " [x] Requesting fib(30)"
fibonacci_rpc.call(30) do |response|
puts " [.] Got #{response}"
end
end