mdp 0.1

data/ext/majordomo/worker.h

@@ -0,0 +1,43 @@
+#ifndef MAJORDOMO_WORKER_H
+#define MAJORDOMO_WORKER_H
+
+typedef struct {
+    mdp_worker_t *worker;
+    VALUE broker;
+    VALUE service;
+    VALUE heartbeat;
+    VALUE reconnect;
+#ifndef HAVE_RB_THREAD_BLOCKING_REGION
+    zlist_t *recv_buffer;
+#endif
+} rb_majordomo_worker_t;
+
+#define MAJORDOMO_WORKER_HEARTBEAT 2500
+#define MAJORDOMO_WORKER_RECONNECT 2500
+
+#define GetMajordomoWorker(obj) \
+    rb_majordomo_worker_t *worker = NULL; \
+    Data_Get_Struct(obj, rb_majordomo_worker_t, worker); \
+    if (!worker) rb_raise(rb_eTypeError, "uninitialized Majordomo worker!"); \
+    if (!worker->worker) rb_raise(rb_eRuntimeError, "Majordomo worker has already been closed!");
+
+struct nogvl_md_worker_new_args {
+    char *broker;
+    char *service;
+    int verbose;
+};
+
+struct nogvl_md_worker_recv_args {
+    rb_majordomo_worker_t *worker;
+    zframe_t *reply;
+};
+
+struct nogvl_md_worker_send_args {
+    mdp_worker_t *worker;
+    zmsg_t *progress;
+    zframe_t *reply_to;
+};
+
+void _init_majordomo_worker();
+
+#endif

data/lib/majordomo.rb

@@ -0,0 +1,8 @@
+# encoding: utf-8
+
+# Prefer compiled Rubinius bytecode in .rbx/
+ENV["RBXOPT"] = "-Xrbc.db"
+
+require "majordomo_ext"
+
+require 'majordomo/version' unless defined? Majordomo::VERSION

data/lib/majordomo/version.rb

@@ -0,0 +1,3 @@
+module Majordomo
+  VERSION = '0.1'
+end

data/mdp.gemspec

@@ -0,0 +1,22 @@
+# encoding: utf-8
+
+require File.expand_path('../lib/majordomo/version', __FILE__)
+
+Gem::Specification.new do |s|
+  s.name = "mdp"
+  s.version = Majordomo::VERSION
+  s.summary = "Majordomo for Ruby"
+  s.description = "Majordomo - A Ruby binding for libmdp (Majordomo implementation in C)"
+  s.authors = ["Lourens Naudé"]
+  s.email = ["lourens@methodmissing.com"]
+  s.homepage = "http://github.com/methodmissing/majordomo"
+  s.date = Time.now.utc.strftime('%Y-%m-%d')
+  s.platform = Gem::Platform::RUBY
+  s.extensions = "ext/majordomo/extconf.rb"
+  s.has_rdoc = true
+  s.files = `git ls-files`.split("\n")
+  s.test_files = `git ls-files test`.split("\n")
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.add_development_dependency('rake-compiler', '~> 0.8.0')
+end

data/test/helper.rb

@@ -0,0 +1,20 @@
+# encoding: utf-8
+
+require 'test/unit'
+require 'majordomo'
+
+class MajordomoTestCase < Test::Unit::TestCase
+  undef_method :default_test if method_defined? :default_test
+
+  BROKER = "tcp://localhost:5555"
+
+  if ENV['STRESS_GC']
+    def setup
+      GC.stress = true
+    end
+
+    def teardown
+      GC.stress = false
+    end
+  end
+end

data/test/test_client.rb

@@ -0,0 +1,51 @@
+# encoding: utf-8
+
+require File.join(File.dirname(__FILE__), 'helper')
+
+class TestClient < MajordomoTestCase
+  def test_init
+    client = Majordomo::Client.new(BROKER)
+    assert_instance_of Majordomo::Client, client
+    client.close
+
+    assert_raises TypeError do
+      Majordomo::Client.new(:invalid)
+    end
+
+    client = Majordomo::Client.new(BROKER, true)
+  ensure
+    client.close
+  end
+
+  def test_broker
+    client = Majordomo::Client.new(BROKER)
+    assert_equal BROKER, client.broker
+  ensure
+    client.close
+  end
+
+  def test_timeout
+    client = Majordomo::Client.new(BROKER)
+    assert_equal 2500, client.timeout
+    client.timeout = 1000
+    assert_equal 1000, client.timeout
+  ensure
+    client.close
+  end
+
+  def test_close
+    client = Majordomo::Client.new(BROKER)
+    assert_nil client.close
+    assert_raises RuntimeError do
+      client.timeout
+    end
+  end
+
+  def test_send
+    client = Majordomo::Client.new(BROKER, true)
+    client.timeout = 100;
+    assert client.send("test", "message")
+  ensure
+    client.close
+  end
+end

data/test/test_threading.rb

