Contents

require 'ostruct'

module Protobuf
  module Rpc
    class ZmqRunner
      include ::Protobuf::Logger::LogMethods

      def initialize(options)
        @options = case
                   when options.is_a?(OpenStruct) then
                     options.marshal_dump
                   when options.respond_to?(:to_hash) then
                     options.to_hash
                   else
                     raise "Cannot parser Zmq Server - server options"
                   end

      end

      def run
        @server = ::Protobuf::Rpc::Zmq::Server.new(@options)
        register_signals
        @server.run do 
          yield if block_given?
        end
      end

      def running?
        @server.try :running?
      end

      def stop
        @server.try :stop
      end

      private

      def register_signals
        trap(:TTIN) do
          @server.add_worker
          log_info { "Increased worker size to: #{@server.total_workers}" }
        end

        trap(:TTOU) do
          log_info { "Current worker size: #{@server.workers.size}" }
          log_info { "Current worker size: #{@server.busy_worker_count}" }
        end
      end
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
protobuf-2.8.13	lib/protobuf/rpc/servers/zmq_runner.rb
protobuf-2.8.12	lib/protobuf/rpc/servers/zmq_runner.rb
protobuf-2.8.11	lib/protobuf/rpc/servers/zmq_runner.rb