pub_sub 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,4 @@
+source :rubygems
+
+# Specify your gem's dependencies in pub_sub.gemspec
+gemspec

data/README.rdoc

@@ -0,0 +1,9 @@
+= pub_sub
+
+Minimal client lib for RabbitMQ publisher/subscriber.
+
+== Copyright
+
+Copyright (c) 2012 Simon Horne. See LICENSE.txt for
+further details.
+

data/Rakefile

@@ -0,0 +1,2 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks

data/lib/pub_sub.rb

@@ -0,0 +1,10 @@
+require 'bunny'
+require "pub_sub/version"
+require "pub_sub/bunny_config"
+require "pub_sub/publisher"
+require "pub_sub/fanout_publisher"
+require "pub_sub/subscriber"
+
+module PubSub
+  # Your code goes here...
+end

data/lib/pub_sub/bunny_config.rb

@@ -0,0 +1,16 @@
+module PubSub
+  module BunnyConfig
+    def bunny_config
+      defaults = { 
+        :host => 'localhost',
+        :port => 5672,
+        :user => 'guest',
+        :pass => 'guest',
+
+        :spec => '09',
+        :heartbeat => 60
+      }
+      defined?(BUNNY_CONFIG) ? defaults.merge(BUNNY_CONFIG) : defaults
+    end
+  end
+end

data/lib/pub_sub/fanout_publisher.rb

@@ -0,0 +1,59 @@
+module PubSub
+  module FanoutPublisher
+    include BunnyConfig
+
+    RETRY_WAIT = 5
+
+    def logger=(logger)
+      @logger = logger
+    end
+
+    def logger
+      @logger ||= begin
+        logger = Logger.new(STDOUT)
+        logger.level = Logger::WARN
+        logger
+      end
+    end
+
+    def bunny
+      @bunny ||= begin
+        logger.info("PubSub:FanoutPublisher#bunny: initializing and starting new bunny instance")
+        
+        b = Bunny.new(bunny_config)
+        b.start
+        b
+      end
+    end
+
+    def publish(msg, exchange_name)      
+      begin
+
+        begin
+          exchange = bunny.exchange(exchange_name, :type => :fanout, :durable => true)
+        rescue StandardError, Timeout::Error => error
+          logger.info("PubSub:FanoutPublisher#publish: resetting and reconnecting to #{exchange_name}, because - #{error}")
+          sleep RETRY_WAIT
+          reset
+          retry
+        end
+
+        # non-persistent to a fanout exchange
+        exchange.publish(msg)
+
+      rescue StandardError, Timeout::Error => error
+        logger.info("PubSub:FanoutPublisher#publish: error encountered while publishing to #{exchange_name} with routing_key #{routing_key}, retrying - #{error}")
+        sleep RETRY_WAIT
+        reset
+        retry
+      end
+    end
+
+    def reset
+      logger.info("PubSub:FanoutPublisher#reset: stopping bunny")
+
+      @bunny.stop rescue nil
+      @bunny = nil
+    end
+  end
+end

data/lib/pub_sub/publisher.rb

@@ -0,0 +1,73 @@
+module PubSub
+  module Publisher
+    include BunnyConfig
+
+    RETRY_WAIT = 5 
+
+    def logger=(logger)
+      @logger = logger
+    end
+
+    def logger
+      @logger ||= begin
+        logger = Logger.new(STDOUT)
+        logger.level = Logger::WARN
+        logger
+      end
+    end
+
+    def transactional=(transactional)
+      @transactional=transactional
+    end
+
+    def bunny
+      @bunny ||= begin
+        logger.info("PubSub:Publisher#bunny: initializing and starting new bunny instance")
+        
+        b = Bunny.new(bunny_config)
+        b.start
+        
+        if(@transactional)
+          logger.info("PubSub:Publisher#bunny: publisher is transactional, calling tx_select")
+          b.tx_select
+        end
+
+        b
+      end
+    end
+
+    def publish(msg, exchange_name, routing_key)      
+      begin
+
+        begin
+          exchange = bunny.exchange(exchange_name, :type => :topic, :durable => true)
+        rescue StandardError, Timeout::Error => error
+          logger.info("PubSub:Publisher#publish: resetting and reconnecting to #{exchange_name}, because - #{error}")
+          sleep RETRY_WAIT
+          reset
+          retry
+        end
+
+        exchange.publish(msg, :key => routing_key, :persistent => true)
+
+        if(@transactional)
+          logger.debug("PubSub:Publisher#bunny: publisher is transactional, calling tx_commit")
+          bunny.tx_commit
+        end
+
+      rescue StandardError, Timeout::Error => error
+        logger.info("PubSub:Publisher#publish: error encountered while publishing to #{exchange_name} with routing_key #{routing_key}, retrying - #{error}")
+        sleep RETRY_WAIT
+        reset
+        retry
+      end
+    end
+
+    def reset
+      logger.info("PubSub:Publisher#reset: stopping bunny")
+
+      @bunny.stop rescue nil
+      @bunny = nil
+    end
+  end
+end

