jerryluk-rabbitmq-jruby-client 0.1.6 → 0.1.7

data/README

@@ -32,11 +32,17 @@ queue.publish('message body')
 # Retrieve a message from the queue
 message = queue.retrieve
 
-# Subscribe to a queue with callback
+# Subscribe to a queue with callback. The callback will be run in a new thread for each new message. 
 queue.subscribe do |message|
   # do something with message
 end
 
+# Subscribe to a queue in a loop. No new thread will be created.
+queue.loop_subscribe do |message|
+  # do something with message
+end
+
+
 Installing RabbitMQ on OS X
 ===========================
 1. Install MacPorts

data/lib/rabbitmq_client.rb

@@ -12,7 +12,9 @@ class RabbitMQClient
   include_class('com.rabbitmq.client.Channel')
   include_class('com.rabbitmq.client.Consumer')
   include_class('com.rabbitmq.client.DefaultConsumer')
+  include_class('com.rabbitmq.client.QueueingConsumer')
   include_class('com.rabbitmq.client.MessageProperties')
+  include_class('java.lang.InterruptedException')
   include_class('java.lang.String') { |package, name| "J#{name}" }
   
   class RabbitMQClientError < StandardError;end
@@ -88,6 +90,22 @@ class RabbitMQClient
       @channel.basic_consume(@name, no_ack, QueueConsumer.new(@channel, block))
     end
     
+    def loop_subscribe(&block)
+      no_ack = false
+      consumer = QueueingConsumer.new(@channel)
+      @channel.basic_consume(@name, no_ack, consumer)
+      loop do
+        begin
+          delivery = consumer.next_delivery
+          message_body = Marshal.load(String.from_java_bytes(delivery.get_body))
+          block.call message_body
+          @channel.basic_ack(delivery.get_envelope.get_delivery_tag, false)
+        rescue InterruptedException => ie
+          next
+        end
+      end
+    end
+    
     protected
     def auto_bind
       unless @exchange

data/spec/rabbitmq_client_spec.rb

@@ -42,6 +42,8 @@ describe RabbitMQClient do
       @queue.bind(@exchange)
       @queue.publish('Hello World')
       @queue.retrieve.should == 'Hello World'
+      @queue.publish('人大')
+      @queue.retrieve.should == '人大'
     end
     
     it "should able to subscribe with a callback function" do
@@ -56,6 +58,25 @@ describe RabbitMQClient do
       a.should == 3
     end
     
+    it "should able to subscribe to a queue using loop_subscribe" do
+      a = 0
+      @queue.bind(@exchange)
+      Thread.new do
+        begin
+          timeout(1) do
+            @queue.loop_subscribe do |v|
+              a += v.to_i
+            end
+          end
+        rescue Timeout::Error => e
+        end
+      end
+      @queue.publish("1")
+      @queue.publish("2")
+      sleep 2
+      a.should == 3
+    end
+    
     it "should raise an exception if binding a persistent queue with non-persistent exchange and vice versa" do
       persistent_queue = @client.queue('test_queue1', true)
       persistent_exchange = @client.exchange('test_exchange1', 'fanout', true)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: jerryluk-rabbitmq-jruby-client
 version: !ruby/object:Gem::Version 
-  version: 0.1.6
+  version: 0.1.7
 platform: ruby
 authors: 
 - Jerry Luk