rb-zmq 0.0.1

data/History.txt

@@ -0,0 +1,6 @@
+=== 1.0.0 / 2010-02-01
+
+* 1 major enhancement
+
+  * Birthday!
+

data/README.txt

@@ -0,0 +1,40 @@
+= rb_zmq
+
+* http://www.zeromq.org
+
+== DESCRIPTION:
+
+This gem comprises the ruby bindings for ZeroMQ
+
+== FEATURES/PROBLEMS:
+
+* Posix signal handling is a bit goofy, so things like handling SIGTERM (ctrl-c) at the console have to be done by hand.
+
+== SYNOPSIS:
+
+  bind_to = 'tcp://127.0.0.1:5555'
+  message_size = 1
+  roundtrip_count = 100
+        
+  ctx = Context.new(1, 1, 0)
+  s = Socket.new(ctx, REP);
+  s.bind(bind_to);
+
+  for i in 0...roundtrip_count do
+      msg = s.recv(0)
+      s.send(msg, 0)
+  end
+
+  sleep 1
+
+
+== REQUIREMENTS:
+
+* The libzmq must be installed & loadable
+
+== INSTALL:
+
+* gem install rb-zmq
+
+== LICENSE:
+  This software is licensed under the same terms as ZeroMQ itself.

data/extconf.rb

@@ -0,0 +1,28 @@
+#
+#    Copyright (c) 2007-2010 iMatix Corporation
+#
+#    This file is part of 0MQ.
+#
+#    0MQ is free software; you can redistribute it and/or modify it under
+#    the terms of the Lesser GNU General Public License as published by
+#    the Free Software Foundation; either version 3 of the License, or
+#    (at your option) any later version.
+#
+#    0MQ is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    Lesser GNU General Public License for more details.
+#
+#    You should have received a copy of the Lesser GNU General Public License
+#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+require 'mkmf'
+dir_config('zmq')
+if have_library('zmq', 'zmq_init') and have_header('zmq.h')
+  puts "cool, I found your zmq install..."
+  create_makefile("zmq")
+else
+  raise "Couldn't find zmq library. try setting --with-zmq-dir=<path> to tell me where it is."
+end
+
+

data/lib/rb_zmq.rb

@@ -0,0 +1 @@
+require 'librbzmq'

data/rbzmq.cpp