data/lib/pub_sub/subscriber.rb

@@ -0,0 +1,62 @@
+module PubSub
+  module Subscriber
+    include BunnyConfig
+
+    RETRY_WAIT = 5 
+
+    def logger=(logger)
+      @logger = logger
+    end
+
+    def logger
+      @logger ||= begin
+        logger = Logger.new(STDOUT)
+        logger.level = Logger::WARN
+        logger
+      end
+    end
+
+    def queue_name=(queue_name)
+      @queue_name = queue_name
+    end
+
+    def bunny
+      @bunny ||= begin
+        logger.info("PubSub:Subscriber#bunny: initializing and starting new bunny instance")
+        
+        b = Bunny.new(bunny_config)
+        b.start
+        b.qos
+        b
+      end
+    end
+
+    def reset
+      logger.info("PubSub:Subscriber#reset: unsubscribing from queue and stopping bunny")
+      
+      @queue.unsubscribe rescue nil
+      @queue = nil
+      @bunny.stop rescue nil
+      @bunny = nil
+    end
+
+    # simple subsciption loop on a durable queue, automatic acks
+    # creates the queue if it does not already exist
+    def subscribe(&block)
+      raise "queue_name must be set before calling subscribe" unless @queue_name
+
+      logger.info("PubSub:Subscriber#subscribe: subscribing to queue - #{@queue_name}")
+      begin
+        @queue = bunny.queue(@queue_name, :durable => true)
+        @queue.subscribe(:ack => true) do |msg|
+          block.call(msg)
+        end
+      rescue StandardError, Timeout::Error => error
+        logger.info("PubSub:Subscriber: error encountered while subscribing to #{@queue_name}, retrying - #{error}")
+        sleep RETRY_WAIT
+        reset
+        retry
+      end
+    end
+  end
+end

data/lib/pub_sub/version.rb

@@ -0,0 +1,3 @@
+module PubSub
+  VERSION = "0.0.1"
+end

data/pub_sub.gemspec

@@ -0,0 +1,26 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "pub_sub/version"
+
+Gem::Specification.new do |s|
+  s.name        = "pub_sub"
+  s.version     = PubSub::VERSION
+  s.authors     = ["Simon Horne"]
+  s.email       = ["simon@soulware.co.uk"]
+  s.homepage    = ""
+  s.summary     = %q{Minimal Publisher/Subscriber}
+  s.description = %q{Minimal Publisher/Subscriber}
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  # specify any dependencies here; for example:
+  # s.add_development_dependency "rspec"
+  # s.add_runtime_dependency "rest-client"
+
+  s.add_development_dependency "rake"
+
+  s.add_runtime_dependency "bunny", ["= 0.7.8"]
+end

metadata

@@ -0,0 +1,107 @@
+--- !ruby/object:Gem::Specification 
+name: pub_sub
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Simon Horne
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-03-30 00:00:00 -04:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rake
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: bunny
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 19
+        segments: 
+        - 0
+        - 7
+        - 8
+        version: 0.7.8
+  type: :runtime
+  version_requirements: *id002
+description: Minimal Publisher/Subscriber
+email: 
+- simon@soulware.co.uk
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- README.rdoc
+- Rakefile
+- lib/pub_sub.rb
+- lib/pub_sub/bunny_config.rb
+- lib/pub_sub/fanout_publisher.rb
+- lib/pub_sub/publisher.rb
+- lib/pub_sub/subscriber.rb
+- lib/pub_sub/version.rb
+- pub_sub.gemspec
+has_rdoc: true
+homepage: ""
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Minimal Publisher/Subscriber
+test_files: []
+