jerryluk-rabbitmq-jruby-client 0.1.2 → 0.1.3

data/MIT-LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2009 Jerry Luk <jerryluk@gmail.com>
+
+Permission is hereby granted, free of charge, to any person
+obtaining a copy of this software and associated documentation
+files (the "Software"), to deal in the Software without
+restriction, including without limitation the rights to use,
+copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the
+Software is furnished to do so, subject to the following
+conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
+OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
+HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
+WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+OTHER DEALINGS IN THE SOFTWARE.

data/README

@@ -0,0 +1,22 @@
+
+This project is a JRuby wrapper of the Java RabbitMQ Client from Rabbit Technologies
+
+http://www.rabbitmq.com/
+
+Using this library does not require you to use EventMachine for subscribing a RabbitMQ queue.
+
+To use RabbitMQ JRuby Client with Workling:
+1. Copy workling_rabbitmq_client.rb and workling_rabbitmq_subscriber.rb to lib
+Create an initializer 'rabbitmq.rb' in 'config/initializers'
+
+require 'workling_rabbitmq_client'
+Workling::Remote.dispatcher = Workling::Remote::Runners::ClientRunner.new
+Workling::Remote.dispatcher.client = Workling::Clients::WorklingRabbitMQClient.new
+
+Optional: To use subscriber
+Modify vendor/plugins/workling/script/listen.rb, added this after the require statements
+
+require 'workling_rabbitmq_subscriber'
+
+To start Workling: (You can use script/workling_client because it uses daemons)
+jruby vendor/plugins/workling/script/listen.rb

data/lib/rabbitmq_client.rb

@@ -0,0 +1,151 @@
+require 'java'
+require File.dirname(__FILE__) + '/junit.jar'
+require File.dirname(__FILE__) + '/commons-cli-1.1.jar'
+require File.dirname(__FILE__) + '/commons-io-1.2.jar'
+require File.dirname(__FILE__) + '/rabbitmq-client.jar'
+
+class RabbitMQClient
+  include ObjectSpace
+  include_class('com.rabbitmq.client.Connection')
+  include_class('com.rabbitmq.client.ConnectionParameters')
+  include_class('com.rabbitmq.client.ConnectionFactory')
+  include_class('com.rabbitmq.client.Channel')
+  include_class('com.rabbitmq.client.Consumer')
+  include_class('com.rabbitmq.client.DefaultConsumer')
+  include_class('java.lang.String') { |package, name| "J#{name}" }
+  
+  class QueueConsumer < DefaultConsumer
+    def initialize(channel, block)
+      @channel = channel
+      @block = block
+      super(channel)
+    end
+    
+    def handleDelivery(consumer_tag, envelope, properties, body)
+      delivery_tag = envelope.get_delivery_tag
+      message_body = Marshal.load(String.from_java_bytes(body))
+      @block.call message_body
+      @channel.basic_ack(delivery_tag, false)
+    end
+  end
+  
+  class Queue
+    def initialize(name, channel)
+      @name = name
+      @channel = channel
+      @channel.queue_declare(name)
+      self
+    end
+    
+    def bind(exchange, routing_key=nil)
+      @routing_key = routing_key || "#{@name}_#{Time.new.to_i.to_s}_#{rand(100).to_s}"
+      @exchange = exchange
+      @channel.queue_bind(@name, @exchange.name, @routing_key)
+      self
+    end
+    
+    def publish(message_body, props=nil)
+      auto_bind
+      message_body_byte = Marshal.dump(message_body).to_java_bytes
+      @channel.basic_publish(@exchange.name, @routing_key, props, message_body_byte)
+      message_body
+    end
+    
+    def retrieve
+      auto_bind
+      message_body = nil
+      no_ack = false
+      response = @channel.basic_get(@name, no_ack)
+      if response
+        props = response.get_props
+        message_body = Marshal.load(String.from_java_bytes(response.get_body))
+        delivery_tag = response.get_envelope.get_delivery_tag
+        @channel.basic_ack(delivery_tag, false)
+      end
+      message_body
+    end
+    
+    def subscribe(&block)
+      no_ack = false
+      @channel.basic_consume(@name, no_ack, QueueConsumer.new(@channel, block))
+    end
+    
+    protected
+    def auto_bind
+      unless @exchange
+        exchange = Exchange.new("@name_exchange", 'fanout', @channel)
+        self.bind(exchange)
+      end
+    end
+  end
+  
+  class Exchange
+    attr_reader :name
+    
+    def initialize(name, type, channel)
+      @name = name
+      @type = type
+      @channel = channel
+      @channel.exchange_declare(@name, type.to_s)
+      self
+    end
+  end
+  
+  # Class Methods
+  class << self
+  end
+  
+  attr_reader :channel
+  attr_reader :connection
+  
+  # Instance Methods
+  def initialize(options={})
+    # server address
+    @host = options[:host] || '127.0.0.1'
+    @port = options[:port] || 5672
+    
+    # login details
+    @username = options[:username] || 'guest'
+    @password = options[:password] || 'guest'
+    @vhost = options[:vhost] || '/'
+    
+    # queues and exchanges
+    @queues = {}
+    @exchanges = {}
+    
+    connect
+    # Disconnect before the object is destroyed
+    define_finalizer(self, lambda {|id| self.disconnect if self.connected? })
+    self
+  end
+  
+  def connect
+    params = ConnectionParameters.new
+    params.set_username(@username)
+    params.set_password(@password)
+    params.set_virtual_host(@vhost)
+    params.set_requested_heartbeat(0)
+    conn_factory = ConnectionFactory.new(params)
+    @connection = conn_factory.new_connection(@host, @port)
+    @channel = @connection.create_channel
+  end
+  
+  def disconnect
+    @channel.close
+    @connection.close
+    @connection = nil
+  end
+  
+  def connected?
+    @connection != nil
+  end
+  
+  def queue(name)
+    @queues[name] ||= Queue.new(name, @channel)
+  end
+  
+  def exchange(name, type='fanout')
+    @exchanges[name] ||= Exchange.new(name, type, @channel)
+  end
+end
+

