ffi-rzmq 0.8.2 → 0.9.0

data/examples/v3api/local_lat.rb

@@ -0,0 +1,59 @@
+#
+#    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 File.join(File.dirname(__FILE__), '..', '..', 'lib', 'ffi-rzmq')
+
+if ARGV.length < 3
+  puts "usage: ruby local_lat.rb <connect-to> <message-size> <roundtrip-count>"
+  exit
+end
+
+bind_to = ARGV[0]
+message_size = ARGV[1].to_i
+roundtrip_count = ARGV[2].to_i
+
+def assert(rc)
+  raise "Last API call failed at #{caller(1)}" unless rc >= 0
+end
+
+begin
+  ctx = ZMQ::Context.new
+  s = ctx.socket(ZMQ::REP)
+rescue ContextError => e
+  STDERR.puts "Failed to allocate context or socket!"
+  raise
+end
+
+assert(s.setsockopt(ZMQ::LINGER, 100))
+assert(s.setsockopt(ZMQ::RCVHWM, 100))
+assert(s.setsockopt(ZMQ::SNDHWM, 100))
+
+assert(s.bind(bind_to))
+
+roundtrip_count.times do
+  string = ''
+  assert(s.recv_string(string, 0))
+
+  raise "Message size doesn't match, expected [#{message_size}] but received [#{string.size}]" if message_size != string.size
+
+  assert(s.send_string(string, 0))
+end
+
+assert(s.close)
+
+ctx.terminate

data/examples/v3api/local_lat_poll.rb

@@ -0,0 +1,66 @@
+
+require File.join(File.dirname(__FILE__), '..', '..', 'lib', 'ffi-rzmq')
+
+if ARGV.length < 3
+  puts "usage: ruby local_lat.rb <connect-to> <message-size> <roundtrip-count>"
+  exit
+end
+
+link = ARGV[0]
+message_size = ARGV[1].to_i
+roundtrip_count = ARGV[2].to_i
+
+def assert(rc)
+  raise "Last API call failed at #{caller(1)}" unless rc >= 0
+end
+
+begin
+  ctx = ZMQ::Context.new
+  s1 = ctx.socket(ZMQ::REQ)
+  s2 = ctx.socket(ZMQ::REP)
+rescue ContextError => e
+  STDERR.puts "Failed to allocate context or socket!"
+  raise
+end
+
+assert(s1.setsockopt(ZMQ::LINGER, 100))
+assert(s2.setsockopt(ZMQ::LINGER, 100))
+
+assert(s1.connect(link))
+assert(s2.bind(link))
+
+poller = ZMQ::Poller.new
+poller.register_readable(s2)
+poller.register_readable(s1)
+
+
+start_time = Time.now
+
+# kick it off
+message = ZMQ::Message.new("a" * message_size)
+assert(s1.sendmsg(message, ZMQ::DONTWAIT))
+
+i = roundtrip_count
+
+until i.zero?
+  i -= 1
+
+  assert(poller.poll_nonblock)
+
+  poller.readables.each do |socket|
+    received_message = ''
+    assert(socket.recv_string(received_message, ZMQ::DONTWAIT))
+    assert(socket.sendmsg(ZMQ::Message.new(received_message), ZMQ::DONTWAIT))
+  end
+end
+
+elapsed_usecs = (Time.now.to_f - start_time.to_f) * 1_000_000
+latency = elapsed_usecs / roundtrip_count / 2
+
+puts "mean latency: %.3f [us]" % latency
+puts "received all messages in %.3f seconds" % (elapsed_usecs / 1_000_000)
+
+assert(s1.close)
+assert(s2.close)
+
+ctx.terminate

data/examples/v3api/local_throughput.rb

