Sha256: 16b7b8923cc19b547cf08f805d3f90cec95147bf93bbc82e0304c9dd44e775ac

Contents?: true

Size: 1.79 KB

Versions: 8

Compression:

Stored size: 1.79 KB

Contents

require 'protobuf/rpc/servers/zmq/broker'
require 'protobuf/rpc/servers/zmq/worker'
require 'protobuf/rpc/servers/zmq/util'

module Protobuf
  module Rpc
    module Zmq
      class Server
        include ::Protobuf::Rpc::Zmq::Util

        ##
        # Class Methods
        #
        def self.run(options = {})
          @options = options

          unless options[:workers_only]
            log_debug { sign_message("initializing broker") }
            @broker = ::Protobuf::Rpc::Zmq::Broker.new(options)
          end

          local_worker_threads = options[:threads]
          log_debug { sign_message("starting server workers") }

          local_worker_threads.times do
            self.start_worker
          end

          @running = true
          log_debug { sign_message("server started") }
          while self.running? do
            if options[:workers_only]
              sleep 5
              Thread.pass
            else
              @broker.poll
            end
          end
        ensure
          @broker.teardown if @broker
        end

        def self.running?
          !!@running
        end

        def self.start_worker
          @threads << Thread.new(@options) { |options|
            begin
              ::Protobuf::Rpc::Zmq::Worker.new(options).run
            rescue => e
              message =  "Worker Failed, spawning new worker: #{e.inspect}\n #{e.backtrace.join($/)}"
              $stderr.puts message
              log_error { message }

              retry if ::Protobuf::Rpc::Zmq::Server.running?
            end
          }
        end

        def self.stop
          @running = false

          @threads.each do |t|
            t.join(5) || t.kill
          end
        end

        def self.threads
          @threads
        end

        @threads ||= []
      end
    end
  end
end

Version data entries

8 entries across 8 versions & 1 rubygems

Version Path
protobuf-2.7.7-java lib/protobuf/rpc/servers/zmq/server.rb
protobuf-2.7.7 lib/protobuf/rpc/servers/zmq/server.rb
protobuf-2.7.6-java lib/protobuf/rpc/servers/zmq/server.rb
protobuf-2.7.6 lib/protobuf/rpc/servers/zmq/server.rb
protobuf-2.7.5-java lib/protobuf/rpc/servers/zmq/server.rb
protobuf-2.7.5 lib/protobuf/rpc/servers/zmq/server.rb
protobuf-2.7.4-java lib/protobuf/rpc/servers/zmq/server.rb
protobuf-2.7.4 lib/protobuf/rpc/servers/zmq/server.rb