buster 0.1.1

data/.gitignore

@@ -0,0 +1,6 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*
+*.swp
+vendor/bundle

data/Gemfile

@@ -0,0 +1,8 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in buster.gemspec
+gemspec
+
+group :development do
+  gem 'rspec'
+end

data/README.md

@@ -0,0 +1,36 @@
+Buster
+---
+
+A Ruby Service Bus
+
+License
+---
+
+Copyright (c) 2012, Jason Staten, PeerIntel
+
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+- Redistributions of source code must retain the above copyright notice, this
+  list of conditions and the following disclaimer.
+
+- Redistributions in binary form must reproduce the above copyright notice,
+  this list of conditions and the following disclaimer in the documentation
+  and/or other materials provided with the distribution.
+
+- Neither the name of PeerIntel nor the names of its contributors may be used
+  to endorse or promote products derived from this software without specific
+  prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
+ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
+THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

data/Rakefile

@@ -0,0 +1,4 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new

data/buster.gemspec

@@ -0,0 +1,23 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "buster/version"
+
+Gem::Specification.new do |s|
+  s.name        = "buster"
+  s.version     = Buster::VERSION
+  s.authors     = ["Jason Staten"]
+  s.email       = ["jstaten07@gmail.com"]
+  s.homepage    = ""
+  s.summary     = %q{Service Bus}
+  s.description = %q{Service Bus}
+
+  s.rubyforge_project = "buster"
+
+  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"]
+
+  s.add_runtime_dependency "ffi-rzmq"
+  s.add_runtime_dependency "msgpack"
+end

data/lib/buster.rb

@@ -0,0 +1,40 @@
+%w{host sender worker poller command_handler router context version}.each { |r| require "buster/#{r}" }
+
+require 'ffi-rzmq'
+require 'msgpack'
+
+module Buster
+  class << self
+    attr_accessor :local_endpoint
+
+    def routes
+      @routes ||= {}
+    end
+
+    def start
+      raise "Bus is already started" if @host
+      @host = Buster::Host.new(routes, context, self.local_endpoint)
+      @host.start
+    end
+
+    def stop
+      raise "Bus is not running" unless @host
+      @host.stop
+      @host = nil
+    end
+
+    def fire(name, props = {})
+      sender.fire(name, props)
+    end
+
+    private
+
+    def sender
+      @sender ||= Buster::Sender.new(context)
+    end
+
+    def context
+      @context ||= Buster::Context.new(ZMQ::Context.create)
+    end
+  end
+end

data/lib/buster/command_handler.rb

@@ -0,0 +1,11 @@
+module Buster
+  module CommandHandler
+    attr_accessor :reply_action
+
+    def reply(message_name, props = {})
+      raise 'No reply_action available' unless self.reply_action
+
+      reply_action.call(message_name, props)
+    end
+  end
+end

data/lib/buster/context.rb

@@ -0,0 +1,25 @@
+module Buster
+  class Context
+
+    attr_accessor :context
+
+    def initialize(context)
+      @context = context
+      @sockets = []
+    end
+
+    def socket(type)
+      socket = @context.socket(type)
+      @sockets << socket
+      socket
+    end
+
+    def terminate
+      @sockets.reverse.each do |s|
+        s.setsockopt(ZMQ::LINGER, 0)
+        s.close
+      end
+      @context.terminate
+    end
+  end
+end

data/lib/buster/host.rb

@@ -0,0 +1,36 @@
+module Buster
+  class Host
+
+    def initialize(routes, context, local_endpoint = nil)
+      @local_endpoint = local_endpoint ||"inproc://local_endpoint"
+      @context = context
+      @router = Buster::Router.new(@context, routes)
+      @poller = Buster::Poller.new
+    end
+
+    def start
+      frontend = @context.socket(ZMQ::ROUTER)
+      frontend.bind(@local_endpoint)
+
+      backend = @context.socket(ZMQ::DEALER)
+      backend.bind("inproc://workers")
+
+      @router.connect_routes backend, @poller
+
+      Buster::Worker.start_worker @context
+
+      @poller.pipe frontend, backend
+      @poller.pipe backend, frontend
+
+      poll_thread = Thread.new { @poller.start }
+      poll_thread.join
+
+      @context.terminate
+    end
+
+    def stop
+      @poller.stop
+    end
+
+  end
+end

data/lib/buster/poller.rb

@@ -0,0 +1,40 @@
+module Buster
+  class Poller
+    def initialize
+      @poller = ZMQ::Poller.new
+    end
+
+    def start
+      @run = true
+      while @run
+        @poller.poll(500)
+        @poller.readables.each do |socket|
+          action = @readable_actions[socket]
+          action.call(socket) if action
+        end
+      end
+    end
+
+    def stop
+      @run = false
+    end
+
+    def register(socket, &block)
+      @poller.register(socket, ZMQ::POLLIN)
+      readable_actions[socket] = block
+    end
+
+    def pipe(incoming, outgoing)
+      register(incoming) do |s|
+        s.recv_strings(msgs = [])
+        outgoing.send_strings(msgs, ZMQ::NOBLOCK)
+      end
+    end
+
+    private
+
+    def readable_actions
+      @readable_actions ||= {}
+    end
+  end
+end