@@ -0,0 +1,32 @@
+# encoding: utf-8
+
+require File.join(File.dirname(__FILE__), 'helper')
+
+class TestThreading < MajordomoTestCase
+  def test_send_recv
+    threads = []
+    req_received, rep_received = 0, 0
+    threads << Thread.new do
+      client = Majordomo::Client.new(BROKER, true)
+      client.timeout = 100
+      100.times do
+        client.send("thread_test", "ping")
+        reply = client.recv("thread_test")
+        rep_received += 1 if reply
+      end
+    end
+
+    threads << Thread.new do
+      worker = Majordomo::Worker.new(BROKER, "thread_test", true)
+      100.times do
+        request, reply_to = worker.recv
+        req_received += 1 if request
+        worker.send("pong", reply_to)
+      end
+    end
+
+    threads.map(&:join)
+    assert_equal req_received, 100
+    assert_equal rep_received, 100
+  end
+end

data/test/test_worker.rb

@@ -0,0 +1,86 @@
+# encoding: utf-8
+
+require File.join(File.dirname(__FILE__), 'helper')
+
+class TestWorker < MajordomoTestCase
+  def test_init
+    worker = Majordomo::Worker.new(BROKER, "test_init")
+    assert_instance_of Majordomo::Worker, worker
+    worker.close
+
+    assert_raises TypeError do
+      Majordomo::Worker.new(:invalid, "test")
+    end
+
+    assert_raises TypeError do
+      Majordomo::Worker.new(BROKER, :invalid)
+    end
+
+    worker = Majordomo::Worker.new(BROKER, "test", true)
+  ensure
+    worker.close
+  end
+
+  def test_broker
+    worker = Majordomo::Worker.new(BROKER, "test")
+    assert_equal BROKER, worker.broker
+  ensure
+    worker.close
+  end
+
+  def test_service
+    service = "test_service"
+    worker = Majordomo::Worker.new(BROKER, service)
+    assert_equal service, worker.service
+  ensure
+    worker.close
+  end
+
+  def test_heartbeat
+    worker = Majordomo::Worker.new(BROKER, "test_heartbeat")
+    assert_equal 2500, worker.heartbeat
+    worker.heartbeat = 100
+    assert_equal 100, worker.heartbeat
+  ensure
+    worker.close
+  end
+
+  def test_reconnect
+    worker = Majordomo::Worker.new(BROKER, "test_reconnect")
+    assert_equal 2500, worker.reconnect
+    worker.reconnect = 100
+    assert_equal 100, worker.reconnect
+  ensure
+    worker.close
+  end
+
+  def test_close
+    worker = Majordomo::Worker.new(BROKER, "test_close")
+    assert_nil worker.close
+
+    assert_raises RuntimeError do
+      worker.reconnect
+    end
+  end
+
+  def test_recv
+    service = "test_revc"
+    worker = Majordomo::Worker.new(BROKER, service, true)
+    worker.heartbeat = 100
+    worker.reconnect = 100
+
+    client = Majordomo::Client.new(BROKER, true)
+    client.timeout = 100
+    assert client.send service, "request"
+
+    request = worker.recv
+    assert_instance_of Array, request
+    assert_equal "request", request[0]
+
+    assert worker.send("reply", request[1])
+    assert_equal "reply", client.recv(service)
+  ensure
+    client.close
+    worker.close
+  end
+end

metadata

@@ -0,0 +1,92 @@
+--- !ruby/object:Gem::Specification
+name: mdp
+version: !ruby/object:Gem::Version
+  version: '0.1'
+  prerelease: 
+platform: ruby
+authors:
+- Lourens Naudé
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-09-06 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rake-compiler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.8.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.8.0
+description: Majordomo - A Ruby binding for libmdp (Majordomo implementation in C)
+email:
+- lourens@methodmissing.com
+executables: []
+extensions:
+- ext/majordomo/extconf.rb
+extra_rdoc_files: []
+files:
+- .gitignore
+- CHANGELOG.rdoc
+- Gemfile
+- Gemfile.lock
+- README.rdoc
+- Rakefile
+- ext/majordomo/client.c
+- ext/majordomo/client.h
+- ext/majordomo/extconf.rb
+- ext/majordomo/jruby.h
+- ext/majordomo/majordomo_ext.c
+- ext/majordomo/majordomo_ext.h
+- ext/majordomo/prelude.h
+- ext/majordomo/rubinius.h
+- ext/majordomo/ruby18.h
+- ext/majordomo/ruby19.h
+- ext/majordomo/worker.c
+- ext/majordomo/worker.h
+- lib/majordomo.rb
+- lib/majordomo/version.rb
+- mdp.gemspec
+- test/helper.rb
+- test/test_client.rb
+- test/test_threading.rb
+- test/test_worker.rb
+homepage: http://github.com/methodmissing/majordomo
+licenses: []
+post_install_message: 
+rdoc_options:
+- --charset=UTF-8
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Majordomo for Ruby
+test_files:
+- test/helper.rb
+- test/test_client.rb
+- test/test_threading.rb
+- test/test_worker.rb