@@ -0,0 +1,286 @@
+/*
+    Copyright (c) 2007-2010 iMatix Corporation
+
+    This file is part of 0MQ.
+
+    0MQ is free software; you can redistribute it and/or modify it under
+    the terms of the Lesser GNU General Public License as published by
+    the Free Software Foundation; either version 3 of the License, or
+    (at your option) any later version.
+
+    0MQ is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    Lesser GNU General Public License for more details.
+
+    You should have received a copy of the Lesser GNU General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include <assert.h>
+#include <errno.h>
+#include <string.h>
+#include <ruby.h>
+
+#include "zmq.h"
+
+static void context_free (void *ctx)
+{
+    if (ctx) {
+       int rc = zmq_term (ctx);
+       assert (rc == 0);
+    }
+}
+
+static VALUE context_alloc (VALUE class_)
+{
+    return rb_data_object_alloc (class_, NULL, 0, context_free);
+}
+
+static VALUE context_initialize (VALUE self_, VALUE app_threads_,
+    VALUE io_threads_, VALUE flags_)
+{
+    assert (!DATA_PTR (self_));
+    void *ctx = zmq_init (NUM2INT (app_threads_), NUM2INT (io_threads_),
+        NUM2INT (flags_));
+    if (!ctx) {
+        rb_raise (rb_eRuntimeError, zmq_strerror (errno));
+        return Qnil;
+    }
+
+    DATA_PTR (self_) = (void*) ctx;
+    return self_;
+}
+
+static void socket_free (void *s)
+{
+    if (s) {
+       int rc = zmq_close (s);
+       assert (rc == 0);
+    }
+}
+
+static VALUE socket_alloc (VALUE class_)
+{
+    return rb_data_object_alloc (class_, NULL, 0, socket_free);
+}
+
+static VALUE socket_initialize (VALUE self_, VALUE context_, VALUE type_)
+{
+    assert (!DATA_PTR (self_));
+
+    if (strcmp (rb_obj_classname (context_), "Context") != 0) {
+        rb_raise (rb_eArgError, "expected Context object");
+        return Qnil;
+    }
+
+    void *s = zmq_socket (DATA_PTR (context_), NUM2INT (type_));
+    if (!s) {
+        rb_raise (rb_eRuntimeError, zmq_strerror (errno));
+        return Qnil;
+    }
+
+    DATA_PTR (self_) = (void*) s;
+    return self_;
+}
+
+
+static VALUE socket_setsockopt (VALUE self_, VALUE option_,
+    VALUE optval_)
+{
+
+    int rc = 0;
+
+    switch (NUM2INT (option_)) {
+    case ZMQ_HWM:
+    case ZMQ_LWM:
+    case ZMQ_SWAP:
+    case ZMQ_AFFINITY:
+    case ZMQ_RATE:
+    case ZMQ_RECOVERY_IVL:
+    case ZMQ_MCAST_LOOP:
+        {
+            long optval = FIX2LONG (optval_);
+
+            //  Forward the code to native 0MQ library.
+            rc = zmq_setsockopt (DATA_PTR (self_), NUM2INT (option_), 
+                (void *) &optval, 4);
+        }
+        break;
+
+    case ZMQ_IDENTITY:
+    case ZMQ_SUBSCRIBE:
+    case ZMQ_UNSUBSCRIBE:
+
+        //  Forward the code to native 0MQ library.
+        rc = zmq_setsockopt (DATA_PTR (self_), NUM2INT (option_), 
+	    (void *) StringValueCStr (optval_), RSTRING_LEN (optval_));	
+        break;
+
+    default:
+        rc = -1;
+        errno = EINVAL;
+    }
+
+    if (rc != 0) {
+        rb_raise (rb_eRuntimeError, zmq_strerror (errno));
+        return Qnil;
+    }
+
+    return self_;
+}
+
+
+static VALUE socket_bind (VALUE self_, VALUE addr_)
+{
+    assert (DATA_PTR (self_));
+
+    int rc = zmq_bind (DATA_PTR (self_), rb_string_value_cstr (&addr_));
+    if (rc != 0) {
+        rb_raise (rb_eRuntimeError, zmq_strerror (errno));
+        return Qnil;
+    }
+
+    return Qnil;
+}
+
+static VALUE socket_connect (VALUE self_, VALUE addr_)
+{
+    assert (DATA_PTR (self_));
+
+    int rc = zmq_connect (DATA_PTR (self_), rb_string_value_cstr (&addr_));
+    if (rc != 0) {
+        rb_raise (rb_eRuntimeError, zmq_strerror (errno));
+        return Qnil;
+    }
+
+    return Qnil;
+}
+
+static VALUE socket_send (VALUE self_, VALUE msg_, VALUE flags_)
+{
+    assert (DATA_PTR (self_));
+
+    Check_Type (msg_, T_STRING);
+
+    zmq_msg_t msg;
+    int rc = zmq_msg_init_size (&msg, RSTRING_LEN (msg_));
+    if (rc != 0) {
+        rb_raise (rb_eRuntimeError, zmq_strerror (errno));
+        return Qnil;
+    }
+    memcpy (zmq_msg_data (&msg), RSTRING_PTR (msg_), RSTRING_LEN (msg_));
+ 
+    rc = zmq_send (DATA_PTR (self_), &msg, NUM2INT (flags_));
+    if (rc != 0 && errno == EAGAIN) {
+        rc = zmq_msg_close (&msg);
+        assert (rc == 0);
+        return Qfalse;
+    }
+
+    if (rc != 0) {
+        rb_raise (rb_eRuntimeError, zmq_strerror (errno));
+        rc = zmq_msg_close (&msg);
+        assert (rc == 0);
+        return Qnil;
+    }
+
+    rc = zmq_msg_close (&msg);
+    assert (rc == 0);
+    return Qtrue;
+}
+
+static VALUE socket_flush (VALUE self_)
+{
+    assert (DATA_PTR (self_));
+
+    int rc = zmq_flush (DATA_PTR (self_));
+    if (rc != 0) {
+        rb_raise (rb_eRuntimeError, zmq_strerror (errno));
+        return Qnil;
+    }
+
+    return Qnil;
+}
+
+static VALUE socket_recv (VALUE self_, VALUE flags_)
+{
+    assert (DATA_PTR (self_));
+
+    zmq_msg_t msg;
+    int rc = zmq_msg_init (&msg);
+    assert (rc == 0);
+
+    rc = zmq_recv (DATA_PTR (self_), &msg, NUM2INT (flags_));
+    if (rc != 0 && errno == EAGAIN) {
+        rc = zmq_msg_close (&msg);
+        assert (rc == 0);
+        return Qnil;
+    }
+
+    if (rc != 0) {
+        rb_raise (rb_eRuntimeError, zmq_strerror (errno));
+        rc = zmq_msg_close (&msg);
+        assert (rc == 0);
+        return Qnil;
+    }
+
+    VALUE message = rb_str_new ((char*) zmq_msg_data (&msg),
+        zmq_msg_size (&msg));
+    rc = zmq_msg_close (&msg);
+    assert (rc == 0);
+    return message;
+}
+
+extern "C" void Init_librbzmq ()
+{	
+    VALUE context_type = rb_define_class ("Context", rb_cObject);
+    rb_define_alloc_func (context_type, context_alloc);
+    rb_define_method (context_type, "initialize",
+        (VALUE(*)(...)) context_initialize, 3);
+
+    VALUE socket_type = rb_define_class ("Socket", rb_cObject);
+    rb_define_alloc_func (socket_type, socket_alloc);
+    rb_define_method (socket_type, "initialize",
+        (VALUE(*)(...)) socket_initialize, 2);
+    rb_define_method (socket_type, "setsockopt",
+        (VALUE(*)(...)) socket_setsockopt, 2);
+    rb_define_method (socket_type, "bind",
+        (VALUE(*)(...)) socket_bind, 1);
+    rb_define_method (socket_type, "connect",
+        (VALUE(*)(...)) socket_connect, 1);
+    rb_define_method (socket_type, "send",
+        (VALUE(*)(...)) socket_send, 2);
+    rb_define_method (socket_type, "flush",
+        (VALUE(*)(...)) socket_flush, 0);
+    rb_define_method (socket_type, "recv",
+        (VALUE(*)(...)) socket_recv, 1);
+
+    rb_define_global_const ("HWM", INT2NUM (ZMQ_HWM));
+    rb_define_global_const ("LWM", INT2NUM (ZMQ_LWM));
+    rb_define_global_const ("SWAP", INT2NUM (ZMQ_SWAP));
+    rb_define_global_const ("AFFINITY", INT2NUM (ZMQ_AFFINITY));
+    rb_define_global_const ("IDENTITY", INT2NUM (ZMQ_IDENTITY));
+    rb_define_global_const ("SUBSCRIBE", INT2NUM (ZMQ_SUBSCRIBE));
+    rb_define_global_const ("UNSUBSCRIBE", INT2NUM (ZMQ_UNSUBSCRIBE));
+    rb_define_global_const ("RATE", INT2NUM (ZMQ_RATE));
+    rb_define_global_const ("RECOVERY_IVL", INT2NUM (ZMQ_RECOVERY_IVL));
+    rb_define_global_const ("MCAST_LOOP", INT2NUM (ZMQ_MCAST_LOOP));
+    rb_define_global_const ("SNDBUF", INT2NUM (ZMQ_SNDBUF));
+    rb_define_global_const ("RCVBUF", INT2NUM (ZMQ_RCVBUF));
+
+    rb_define_global_const ("NOBLOCK", INT2NUM (ZMQ_NOBLOCK));
+    rb_define_global_const ("NOFLUSH", INT2NUM (ZMQ_NOFLUSH));
+
+    rb_define_global_const ("P2P", INT2NUM (ZMQ_P2P));
+    rb_define_global_const ("SUB", INT2NUM (ZMQ_SUB));
+    rb_define_global_const ("PUB", INT2NUM (ZMQ_PUB));
+    rb_define_global_const ("REQ", INT2NUM (ZMQ_REQ));
+    rb_define_global_const ("REP", INT2NUM (ZMQ_REP));
+    rb_define_global_const ("XREQ", INT2NUM (ZMQ_XREQ));
+    rb_define_global_const ("XREP", INT2NUM (ZMQ_XREP));
+    rb_define_global_const ("UPSTREAM", INT2NUM (ZMQ_UPSTREAM));
+    rb_define_global_const ("DOWNSTREAM", INT2NUM (ZMQ_DOWNSTREAM));
+
+    rb_define_global_const ("POLL", INT2NUM (ZMQ_POLL));
+}

data/test/test_rb_zmq.rb

@@ -0,0 +1,10 @@
+require "test/unit"
+
+class TestRbZmq < Test::Unit::TestCase
+  def test_sanity
+    require 'rubygems'
+    gem 'rb-zmq'
+    require 'librbzmq'
+    assert Context.new 1,1,0
+  end
+end

metadata

@@ -0,0 +1,91 @@
+--- !ruby/object:Gem::Specification 
+name: rb-zmq
+version: !ruby/object:Gem::Version 
+  version: 0.0.1
+platform: ruby
+authors: 
+- McClain Looney
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-02-01 00:00:00 -06:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: hoe
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 2.2.5
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: gemcutter
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.3.0
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: hoe
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 2.5.0
+    version: 
+description: This gem comprises the ruby bindings for ZeroMQ
+email: 
+- m@loonsoft.com
+executables: []
+
+extensions: 
+- extconf.rb
+extra_rdoc_files: 
+- History.txt
+- README.txt
+files: 
+- History.txt
+- README.txt
+- extconf.rb
+- rbzmq.cpp
+- lib/rb_zmq.rb
+has_rdoc: true
+homepage: http://www.zeromq.org
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --main
+- README.txt
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: rb-zmq
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: This gem comprises the ruby bindings for ZeroMQ
+test_files: 
+- test/test_rb_zmq.rb