data/spec/rabbitmq_client_spec.rb

@@ -0,0 +1,55 @@
+require 'rubygems'
+require 'spec'
+require File.dirname(__FILE__) + '/../lib/rabbitmq_client'
+
+describe RabbitMQClient do
+  before(:each) do
+    @client = RabbitMQClient.new
+  end
+  
+  it "should able to create a connection" do
+    @client.connection.should_not be_nil
+  end
+  
+  it "should able to create a channel" do
+    @client.channel.should_not be_nil
+  end
+  
+  it "should be able to create a new exchange" do
+    exchange = @client.exchange('test_exchange', 'direct')
+    exchange.should_not be_nil
+  end
+  
+  describe Queue do
+    before(:each) do
+      @queue = @client.queue('test_queue')
+      @exchange = @client.exchange('test_exchange', 'direct')
+    end
+    
+    it "should able to create a queue" do
+      @queue.should_not be_nil
+    end
+    
+    it "should able to bind to an exchange" do
+      @queue.bind(@exchange).should_not be_nil
+    end
+    
+    it "should able to publish and retrieve a message" do
+      @queue.bind(@exchange)
+      @queue.publish('Hello World')
+      @queue.retrieve.should == 'Hello World'
+    end
+    
+    it "should able to subscribe with a callback function" do
+      a = 0
+      @queue.bind(@exchange)
+      @queue.subscribe do |v|
+         a += v.to_i
+      end
+      @queue.publish("1")
+      @queue.publish("2")
+      sleep 1
+      a.should == 3
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: jerryluk-rabbitmq-jruby-client
 version: !ruby/object:Gem::Version 
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors: 
 - Jerry Luk
@@ -21,8 +21,14 @@ extensions: []
 
 extra_rdoc_files: []
 
-files: []
-
+files: 
+- README
+- MIT-LICENSE
+- lib/commons-cli-1.1.jar
+- lib/commons-io-1.2.jar
+- lib/rabbitmq-client.jar
+- lib/junit.jar
+- lib/rabbitmq_client.rb
 has_rdoc: false
 homepage: http://www.linkedin.com/in/jerryluk
 post_install_message: 
@@ -49,5 +55,5 @@ rubygems_version: 1.2.0
 signing_key: 
 specification_version: 2
 summary: A RabbitMQ client for JRuby
-test_files: []
-
+test_files: 
+- spec/rabbitmq_client_spec.rb