Kouhei Sutou
null+****@clear*****
Tue Jan 21 12:06:32 JST 2014
Kouhei Sutou 2014-01-21 12:06:32 +0900 (Tue, 21 Jan 2014) New Revision: 77e17e74e6335110b606e70f0d2a7abfcdbb0f6e https://github.com/droonga/drnbench/commit/77e17e74e6335110b606e70f0d2a7abfcdbb0f6e Message: Teardown subscribers Modified files: lib/drnbench/publish-subscribe/runner.rb Modified: lib/drnbench/publish-subscribe/runner.rb (+14 -4) =================================================================== --- lib/drnbench/publish-subscribe/runner.rb 2014-01-21 11:58:04 +0900 (2daeeae) +++ lib/drnbench/publish-subscribe/runner.rb 2014-01-21 12:06:32 +0900 (1d761cc) @@ -20,7 +20,7 @@ module Drnbench @published_messages = Queue.new setup_server - setup_initial_subscribers + setup_subscribers end def setup_server @@ -36,8 +36,15 @@ module Drnbench @engine.stop end - def setup_initial_subscribers - add_subscribers(@config.start_n_subscribers) + def setup_subscribers + add_subscribers(@config.start_n_subscribers, + @config.n_publishings) + end + + def teardown_subscribers + @subscribers.each do |subscriber| + subscriber.close + end end def run @@ -51,15 +58,18 @@ module Drnbench published_messages end - def add_subscribers(n_subscribers) + def add_subscribers(n_subscribers, n_expected_messages) progressbar = ProgressBar.new("subscribe", n_subscribers, STDERR) n_subscribers.times do |index| message =****@confi*****_subscribe_request client = Droonga::Client.new(:protocol => :http, :host => @config.protocol_adapter.host, :port => @config.protocol_adapter.port) + n_received_messages = 0 client.subscribe(message) do |published_message| @published_messages.push(published_message) + n_received_messages += 1 + break if n_received_messages == n_expected_messages end @subscribers << client progressbar.inc -------------- next part -------------- HTML����������������������������...Download