Sha256: 0f2f7846abaa382a53e0aefab80aa27f6c23ec00e4b935e658473405edcd5450

Contents?: true

Size: 1.45 KB

Versions: 5

Compression:

Stored size: 1.45 KB

Contents

require 'ostruct'
require 'thread'

module Protobuf
  module Rpc
    class ZmqRunner
      include ::Protobuf::Logging

      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
                     fail "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(:TRAP) do
          ::Thread.list.each do |thread|
            logger.info do
              <<-THREAD_TRACE
                #{thread.inspect}:
                  #{thread.backtrace.join($INPUT_RECORD_SEPARATOR)}"
              THREAD_TRACE
            end
          end
        end

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

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

Version data entries

5 entries across 5 versions & 1 rubygems

Version Path
protobuf-3.4.4 lib/protobuf/rpc/servers/zmq_runner.rb
protobuf-3.4.3 lib/protobuf/rpc/servers/zmq_runner.rb
protobuf-3.4.2 lib/protobuf/rpc/servers/zmq_runner.rb
protobuf-3.4.1 lib/protobuf/rpc/servers/zmq_runner.rb
protobuf-3.4.0 lib/protobuf/rpc/servers/zmq_runner.rb