protobuf 2.7.3 → 2.7.4

data/lib/protobuf/rpc/servers/zmq/broker.rb

@@ -6,50 +6,84 @@ module Protobuf
     module Zmq
       class Broker
         include ::Protobuf::Rpc::Zmq::Util
-        attr_reader :frontend, :backend, :poller, :context, :available_workers
+        attr_reader :frontend, :backend, :poller, :context, :available_workers, :options, :expected_worker_count
 
         ##
         # Constructor
         #
         def initialize(options = {})
           @available_workers = []
+          @options = options.deep_dup
+          @expected_worker_count = @options[:threads]
           @context = ::ZMQ::Context.new
-          @frontend = setup_frontend(options)
-          @backend = setup_backend(options)
           @poller = ::ZMQ::Poller.new
-          @poller.register(frontend, ::ZMQ::POLLIN)
-          @poller.register(backend, ::ZMQ::POLLIN)
+          setup_backend
         end
 
         ##
         # Instance Methods
         #
         def poll
-          if available_workers.size > 0
-            poller.register(frontend, ::ZMQ::POLLIN) if poller.size < 2
+          if frontend.nil?
+            if local_workers_have_started?            
+              # only open the front end when the workers are done booting
+              log_info { "Starting frontend socket in broker, all workers ready!" }
+              setup_frontend 
+            end
           else
-            poller.delete(frontend)            
+            # Start checking the poller after startup
+            if available_workers.size > 0
+              poller.register(frontend, ::ZMQ::POLLIN) if poller.size < 2
+            else
+              poller.delete(frontend)
+            end
           end
 
           poller.poll(1000)
           poller.readables.each do |socket|
             case socket
-            when frontend then
-              move_to_backend(socket)
             when backend then
               move_to_frontend(socket)
+            when frontend then
+              move_to_backend(socket)
             end
           end
         end
 
+        def setup_backend
+          host = options[:host]
+          port = options[:worker_port]
+
+          zmq_backend = context.socket(::ZMQ::ROUTER)
+          zmq_error_check(zmq_backend.bind(bind_address(host, port)))
+
+          @backend = zmq_backend
+          @poller.register(@backend, ::ZMQ::POLLIN)
+        end
+
+        def setup_frontend
+          host = options[:host]
+          port = options[:port]
+
+          zmq_frontend = context.socket(::ZMQ::ROUTER)
+          zmq_error_check(zmq_frontend.bind(bind_address(host, port)))
+
+          @frontend = zmq_frontend
+          @poller.register(@frontend, ::ZMQ::POLLIN)
+        end
+
         def teardown
-          frontend.close
-          backend.close
-          context.terminate
+          frontend.try(:close)
+          backend.try(:close)
+          context.try(:terminate)
         end
 
         private
 
+          def local_workers_have_started?
+            @local_workers_have_started ||= available_workers.size >= expected_worker_count
+          end
+
           def move_to_backend(socket)
             message_array = []
             zmq_error_check(socket.recv_strings(message_array))
@@ -73,7 +107,9 @@ module Protobuf
             available_workers << message_array[0]
 
             # messages should be [ "uuid of socket", "", "READY_MESSAGE || uuid of client socket"]
-            unless message_array[2] == ::Protobuf::Rpc::Zmq::WORKER_READY_MESSAGE
+            if message_array[2] == ::Protobuf::Rpc::Zmq::WORKER_READY_MESSAGE
+              log_info { "Worker #{available_workers.size} of #{expected_worker_count} ready!" }
+            else
               frontend_message_set = [
                 message_array[2], # client UUID
                 "",
@@ -84,24 +120,6 @@ module Protobuf
             end
           end
 
-          def setup_backend(options = {})
-            host = options[:host]
-            port = options[:worker_port]
-
-            zmq_backend = context.socket(::ZMQ::ROUTER)
-            zmq_error_check(zmq_backend.bind(bind_address(host, port)))
-            zmq_backend
-          end
-
-          def setup_frontend(options = {})
-            host = options[:host]
-            port = options[:port]
-
-            zmq_frontend = context.socket(::ZMQ::ROUTER)
-            zmq_error_check(zmq_frontend.bind(bind_address(host, port)))
-            zmq_frontend
-          end
-
           def bind_address(host, port)
             "tcp://#{resolve_ip(host)}:#{port}"
           end

data/lib/protobuf/rpc/servers/zmq/server.rb

@@ -62,7 +62,7 @@ module Protobuf
           @running = false
 
           @threads.each do |t|
-            t.join
+            t.join(5) || t.kill
           end
         end
 

data/lib/protobuf/version.rb

@@ -1,4 +1,4 @@
 module Protobuf
-  VERSION = '2.7.3'
+  VERSION = '2.7.4'
   PROTOC_VERSION = '2.4.1'
 end

data/spec/lib/protobuf/rpc/servers/zmq/broker_spec.rb

@@ -17,10 +17,6 @@ describe ::Protobuf::Rpc::Zmq::Broker do
     subject.context.should be_a(::ZMQ::Context)
   end
 
-  it 'sets up a frontend socket' do
-    subject.frontend.should be_a(::ZMQ::Socket)
-  end
-
   it 'sets up a backend socket' do
     subject.backend.should be_a(::ZMQ::Socket)
   end

data/spec/lib/protobuf/rpc/servers/zmq/server_spec.rb

@@ -27,7 +27,7 @@ describe Protobuf::Rpc::Zmq::Server do
 
     it 'lets all threads stop' do
       thread_mock = double(Thread)
-      thread_mock.should_receive(:join)
+      thread_mock.should_receive(:join).and_return(thread_mock)
       described_class.instance_variable_set(:@threads, [thread_mock])
       described_class.stop
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: protobuf
 version: !ruby/object:Gem::Version
-  version: 2.7.3
+  version: 2.7.4
   prerelease: 
 platform: ruby
 authors:
@@ -10,7 +10,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-03-04 00:00:00.000000000 Z
+date: 2013-03-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -499,7 +499,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 507016021282761901
+      hash: 933543426026237728
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -508,7 +508,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 507016021282761901
+      hash: 933543426026237728
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.25