data/lib/buster/router.rb

@@ -0,0 +1,34 @@
+module Buster
+  class Router
+
+    def initialize(context, routes = {})
+      @context = context
+      @routes = routes
+    end
+
+    def connect_routes(reply_socket, poller)
+      local = @context.socket(ZMQ::DEALER)
+      local.bind("inproc://routes")
+
+      sockets = @routes.map do |pattern, uri|
+        remote = @context.socket(ZMQ::DEALER)
+        result = remote.connect(uri)
+
+        poller.pipe remote, reply_socket
+        [pattern, remote]
+      end
+
+      poller.register(local) do |s|
+        s.recv_strings(msgs = [])
+        message_name = msgs[0]
+        #TODO: More robust routing
+        remote = sockets.detect([nil,nil]){|x| x[0].match message_name}[1]
+        if remote.nil?
+          puts "No remote matching '#{message_name}'"
+          return
+        end
+        remote.send_strings(msgs)
+      end
+    end
+  end
+end

data/lib/buster/sender.rb

@@ -0,0 +1,20 @@
+module Buster
+  class Sender
+    def initialize(context)
+      @context = context
+    end
+
+    def fire(name, props = {})
+      bin = MessagePack.pack(props)
+      sender.send_strings([name.to_s, bin])
+    end
+
+    private
+
+    def sender
+      s = @context.socket(ZMQ::DEALER)
+      s.connect("inproc://routes")
+      s
+    end
+  end
+end

data/lib/buster/version.rb

@@ -0,0 +1,3 @@
+module Buster
+  VERSION = "0.1.1"
+end

data/lib/buster/worker.rb

@@ -0,0 +1,49 @@
+module Buster
+  class Worker
+    def initialize(context)
+      @context = context
+    end
+
+    def run
+      worker = @context.socket(ZMQ::DEALER)
+      worker.connect("inproc://workers")
+      while true
+        worker.recv_strings(msgs = [])
+        return if msgs.length < 2
+
+        body = MessagePack.unpack(msgs.pop)
+        message_name = msgs.pop
+        reply_id = msgs.pop
+
+        handler = find_handler message_name
+        if handler.nil?
+          puts "No handler found for #{message_name}"
+          return
+        end
+
+        handler.reply_action = lambda do |name,props|
+          worker.send_strings([reply_id, name.to_s, MessagePack.pack(props)])
+        end
+
+        handler.execute body
+      end
+      worker.close
+    end
+
+
+    def self.start_worker(context)
+      worker = Worker.new(context)
+      t = Thread.new { worker.run }
+      t.run
+    end
+
+    private
+
+    def find_handler(name)
+     camelized = name.split(/[^a-z0-9]/i).map{|w| w.capitalize}.join
+     const_name = "#{camelized}Handler"
+     Object.const_get(const_name).new if Object.const_defined?(const_name)
+    end
+
+  end
+end

data/spec/integration_spec.rb