@@ -0,0 +1,65 @@
+
+require File.join(File.dirname(__FILE__), '..', '..', 'lib', 'ffi-rzmq')
+
+if ARGV.length != 3
+  puts "usage: ruby local_throughtput.rb <bind-to> <message-size> <message-count>"
+  Process.exit
+end
+
+def assert(rc)
+  raise "Last API call failed at #{caller(1)}" unless rc >= 0
+end
+
+bind_to = ARGV[0]
+message_size = ARGV[1].to_i
+message_count = ARGV[2].to_i
+
+begin
+  ctx = ZMQ::Context.new
+  s = ZMQ::Socket.new(ctx.pointer, ZMQ::SUB)
+rescue ContextError => e
+  STDERR.puts "Failed to allocate context or socket!"
+  raise
+end
+
+#assert(s.setsockopt(ZMQ::LINGER, 100))
+assert(s.setsockopt(ZMQ::SUBSCRIBE, ""))
+#assert(s.setsockopt(ZMQ::RCVHWM, 0))
+#assert(s.setsockopt(ZMQ::SNDHWM, 0))
+
+assert(s.bind(bind_to))
+sleep 1
+
+msg = ZMQ::Message.new
+msg = ''
+assert(s.recv_string(msg))
+#assert(s.recvmsg(msg))
+
+start_time = Time.now
+
+i = 1
+while i < message_count
+  #assert(s.recvmsg(msg))
+  assert(s.recv_string(msg))
+  puts i
+  i += 1
+end
+
+end_time = Time.now
+
+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
+
+assert(s.close)
+
+ctx.terminate

data/examples/v3api/publish_subscribe.rb

@@ -0,0 +1,82 @@
+require File.join(File.dirname(__FILE__), '..', '..', 'lib', 'ffi-rzmq')
+
+
+def assert(rc)
+  raise "Last API call failed at #{caller(1)}" unless rc >= 0
+end
+
+link = "tcp://127.0.0.1:5555"
+
+begin
+  ctx = ZMQ::Context.new
+  s1 = ctx.socket(ZMQ::PUB)
+  s2 = ctx.socket(ZMQ::SUB)
+  s3 = ctx.socket(ZMQ::SUB)
+  s4 = ctx.socket(ZMQ::SUB)
+  s5 = ctx.socket(ZMQ::SUB)
+rescue ContextError => e
+  STDERR.puts "Failed to allocate context or socket!"
+  raise
+end
+
+assert(s1.setsockopt(ZMQ::LINGER, 100))
+assert(s2.setsockopt(ZMQ::SUBSCRIBE, '')) # receive all
+assert(s3.setsockopt(ZMQ::SUBSCRIBE, 'animals')) # receive any starting with this string
+assert(s4.setsockopt(ZMQ::SUBSCRIBE, 'animals.dog'))
+assert(s5.setsockopt(ZMQ::SUBSCRIBE, 'animals.cat'))
+
+assert(s1.bind(link))
+assert(s2.connect(link))
+assert(s3.connect(link))
+assert(s4.connect(link))
+assert(s5.connect(link))
+
+sleep 1
+
+topic = "animals.dog"
+payload = "Animal crackers!"
+
+s1.identity = "publisher-A"
+puts "sending"
+# use the new multi-part messaging support to
+# automatically separate the topic from the body
+assert(s1.send_string(topic, ZMQ::SNDMORE))
+assert(s1.send_string(payload, ZMQ::SNDMORE))
+assert(s1.send_string(s1.identity))
+
+topic = ''
+assert(s2.recv_string(topic))
+
+body = ''
+assert(s2.recv_string(body)) if s2.more_parts?
+
+identity = ''
+assert(s2.recv_string(identity)) if s2.more_parts?
+puts "s2 received topic [#{topic}], body [#{body}], identity [#{identity}]"
+
+
+
+topic = ''
+assert(s3.recv_string(topic))
+
+body = ''
+assert(s3.recv_string(body)) if s3.more_parts?
+puts "s3 received topic [#{topic}], body [#{body}]"
+
+topic = ''
+assert(s4.recv_string(topic))
+
+body = ''
+assert(s4.recv_string(body)) if s4.more_parts?
+puts "s4 received topic [#{topic}], body [#{body}]"
+
+s5_string = ''
+rc = s5.recv_string(s5_string, ZMQ::DONTWAIT)
+eagain = (rc == -1 && ZMQ::Util.errno == ZMQ::EAGAIN)
+puts(eagain ? "s5 received no messages" : "s5 FAILED")
+
+[s1, s2, s3, s4, s5].each do |socket|
+  assert(socket.close)
+end
+
+ctx.terminate

data/examples/v3api/remote_lat.rb

