1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
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
|
