Sha256: c3643a33e692baacba637fc58891b0531488922766a693419a01f1462b1ba26d

Contents?: true

Size: 1.97 KB

Versions: 30

Compression:

Stored size: 1.97 KB

Contents

require 'protobuf/rpc/connectors/base'

module Protobuf
  module Rpc
    module Connectors
      class Socket < Base
        include Protobuf::Rpc::Connectors::Common
        include Protobuf::Logger::LogMethods

        def send_request
          timeout_wrap do
            setup_connection
            connect_to_rpc_server
            post_init
            read_response
          end
        end

        def log_signature
          @_log_signature ||= "[client-#{self.class}]"
        end

        private

        def close_connection
          @socket.close
          log_debug { sign_message('Connector closed')  }
        end

        def connect_to_rpc_server
          @socket = TCPSocket.new(options[:host], options[:port])
          log_debug { sign_message("Connection established #{options[:host]}:#{options[:port]}")  }
        end

        # Method to determine error state, must be used with Connector api
        def error?
          return true if @error
          log_debug { sign_message("Error state : #{@socket.closed?}")  }
          @socket.closed?
        end

        def read_data
          size_io = StringIO.new

          until (size_reader = @socket.getc) == "-"
            size_io << size_reader
          end
          str_size_io = size_io.string

          "#{str_size_io}-#{@socket.read(str_size_io.to_i)}"
        end

        def read_response
          log_debug { sign_message("error? is #{error?}") }
          return if error?
          response_buffer = ::Protobuf::Rpc::Buffer.new(:read)
          response_buffer << read_data
          @response_data = response_buffer.data
          parse_response if response_buffer.flushed?
        end

        def send_data
          return if error?
          request_buffer = ::Protobuf::Rpc::Buffer.new(:write)
          request_buffer.set_data(@request_data)
          @socket.write(request_buffer.write)
          @socket.flush
          log_debug { sign_message("write closed") }
        end
      end
    end
  end
end

Version data entries

30 entries across 30 versions & 2 rubygems

Version Path
protobuffy-3.6.0 lib/protobuf/rpc/connectors/socket.rb
protobuffy-3.5.1 lib/protobuf/rpc/connectors/socket.rb
protobuffy-3.4.0 lib/protobuf/rpc/connectors/socket.rb
protobuffy-3.3.0 lib/protobuf/rpc/connectors/socket.rb
protobuffy-3.2.0 lib/protobuf/rpc/connectors/socket.rb
protobuffy-3.1.0 lib/protobuf/rpc/connectors/socket.rb
protobuf-3.2.1 lib/protobuf/rpc/connectors/socket.rb
protobuf-3.2.0 lib/protobuf/rpc/connectors/socket.rb
protobuf-3.1.0 lib/protobuf/rpc/connectors/socket.rb
protobuf-3.0.5 lib/protobuf/rpc/connectors/socket.rb
protobuf-3.0.4 lib/protobuf/rpc/connectors/socket.rb
protobuf-2.8.13 lib/protobuf/rpc/connectors/socket.rb
protobuf-3.0.3 lib/protobuf/rpc/connectors/socket.rb
protobuf-3.0.2 lib/protobuf/rpc/connectors/socket.rb
protobuf-3.0.1 lib/protobuf/rpc/connectors/socket.rb
protobuf-3.0.0 lib/protobuf/rpc/connectors/socket.rb
protobuf-3.0.0.rc2 lib/protobuf/rpc/connectors/socket.rb
protobuf-3.0.0.rc1 lib/protobuf/rpc/connectors/socket.rb
protobuf-2.8.12 lib/protobuf/rpc/connectors/socket.rb
protobuf-2.8.11 lib/protobuf/rpc/connectors/socket.rb