@@ -0,0 +1,71 @@
+#
+#    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 File.join(File.dirname(__FILE__), '..', '..', 'lib', 'ffi-rzmq')
+
+if ARGV.length < 3
+  puts "usage: ruby remote_lat.rb <connect-to> <message-size> <roundtrip-count>"
+  exit
+end
+
+def assert(rc)
+  raise "Last API call failed at #{caller(1)}" unless rc >= 0
+end
+
+connect_to = ARGV[0]
+message_size = ARGV[1].to_i
+roundtrip_count = ARGV[2].to_i
+
+begin
+  ctx = ZMQ::Context.new
+  s = ctx.socket(ZMQ::REQ)
+rescue ContextError => e
+  STDERR.puts "Failed to allocate context or socket!"
+  raise
+end
+
+assert(s.setsockopt(ZMQ::LINGER, 100))
+assert(s.connect(connect_to))
+
+msg = "#{ '3' * message_size }"
+
+start_time = Time.now
+
+roundtrip_count.times do
+  assert(s.send_string(msg, 0))
+
+  msg = ''
+  assert(s.recv_string(msg, 0))
+
+  raise "Message size doesn't match, expected [#{message_size}] but received [#{msg.size}]" if message_size != msg.size
+end
+
+end_time = Time.now
+elapsed_secs = (end_time.to_f - start_time.to_f)
+elapsed_usecs = elapsed_secs * 1000000
+latency = elapsed_usecs / roundtrip_count / 2
+
+puts "message size: %i [B]" % message_size
+puts "roundtrip count: %i" % roundtrip_count
+puts "throughput (msgs/s): %i" % (roundtrip_count / elapsed_secs)
+puts "mean latency: %.3f [us]" % latency
+
+assert(s.close)
+
+ctx.terminate

data/examples/v3api/remote_throughput.rb

@@ -0,0 +1,47 @@
+
+require File.join(File.dirname(__FILE__), '..', '..', 'lib', 'ffi-rzmq')
+
+if ARGV.length != 3
+  puts "usage: ruby remote_throughput.rb <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
+
+def assert(rc)
+  raise "Last API call failed at #{caller(1)}" unless rc >= 0
+end
+
+begin
+  ctx = ZMQ::Context.new
+  s = ZMQ::Socket.new(ctx.pointer, ZMQ::PUB)
+rescue ContextError => e
+  STDERR.puts "Could not allocate a context or socket!"
+  raise
+end
+
+#assert(s.setsockopt(ZMQ::LINGER, 1_000))
+#assert(s.setsockopt(ZMQ::RCVHWM, 0))
+#assert(s.setsockopt(ZMQ::SNDHWM, 0))
+assert(s.connect(connect_to))
+
+# the sleep gives the downstream SUB socket a chance to register its
+# subscription filters with this PUB socket
+sleep 1
+
+contents = "#{'0'*message_size}"
+
+i = 0
+while i < message_count
+  msg = ZMQ::Message.new(contents)
+  assert(s.sendmsg(msg))
+  puts i
+  i += 1
+end
+
+sleep 10
+assert(s.close)
+
+ctx.terminate

data/examples/v3api/reqrep_poll.rb

@@ -0,0 +1,62 @@
+
+require File.join(File.dirname(__FILE__), '..', '..', 'lib', 'ffi-rzmq')
+
+
+def assert(rc)
+  raise "Last API call failed at #{caller(1)}" unless rc >= 0
+end
+
+link = "tcp://127.0.0.1:5554"
+
+begin
+  ctx = ZMQ::Context.new
+  s1 = ctx.socket(ZMQ::REQ)
+  s2 = ctx.socket(ZMQ::REP)
+rescue ContextError => e
+  STDERR.puts "Failed to allocate context or socket!"
+  raise
+end
+
+assert(s1.setsockopt(ZMQ::LINGER, 100))
+assert(s2.setsockopt(ZMQ::LINGER, 100))
+
+assert(s1.connect(link))
+assert(s2.bind(link))
+
+poller = ZMQ::Poller.new
+poller.register_readable(s2)
+poller.register_writable(s1)
+
+start_time = Time.now
+@unsent = true
+
+until @done do
+  assert(poller.poll_nonblock)
+
+  # send the message after 5 seconds
+  if Time.now - start_time > 5 && @unsent
+    payload = "#{ '3' * 1024 }"
+
+    puts "sending payload nonblocking"
+    assert(s1.send_string(payload, ZMQ::DONTWAIT))
+    @unsent = false
+  end
+
+  # check for messages after 1 second
+  if Time.now - start_time > 1
+    poller.readables.each do |sock|
+      received_msg = ''
+      assert(sock.recv_string(received_msg, ZMQ::DONTWAIT))
+
+      puts "message received [#{received_msg}]"
+      @done = true
+    end
+  end
+end
+
+puts "executed in [#{Time.now - start_time}] seconds"
+
+assert(s1.close)
+assert(s2.close)
+
+ctx.terminate