diff --git a/run_test.rb b/run_test.rb index 04636ee..e109dbd 100755 --- a/run_test.rb +++ b/run_test.rb @@ -54,11 +54,14 @@ sources.each do |type, host| sky = (type == :firehose) ? Skyfall::Firehose.new(host) : Skyfall::Jetstream.new(host) events = 0 + reconnects = 0 + errors = 0 + users = Set.new minute = Time.now.to_i / verbose if verbose + connected = false - reconnects = 0 - errors = 0 + got_first_event = false sky.on_connecting do puts "[#{Time.now}] #{host}: Connecting..." @@ -84,7 +87,13 @@ sources.each do |type, host| users << msg.did if verbose + if !got_first_event + puts "[#{Time.now}] #{host}: Starting at seq #{msg.seq}" + got_first_event = true + end + now = Time.now.to_i / verbose + if now > minute puts "[#{Time.now}] #{host.ljust(maxlen)} | events: #{events.to_s.ljust(8)} | users: #{users.size}" minute = now @@ -95,7 +104,8 @@ sources.each do |type, host| trap('SIGINT') { sky.disconnect } sky.connect - puts "[#{Time.now}] #{host}: Finished." + + puts "[#{Time.now}] #{host}: Finished at seq #{sky.cursor}" output.puts(JSON.generate({ events: events, users: users.size, connected: connected, errors: errors, reconnects: reconnects }))