diff options
Diffstat (limited to 'perf/ruby/remote_lat.rb')
-rw-r--r-- | perf/ruby/remote_lat.rb | 61 |
1 files changed, 61 insertions, 0 deletions
diff --git a/perf/ruby/remote_lat.rb b/perf/ruby/remote_lat.rb new file mode 100644 index 0000000..953e579 --- /dev/null +++ b/perf/ruby/remote_lat.rb @@ -0,0 +1,61 @@ +# +# 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/>. + +require 'librbzmq' + +class AssertionFailure < StandardError +end + +def assert(bool, message = 'assertion failure') + raise AssertionFailure.new(message) unless bool +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 + + |