module BERTRPC class Action include Encodes def initialize(svc, req, mod, fun, args) @svc = svc @req = req @mod = mod @fun = fun @args = args end def execute bert_request = encode_ruby_request(t[@req.kind, @mod, @fun, @args]) bert_response = transaction(bert_request) decode_bert_response(bert_response) end #private def write(sock, bert) sock.write([bert.length].pack("N")) sock.write(bert) end def transaction(bert_request) sock = connect_to(@svc.host, @svc.port, @svc.timeout) if @req.options if @req.options[:cache] && @req.options[:cache][0] == :validation token = @req.options[:cache][1] info_bert = encode_ruby_request([:info, :cache, [:validation, token]]) write(sock, info_bert) end end write(sock, bert_request) lenheader = sock.read(4) raise ProtocolError.new(ProtocolError::NO_HEADER) unless lenheader len = lenheader.unpack('N').first bert_response = sock.read(len) raise ProtocolError.new(ProtocolError::NO_DATA) unless bert_response sock.close bert_response rescue Errno::ECONNREFUSED raise ConnectionError.new("Unable to connect to #{@svc.host}:#{@svc.port}") rescue Timeout::Error raise ReadTimeoutError.new(@svc.host, @svc.port, @svc.timeout) end # Creates a socket object which does speedy, non-blocking reads # and can perform reliable read timeouts. # # Raises Timeout::Error on timeout. # # +host+ String address of the target TCP server # +port+ Integer port of the target TCP server # +timeout+ Optional Integer (in seconds) of the read timeout def connect_to(host, port, timeout = nil) io = BufferedIO.new(TCPSocket.new(host, port)) io.read_timeout = timeout io end end end