summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMartin Sustrik <sustrik@fastmq.commkdir>2009-09-07 11:10:58 +0200
committerMartin Sustrik <sustrik@fastmq.commkdir>2009-09-07 11:10:58 +0200
commit87ccbb9fae0f9234f82f9a9ec6a215f5e73209d3 (patch)
treec2b7da25c754c3a0c1c46b8f07a2148c34fc9232
parent67253f3186488db93dee23bd9194419f58f0b1d3 (diff)
parentd62c74232a4c1edd38004933afd0c7dc119bd98b (diff)
Merge branch 'master' of git@github.com:sustrik/zeromq2
-rw-r--r--perf/ruby/local_lat.rb66
-rw-r--r--perf/ruby/local_thr.rb90
-rw-r--r--perf/ruby/remote_lat.rb64
-rw-r--r--perf/ruby/remote_thr.rb48
-rw-r--r--ruby/Makefile.am2
-rw-r--r--ruby/rbzmq.cpp292
-rw-r--r--ruby/zmq.cpp460
7 files changed, 382 insertions, 640 deletions
diff --git a/perf/ruby/local_lat.rb b/perf/ruby/local_lat.rb
index 3079b2e..b71af80 100644
--- a/perf/ruby/local_lat.rb
+++ b/perf/ruby/local_lat.rb
@@ -18,58 +18,24 @@
require 'librbzmq'
-class AssertionFailure < StandardError
+if ARGV.length != 3
+ puts "usage: local_lat <bind-to> <message-size> <roundtrip-count>"
+ Process.exit
end
-
-def assert(bool, message = 'assertion failure')
- raise AssertionFailure.new(message) unless bool
-end
-
- if ARGV.length != 4
- puts "usage: local_lat <in-interface> <out-interface> <message-size>
- <roundtrip-count>"
- Process.exit
- end
-
- in_interface = ARGV[0]
- out_interface = ARGV[1]
- message_size = ARGV[2]
- roundtrip_count = ARGV[3]
-
- # Print out the test parameters.
- puts "message size: #{message_size} [B]"
- puts "roundtrip count: #{roundtrip_count}"
-
- # Create 0MQ transport.
- rb_zmq = Zmq.new()
- # Create the wiring.
- context = rb_zmq.context(1,1)
- in_socket = rb_zmq.socket(context, ZMQ_SUB)
- out_socket = rb_zmq.socket(context, ZMQ_PUB)
-
- # Bind.
- rb_zmq.bind(in_socket, in_interface.to_s)
- rb_zmq.bind(out_socket, out_interface.to_s)
-
- # Create message data to send.
- out_msg = rb_zmq.msg_init_size(message_size.to_i)
-
- # Get initial timestamp.
- start_time = Time.now
+bind_to = ARGV[0]
+message_size = ARGV[1].to_i
+roundtrip_count = ARGV[2].to_i
+
+ctx = Context.new(1, 1)
+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
- # The message loop.
- for i in 0...roundtrip_count.to_i do
- rb_zmq.send(out_socket, out_msg, ZMQ_NOBLOCK)
- in_buf = rb_zmq.recv(in_socket, ZMQ_NOBLOCK)
- assert(rb_zmq.msg_size(in_buf.msg) == message_size.to_i)
- end
+sleep 1
- # Get final timestamp.
- end_time = Time.now
- # Compute and print out the latency.
- latency = (end_time.to_f - start_time.to_f) *1000000 /
- roundtrip_count.to_i / 2
- puts "Your average latency is " + "%0.2f" % latency + "[us]"
-
diff --git a/perf/ruby/local_thr.rb b/perf/ruby/local_thr.rb
index 6e2c62a..4d9357b 100644
--- a/perf/ruby/local_thr.rb
+++ b/perf/ruby/local_thr.rb
@@ -17,74 +17,40 @@
# along with this program. If not, see <http://www.gnu.org/licenses/>.
require 'librbzmq'
-class Context
-end
-class Socket
+if ARGV.length != 3
+ puts "usage: local_thr <bind-to> <message-size> <message-count>"
+ Process.exit
end
-class AssertionFailure < StandardError
-end
+bind_to = ARGV[0]
+message_size = ARGV[1].to_i
+message_count = ARGV[2].to_i
+
+ctx = Context.new(1, 1)
+s = Socket.new(ctx, SUB);
+s.bind(bind_to);
-def assert(bool, message = 'assertion failure')
- raise AssertionFailure.new(message) unless bool
+msg = s.recv(0)
+
+start_time = Time.now
+
+for i in 1...message_count.to_i do
+ msg = s.recv(0)
end
- if ARGV.length != 3
- puts "usage: local_thr <in-interface> <message-size>" + \
- " <message-count>"
- Process.exit
- end
+end_time = Time.now
- in_interface = ARGV[0]
- message_size = ARGV[1]
- message_count = ARGV[2]
-
- # Print out the test parameters.
- puts "message size: " + message_size.to_s + " [B]"
- puts "message count: " + message_count.to_s
-
- # Create 0MQ transport.
- rb_zmq = Zmq.new();
-
- # Create context.
- context = rb_zmq.context(1, 1);
-
- # Create the socket.
- in_socket = rb_zmq.socket(context, ZMQ_SUB);
-
- # Connect.
- rb_zmq.connect(in_socket, in_interface.to_s);
-
- # Receive first message
- data = rb_zmq.recv(in_socket, ZMQ_NOBLOCK);
- assert(rb_zmq.msg_size(data.msg) == message_size.to_i)
-
- # Get initial timestamp.
- start_time = Time.now
-
- # The message loop.
- for i in 0...message_count.to_i-1 do
- data = rb_zmq.recv(in_socket, ZMQ_NOBLOCK);
- assert(rb_zmq.msg_size(data.msg) == message_size.to_i)
- end
-
- # Get terminal timestamp.
- end_time = Time.now
-
- # Compute and print out the throughput.
- if end_time.to_f - start_time.to_f != 0
- message_throughput = message_count.to_i /
- (end_time.to_f - start_time.to_f);
- else
- message_throughput = message_count.to_i
- end
-
- megabit_throughput = message_throughput.to_f * message_size.to_i * 8 /
- 1000000;
- puts "Your average throughput is " + "%0.2f" % message_throughput.to_s +
- " [msg/s]"
- puts "Your average throughput is " + "%0.2f" % megabit_throughput.to_s +
- " [Mb/s]"
+elapsed = (end_time.to_f - start_time.to_f) * 1000000
+if elapsed == 0
+ elapsed = 1
+end
+
+throughput = message_count * 1000000 / elapsed
+megabits = throughput * message_size * 8 / 1000000
+puts "message size: %i [B]" % message_size
+puts "message count: %i" % message_count
+puts "mean throughput: %i [msg/s]" % throughput
+puts "mean throughput: %.3f [Mb/s]" % megabits
diff --git a/perf/ruby/remote_lat.rb b/perf/ruby/remote_lat.rb
index 953e579..9387fe1 100644
--- a/perf/ruby/remote_lat.rb
+++ b/perf/ruby/remote_lat.rb
@@ -18,44 +18,36 @@
require 'librbzmq'
-class AssertionFailure < StandardError
+if ARGV.length != 3
+ puts "usage: remote_lat <connect-to> <message-size> <roundtrip-count>"
+ Process.exit
end
-def assert(bool, message = 'assertion failure')
- raise AssertionFailure.new(message) unless bool
+connect_to = ARGV[0]
+message_size = ARGV[1].to_i
+roundtrip_count = ARGV[2].to_i
+
+ctx = Context.new(1, 1)
+s = Socket.new(ctx, REQ);
+s.connect(connect_to);
+
+msg = "#{'0'*message_size}"
+
+start_time = Time.now
+
+for i in 0...roundtrip_count do
+ s.send(msg, 0)
+ msg = s.recv(0)
end
- if ARGV.length != 4
- puts "usage: remote_lat <in-interface> <out-interface>" + \
- " <message-size> <roundtrip-count>"
- Process.exit
- end
-
- in_interface = ARGV[0]
- out_interface = ARGV[1]
- message_size = ARGV[2]
- roundtrip_count = ARGV[3]
-
- # Create 0MQ transport.
- rb_zmq = Zmq.new()
-
- # Create the wiring.
- context = rb_zmq.context(1,1)
- in_socket = rb_zmq.socket(context, ZMQ_SUB)
- out_socket = rb_zmq.socket(context, ZMQ_PUB)
-
- # Connect.
- rb_zmq.connect(in_socket, in_interface.to_s)
- rb_zmq.connect(out_socket, out_interface.to_s)
-
- # The message loop.
- for i in 0...roundtrip_count.to_i do
- data = rb_zmq.recv(in_socket, ZMQ_NOBLOCK)
- assert(rb_zmq.msg_size(data.msg) == message_size.to_i)
- rb_zmq.send(out_socket, data.msg, ZMQ_NOBLOCK)
- end
-
- # Wait till all messages are sent.
- sleep 2
-
+end_time = Time.now
+
+elapsed = (end_time.to_f - start_time.to_f) * 1000000
+latency = elapsed / roundtrip_count / 2
+
+puts "message size: %i [B]" % message_size
+puts "roundtrip count: %i" % roundtrip_count
+puts "mean latency: %.3f [us]" % latency
+
+
diff --git a/perf/ruby/remote_thr.rb b/perf/ruby/remote_thr.rb
index 6c3392e..7d55f86 100644
--- a/perf/ruby/remote_thr.rb
+++ b/perf/ruby/remote_thr.rb
@@ -18,38 +18,24 @@
require 'librbzmq'
-class AssertionFailure < StandardError
+if ARGV.length != 3
+ puts "usage: remote_thr <connect-to> <message-size> <message-count>"
+ Process.exit
end
+
+connect_to = ARGV[0]
+message_size = ARGV[1].to_i
+message_count = ARGV[2].to_i
+
+ctx = Context.new(1, 1)
+s = Socket.new(ctx, PUB);
+s.connect(connect_to);
+
+msg = "#{'0'*message_size}"
-def assert(bool, message = 'assertion failure')
- raise AssertionFailure.new(message) unless bool
+for i in 0...message_count do
+ s.send(msg, 0)
end
- if ARGV.length != 3
- puts "usage: remote_thr <out-interface> <message-size> <message-count>"
- Process.exit
- end
-
- out_interface = ARGV[0]
- message_size = ARGV[1]
- message_count = ARGV[2]
-
- # Create 0MQ transport.
- rb_zmq = Zmq.new();
-
- # Create the wiring.
- context = rb_zmq.context(1,1);
- out_socket = rb_zmq.socket(context, ZMQ_PUB);
- rb_zmq.bind(out_socket, out_interface.to_s);
-
- # Create message data to send.
- out_msg = rb_zmq.msg_init_size(message_size.to_s);
-
- # The message loop.
- for i in 0...message_count.to_i + 1 do
- rb_zmq.send(out_socket, out_msg, ZMQ_NOBLOCK);
- end
-
- # Wait till all messages are sent.
- sleep 2
-
+sleep 10
+
diff --git a/ruby/Makefile.am b/ruby/Makefile.am
index 7718e24..148daf0 100644
--- a/ruby/Makefile.am
+++ b/ruby/Makefile.am
@@ -3,7 +3,7 @@ INCLUDES = -I$(top_builddir) -I$(top_srcdir)/include -I$(top_builddir)/include
rblib_LTLIBRARIES = librbzmq.la
rblibdir = @RUBYDIR@
-librbzmq_la_SOURCES = zmq.cpp
+librbzmq_la_SOURCES = rbzmq.cpp
librbzmq_la_LDFLAGS = -version-info @RBLTVER@
librbzmq_la_CXXFLAGS = -Wall -pedantic -Werror -Wno-long-long
diff --git a/ruby/rbzmq.cpp b/ruby/rbzmq.cpp
new file mode 100644
index 0000000..5cb7f61
--- /dev/null
+++ b/ruby/rbzmq.cpp
@@ -0,0 +1,292 @@
+/*
+ Copyright (c) 2007-2009 FastMQ Inc.
+
+ 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 <zmq.h>
+#include <ruby.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_)
+{
+ assert (!DATA_PTR (self_));
+ void *ctx = zmq_init (NUM2INT (app_threads_), NUM2INT (io_threads_));
+ if (!ctx) {
+ rb_raise (rb_eRuntimeError, 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, strerror (errno));
+ return Qnil;
+ }
+
+ DATA_PTR (self_) = (void*) s;
+ return self_;
+}
+
+/*
+static VALUE rb_setsockopt (VALUE self_, VALUE socket_, VALUE option_,
+ VALUE optval_)
+{
+ // Get the socket.
+ void* socket;
+ Data_Get_Struct (socket_, void*, socket);
+
+ int rc = 0;
+
+ if (TYPE (optval_) == T_STRING) {
+
+ // Forward the code to native 0MQ library.
+ rc = zmq_setsockopt (socket, NUM2INT (option_),
+ (void *) StringValueCStr (optval_), RSTRING_LEN (optval_));
+
+ }
+ else if (TYPE (optval_) == T_FLOAT) {
+
+ double optval = NUM2DBL (optval_);
+
+ // Forward the code to native 0MQ library.
+ rc = zmq_setsockopt (socket, NUM2INT (option_),
+ (void*) &optval, 8);
+ }
+
+ else if (TYPE (optval_) == T_FIXNUM) {
+
+ long optval = FIX2LONG (optval_);
+
+ // Forward the code to native 0MQ library.
+ rc = zmq_setsockopt (socket, NUM2INT (option_),
+ (void *) &optval, 4);
+
+ }
+
+ else if (TYPE (optval_) == T_BIGNUM) {
+
+ long optval = NUM2LONG (optval_);
+
+ // Forward the code to native 0MQ library.
+ rc = zmq_setsockopt (socket, NUM2INT (option_),
+ (void *) &optval, 4);
+
+ }
+ else if (TYPE (optval_) == T_ARRAY) {
+
+ // Forward the code to native 0MQ library.
+ rc = zmq_setsockopt (socket, NUM2INT (option_),
+ (void *) RARRAY_PTR (optval_), RARRAY_LEN (optval_));
+
+ }
+
+ else if (TYPE (optval_) == T_STRUCT) {
+
+ // Forward the code to native 0MQ library.
+ rc = zmq_setsockopt (socket, NUM2INT (option_),
+ (void *) RSTRUCT_PTR (optval_), RSTRUCT_LEN (optval_));
+
+ }
+ else
+ rb_raise(rb_eRuntimeError, "Unknown type");
+
+ assert (rc == 0);
+
+ 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, 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, 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, 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, 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, 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, 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, 2);
+
+ 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 ("MASK", INT2NUM (ZMQ_MASK));
+ rb_define_global_const ("AFFINITY", INT2NUM (ZMQ_AFFINITY));
+ rb_define_global_const ("IDENTITY", INT2NUM (ZMQ_IDENTITY));
+ 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));
+}
diff --git a/ruby/zmq.cpp b/ruby/zmq.cpp
deleted file mode 100644
index 7ab6143..0000000
--- a/ruby/zmq.cpp
+++ /dev/null
@@ -1,460 +0,0 @@
-/*
- Copyright (c) 2007-2009 FastMQ Inc.
-
- 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 <zmq.h>
-#include <zmq/err.hpp>
-#include <ruby.h>
-
-// Class rb_zmq.
-static VALUE rb_zmq;
-
-// Structure to return received data.
-static VALUE rb_data;
-
-static void rb_free (void *p)
-{
-
-}
-
-static VALUE rb_alloc (VALUE self_)
-{
- VALUE obj;
- obj = Data_Wrap_Struct (self_, 0, rb_free, NULL);
-
- return obj;
-}
-
-static VALUE rb_msg_init (VALUE self_)
-{
- zmq_msg_t *msg;
- msg = new zmq_msg_t;
- VALUE obj;
-
- int rc = zmq_msg_init (msg);
- if (rc == -1) {
- assert (errno == ENOMEM);
- rb_raise(rb_eRuntimeError, "Out of memory");
- }
-
- obj = Data_Wrap_Struct (rb_zmq, 0, rb_free, msg);
- return obj;
-}
-
-static VALUE rb_msg_init_size (VALUE self_, VALUE size_)
-{
- zmq_msg_t *msg;
- msg = new zmq_msg_t;
- VALUE obj;
-
- // Forward the code to zmq library.
- int rc = zmq_msg_init_size (msg, NUM2INT (size_));
- if (rc == -1) {
- assert (errno == ENOMEM);
- rb_raise(rb_eRuntimeError, "Out of memory");
- }
-
- obj = Data_Wrap_Struct (rb_zmq, 0, rb_free, msg);
- return obj;
-}
-
-static VALUE rb_msg_init_data (VALUE self_, VALUE data_, VALUE size_)
-{
- // Get the message.
- zmq_msg_t *msg;
- VALUE obj;
- msg = new zmq_msg_t;
-
- // Forward the code to zmq library.
- int rc = zmq_msg_init_data (msg, StringValueCStr (data_),
- NUM2INT (size_), rb_free);
- assert (rc == 0);
-
- obj = Data_Wrap_Struct (rb_zmq, 0, rb_free, msg);
- return obj;
-}
-
-static VALUE rb_msg_close (VALUE self_, VALUE msg_)
-{
- // Get the message.
- zmq_msg_t* msg;
- Data_Get_Struct (msg_, zmq_msg_t, msg);
-
- // Forward the code to zmq library.
- int rc = zmq_close (msg);
- assert (rc == 0);
-
- return self_;
-}
-
-static VALUE rb_msg_move (VALUE self_, VALUE src_)
-{
- // Get the message.
- zmq_msg_t* src;
- Data_Get_Struct (src_, zmq_msg_t, src);
-
- zmq_msg_t *dest;
- dest = new zmq_msg_t;
- VALUE obj;
-
- // Forward the code to zmq library.
- int rc = zmq_msg_move (dest, src);
- assert (rc == 0);
-
- obj = Data_Wrap_Struct (rb_zmq, 0, rb_free, dest);
- return obj;
-}
-
-
-static VALUE rb_msg_copy (VALUE self_, VALUE src_)
-{
- // Get the message.
- zmq_msg_t* src;
- Data_Get_Struct (src_, zmq_msg_t, src);
-
- zmq_msg_t *dest;
- dest = new zmq_msg_t;
- VALUE obj;
-
- // Forward the code to zmq library.
- int rc = zmq_msg_copy (dest, src);
- assert (rc == 0);
-
- obj = Data_Wrap_Struct (rb_zmq, 0, rb_free, dest);
- return obj;
-}
-
-static VALUE rb_msg_data (VALUE self_, VALUE msg_)
-{
- // Get the message.
- zmq_msg_t* msg;
- Data_Get_Struct (msg_, zmq_msg_t, msg);
-
- const char* data;
-
- // Forward the code to zmq library.
- data = (const char*) zmq_msg_data (msg);
-
- return rb_str_new (data, zmq_msg_size (msg));
-}
-
-static VALUE rb_msg_size (VALUE self_, VALUE msg_)
-{
- // Get the message.
- zmq_msg_t* msg;
- Data_Get_Struct (msg_, zmq_msg_t, msg);
-
- // Forward the code to zmq library.
- return INT2NUM (zmq_msg_size (msg));
-}
-
-static VALUE rb_msg_type (VALUE self_, VALUE msg_)
-{
- // Get the message.
- zmq_msg_t* msg;
- Data_Get_Struct (msg_, zmq_msg_t, msg);
-
- // Forward the code to zmq library.
- return INT2NUM (zmq_msg_type (msg));
-}
-
-static VALUE rb_init (VALUE self_)
-{
- return self_;
-}
-
-static VALUE rb_context (VALUE self_, VALUE app_threads_, VALUE io_threads_)
-{
- void *context;
- VALUE obj;
-
- // Forward the code to zmq library.
- context = zmq_init (NUM2INT (app_threads_), NUM2INT (io_threads_));
- if (context == NULL) {
- assert (errno == EINVAL);
- rb_raise(rb_eRuntimeError, "Invalid argument");
- }
-
- obj = Data_Wrap_Struct (rb_zmq, 0, free, context);
- return self_;
-}
-
-static VALUE rb_term (VALUE self_, VALUE context_)
-{
- // Get the context.
- void* context;
- Data_Get_Struct (context_, void*, context);
-
- // Forward the code to zmq library.
- int rc = zmq_term ((void*) context);
- assert (rc == 0);
-
- return self_;
-}
-
-static VALUE rb_socket (VALUE self_, VALUE context_, VALUE type_)
-{
- // Get the context.
- void* context;
- Data_Get_Struct (context_, void*, context);
-
- void* socket = NULL;
- VALUE obj;
-
- // Forward the call to native 0MQ library.
- socket = zmq_socket(context, NUM2INT (type_));
- if (socket == NULL) {
- assert (errno == EMFILE || errno == EINVAL);
- if (errno == EMFILE)
- rb_raise(rb_eRuntimeError, "Too many threads");
- else
- rb_raise(rb_eRuntimeError, "Invalid argument");
- }
-
- obj = Data_Wrap_Struct(rb_zmq, 0, free, socket);
- return obj;
-}
-
-static VALUE rb_close (VALUE self_, VALUE socket_)
-{
- // Get the message.
- void* socket;
- Data_Get_Struct (socket_, void*, socket);
-
- // Forward the call to native 0MQ library.
- int rc = zmq_close (socket);
- assert (rc == 0);
- return self_;
-}
-
-static VALUE rb_setsockopt (VALUE self_, VALUE socket_, VALUE option_,
- VALUE optval_)
-{
- // Get the socket.
- void* socket;
- Data_Get_Struct (socket_, void*, socket);
-
- int rc = 0;
-
- if (TYPE (optval_) == T_STRING) {
-
- // Forward the code to native 0MQ library.
- rc = zmq_setsockopt (socket, NUM2INT (option_),
- (void *) StringValueCStr (optval_), RSTRING_LEN (optval_));
-
- }
- else if (TYPE (optval_) == T_FLOAT) {
-
- double optval = NUM2DBL (optval_);
-
- // Forward the code to native 0MQ library.
- rc = zmq_setsockopt (socket, NUM2INT (option_),
- (void*) &optval, 8);
- }
-
- else if (TYPE (optval_) == T_FIXNUM) {
-
- long optval = FIX2LONG (optval_);
-
- // Forward the code to native 0MQ library.
- rc = zmq_setsockopt (socket, NUM2INT (option_),
- (void *) &optval, 4);
-
- }
-
- else if (TYPE (optval_) == T_BIGNUM) {
-
- long optval = NUM2LONG (optval_);
-
- // Forward the code to native 0MQ library.
- rc = zmq_setsockopt (socket, NUM2INT (option_),
- (void *) &optval, 4);
-
- }
- else if (TYPE (optval_) == T_ARRAY) {
-
- // Forward the code to native 0MQ library.
- rc = zmq_setsockopt (socket, NUM2INT (option_),
- (void *) RARRAY_PTR (optval_), RARRAY_LEN (optval_));
-
- }
-
- else if (TYPE (optval_) == T_STRUCT) {
-
- // Forward the code to native 0MQ library.
- rc = zmq_setsockopt (socket, NUM2INT (option_),
- (void *) RSTRUCT_PTR (optval_), RSTRUCT_LEN (optval_));
-
- }
- else
- rb_raise(rb_eRuntimeError, "Unknown type");
-
- assert (rc == 0);
-
- return self_;
-}
-
-static VALUE rb_bind (VALUE self_, VALUE socket_, VALUE addr_)
-{
- // Get the socket.
- void* socket;
- Data_Get_Struct (socket_, void*, socket);
-
- // Forward the code to native 0MQ library.
- int rc = zmq_bind (socket, StringValueCStr (addr_));
- if (rc == -1) {
- assert (errno == EINVAL || errno == EADDRINUSE);
- if (errno == EINVAL)
- rb_raise(rb_eRuntimeError, "Invalid argument");
- else
- rb_raise(rb_eRuntimeError, "Address in use");
- }
-
- return self_;
-}
-
-static VALUE rb_connect (VALUE self_, VALUE socket_, VALUE addr_)
-{
- // Get the socket.
- void* socket;
- Data_Get_Struct (socket_, void*, socket);
-
- // Forward the code to native 0MQ library.
- int rc = zmq_connect (socket, StringValueCStr (addr_));
- if (rc == -1) {
- assert (errno == EINVAL || errno == EADDRINUSE);
- if (errno == EINVAL)
- rb_raise(rb_eRuntimeError, "Invalid argument");
- else
- rb_raise(rb_eRuntimeError, "Address in use");
- }
- return self_;
-}
-
-static VALUE rb_send (VALUE self_, VALUE socket_, VALUE msg_, VALUE flags_)
-{
- // Get the socket.
- void* socket;
- Data_Get_Struct (socket_, void*, socket);
-
- // Get the message.
- zmq_msg_t *msg;
- Data_Get_Struct (msg_, zmq_msg_t, msg);
-
- // Forward the code to native 0MQ library.
- int rc = zmq_send (socket, msg, NUM2INT (flags_));
- assert (rc == 0 || (rc == -1 && errno == EAGAIN));
-
- return INT2NUM (rc);
-}
-
-static VALUE rb_flush (VALUE self_, VALUE socket_)
-{
- // Get the socket.
- void* socket;
- Data_Get_Struct (socket_, void*, socket);
-
- // Forward the code to native 0MQ library.
- int rc = zmq_flush (socket);
- assert (rc == 0);
-
- return self_;
-}
-
-static VALUE rb_recv (VALUE self_, VALUE socket_, VALUE flags_)
-{
- // Get the socket.
- void* socket;
- Data_Get_Struct (socket_, void*, socket);
-
- // Get the message.
- zmq_msg_t* msg;
- msg = new zmq_msg_t;
- VALUE obj;
-
- // Forward the code to native 0MQ library.
- int rc = zmq_recv (socket, msg, NUM2INT (flags_));
- assert (rc == 0 || (rc == -1 && errno == EAGAIN));
-
- obj = Data_Wrap_Struct(rb_data, 0, rb_free, msg);
-
- return rb_struct_new (obj, rc, NULL);
-}
-
-extern "C"
-void Init_librbzmq() {
-
- // Define the rb_zmq class.
- rb_zmq = rb_define_class ("Zmq", rb_cObject);
-
- // Define allocation function for rb_zmq class.
- rb_define_alloc_func (rb_zmq, rb_alloc);
-
- // Parameters: <name_of_class> <name_of_method_aaccessible_from_ruby>
- // <name_of_method_in_the_class> <number_of_arguments>
- // number of arguments is alqays except for the 'VALUE self_' argument
- // (this pointer).
- rb_define_method (rb_zmq, "msg_init", (VALUE(*)(...)) rb_msg_init, 0);
- rb_define_method (rb_zmq, "msg_init_size", (VALUE(*)(...))
- rb_msg_init_size, 1);
- rb_define_method (rb_zmq, "msg_init_data", (VALUE(*)(...))
- rb_msg_init_data, 3);
- rb_define_method (rb_zmq, "msg_close", (VALUE(*)(...)) rb_msg_close, 1);
- rb_define_method (rb_zmq, "msg_move", (VALUE(*)(...)) rb_msg_move, 1);
- rb_define_method (rb_zmq, "msg_copy", (VALUE(*)(...)) rb_msg_copy, 1);
- rb_define_method (rb_zmq, "msg_data", (VALUE(*)(...)) rb_msg_data, 1);
- rb_define_method (rb_zmq, "msg_size", (VALUE(*)(...)) rb_msg_size, 1);
- rb_define_method (rb_zmq, "msg_type", (VALUE(*)(...)) rb_msg_type, 1);
-
- rb_define_method (rb_zmq, "initialize", (VALUE(*)(...)) rb_init, 0);
- rb_define_method (rb_zmq, "term", (VALUE(*)(...)) rb_term, 1);
- rb_define_method (rb_zmq, "free", (VALUE(*)(...)) rb_free, 0);
-
- rb_define_method (rb_zmq, "context", (VALUE(*)(...)) rb_context, 2);
- rb_define_method (rb_zmq, "socket", (VALUE(*)(...)) rb_socket, 2);
- rb_define_method (rb_zmq, "close", (VALUE(*)(...)) rb_close, 1);
- rb_define_method (rb_zmq, "setsockopt", (VALUE(*)(...)) rb_setsockopt, 3);
- rb_define_method (rb_zmq, "bind", (VALUE(*)(...)) rb_bind, 2);
- rb_define_method (rb_zmq, "connect", (VALUE(*)(...)) rb_connect, 2);
- rb_define_method (rb_zmq, "send", (VALUE(*)(...)) rb_send, 3);
- rb_define_method (rb_zmq, "flush", (VALUE(*)(...)) rb_flush, 1);
- rb_define_method (rb_zmq, "recv", (VALUE(*)(...)) rb_recv, 2);
-
- // Define structure to hold data that are returned from receive function.
- rb_data = rb_struct_define (NULL, "msg", "rc", NULL);
- rb_define_const (rb_zmq, "DATA", rb_data);
-
- // Define global constants.
- rb_define_global_const ("ZMQ_MAX_VSM_SIZE", INT2NUM (ZMQ_MAX_VSM_SIZE));
- rb_define_global_const ("ZMQ_GAP", INT2NUM (ZMQ_GAP));
- rb_define_global_const ("ZMQ_DELIMITER", INT2NUM (ZMQ_DELIMITER));
- rb_define_global_const ("ZMQ_VSM", INT2NUM (ZMQ_VSM));
- rb_define_global_const ("ZMQ_HWM", INT2NUM (ZMQ_HWM));
- rb_define_global_const ("ZMQ_SWAP", INT2NUM (ZMQ_SWAP));
- rb_define_global_const ("ZMQ_MASK", INT2NUM (ZMQ_MASK));
- rb_define_global_const ("ZMQ_AFFINITY", INT2NUM (ZMQ_AFFINITY));
- rb_define_global_const ("ZMQ_IDENTITY", INT2NUM (ZMQ_IDENTITY));
- rb_define_global_const ("ZMQ_NOBLOCK", INT2NUM (ZMQ_NOBLOCK));
- rb_define_global_const ("ZMQ_NOFLUSH", INT2NUM (ZMQ_NOFLUSH));
- rb_define_global_const ("ZMQ_P2P", INT2NUM (ZMQ_P2P));
- rb_define_global_const ("ZMQ_SUB", INT2NUM (ZMQ_SUB));
- rb_define_global_const ("ZMQ_PUB", INT2NUM (ZMQ_PUB));
- rb_define_global_const ("ZMQ_REQ", INT2NUM (ZMQ_REQ));
- rb_define_global_const ("ZMQ_REP", INT2NUM (ZMQ_REP));
-
-}