@@ -0,0 +1,82 @@
+require 'spec_helper'
+
+describe 'integration' do
+  before :all do
+    start_endpoint :mars
+    start_endpoint :pluto
+    Buster.local_endpoint = 'ipc://earth.ipc'
+    Buster.routes[/mars/] = 'ipc://mars.ipc'
+    Buster.routes[/pluto/] = 'ipc://pluto.ipc'
+    Buster.routes[/station/] = 'ipc://station.ipc'
+    Thread.new { Buster.start }
+    sleep 1 #bus startup delay
+  end
+
+  after :all do
+    Buster.stop
+  end
+
+  let(:mutex) { Mutex.new }
+  let(:baton) { ConditionVariable.new }
+
+  describe 'when sending a hello_mars' do
+    before do
+      Handler 'HelloEarth' do |props|
+        mutex.synchronize do
+          @response = props['name']
+          baton.signal
+        end
+      end
+    end
+
+    it 'should respond in the mars uppercase style' do
+      name = 'Major Tom'
+      Buster.fire :hello_mars, :name => name
+      mutex.synchronize do
+        baton.wait(mutex, 1)
+        @response.should == name.upcase
+      end
+    end
+  end
+
+  describe 'when sending a hello_pluto' do
+    before do
+      Handler 'HelloEarth' do |props|
+        mutex.synchronize do
+          @response = props['name']
+          baton.signal
+        end
+      end
+    end
+
+    it 'should respond in the pluto reversed style' do
+      name = 'Ground Control'
+      Buster.fire :hello_pluto, :name => name
+      mutex.synchronize do
+        baton.wait(mutex, 1)
+        @response.should == name.reverse
+      end
+    end
+  end
+
+  describe 'when sending to a powered down space station' do
+    before do
+      Handler 'TenFour' do |props|
+        mutex.synchronize do
+          @ten_foured = true
+          baton.signal
+        end
+      end
+    end
+
+    it 'should receive response when the station powers up' do
+      Buster.fire :adjust_station_temperature
+      sleep 1 #Oh no! Station isn't started!
+      start_endpoint :station
+      mutex.synchronize do
+        baton.wait(mutex, 1)
+        @ten_foured.should == true
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,51 @@
+require 'rubygems'
+require 'bundler/setup'
+
+$:.unshift File.expand_path('../../lib', __FILE__)
+require 'buster'
+
+Bundler.require :development
+
+Dir[File.join(File.dirname(__FILE__), 'support', '*.rb')].each { |d| require d }
+
+Thread.abort_on_exception = true
+
+$endpoint_pids = []
+
+RSpec.configure do |config|
+  config.mock_with :rspec
+
+  config.after(:all) do
+    $endpoint_pids.each { |p| Process.kill("TERM", p) }
+  end
+end
+
+def Handler(name, &block)
+  const_name = "#{name}Handler"
+  if Object.const_defined?(const_name)
+    klass = Object.const_get(const_name)
+  else
+    klass = Class.new do
+      include Buster::CommandHandler
+
+      def execute(props)
+        @@execute_action.call(props,self)
+      end
+
+      def self.execute_action=(action)
+        @@execute_action = action
+      end
+    end
+    Object.const_set(const_name, klass)
+  end
+
+  klass.execute_action = block
+
+  klass
+end
+
+def start_endpoint(name)
+  pid =  Process.spawn("ruby #{File.join(File.dirname(__FILE__),'test_endpoints', "#{name}.rb")}")
+  $endpoint_pids << pid
+
+end

data/spec/test_endpoints/mars.rb

@@ -0,0 +1,14 @@
+require File.expand_path('../../spec_helper.rb', __FILE__)
+
+class HelloMarsHandler
+  include Buster::CommandHandler
+
+  def execute(props)
+    reply :hello_earth, 'name' => props['name'].upcase
+  end
+end
+
+trap("TERM") { Buster.stop }
+
+Buster.local_endpoint = 'ipc://mars.ipc'
+Buster.start

data/spec/test_endpoints/pluto.rb

@@ -0,0 +1,14 @@
+require File.expand_path('../../spec_helper.rb', __FILE__)
+
+class HelloPlutoHandler
+  include Buster::CommandHandler
+
+  def execute(props)
+    reply :hello_earth, 'name' => props['name'].reverse
+  end
+end
+
+trap("TERM") { Buster.stop }
+
+Buster.local_endpoint = 'ipc://pluto.ipc'
+Buster.start

data/spec/test_endpoints/station.rb

@@ -0,0 +1,14 @@
+require File.expand_path('../../spec_helper.rb', __FILE__)
+
+class AdjustStationTemperatureHandler
+  include Buster::CommandHandler
+
+  def execute(props)
+    reply :ten_four
+  end
+end
+
+trap("TERM") { Buster.stop }
+
+Buster.local_endpoint = 'ipc://station.ipc'
+Buster.start

metadata

@@ -0,0 +1,97 @@
+--- !ruby/object:Gem::Specification
+name: buster
+version: !ruby/object:Gem::Version
+  version: 0.1.1
+  prerelease: 
+platform: ruby
+authors:
+- Jason Staten
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-01-13 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: ffi-rzmq
+  requirement: &11559360 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *11559360
+- !ruby/object:Gem::Dependency
+  name: msgpack
+  requirement: &11558760 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *11558760
+description: Service Bus
+email:
+- jstaten07@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- README.md
+- Rakefile
+- buster.gemspec
+- lib/buster.rb
+- lib/buster/command_handler.rb
+- lib/buster/context.rb
+- lib/buster/host.rb
+- lib/buster/poller.rb
+- lib/buster/router.rb
+- lib/buster/sender.rb
+- lib/buster/version.rb
+- lib/buster/worker.rb
+- spec/integration_spec.rb
+- spec/spec_helper.rb
+- spec/test_endpoints/mars.rb
+- spec/test_endpoints/pluto.rb
+- spec/test_endpoints/station.rb
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: 1543500698659680995
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: 1543500698659680995
+requirements: []
+rubyforge_project: buster
+rubygems_version: 1.8.13
+signing_key: 
+specification_version: 3
+summary: Service Bus
+test_files:
+- spec/integration_spec.rb
+- spec/spec_helper.rb
+- spec/test_endpoints/mars.rb
+- spec/test_endpoints/pluto.rb
+- spec/test_endpoints/station.rb