Module: Thrift::Client

Defined in:
lib/thrift/client.rb

Instance Method Summary collapse

Instance Method Details

#handle_exception(mtype) ⇒ Object 



53
54
55
56
57
58
59
60
 
# File 'lib/thrift/client.rb', line 53

def handle_exception(mtype)
  if mtype == MessageTypes::EXCEPTION
    x = ApplicationException.new
    x.read(@iprot)
    @iprot.read_message_end
    raise x
  end
end

#initialize(iprot, oprot = nil) ⇒ Object 



22
23
24
25
26
 
# File 'lib/thrift/client.rb', line 22

def initialize(iprot, oprot=nil)
  @iprot = iprot
  @oprot = oprot || iprot
  @seqid = 0
end

#receive_message(result_klass) ⇒ Object 



44
45
46
47
48
49
50
51
 
# File 'lib/thrift/client.rb', line 44

def receive_message(result_klass)
  fname, mtype, rseqid = @iprot.read_message_begin
  handle_exception(mtype)
  result = result_klass.new
  result.read(@iprot)
  @iprot.read_message_end
  result
end

#send_message(name, args_class, args = {}) ⇒ Object 



28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
 
# File 'lib/thrift/client.rb', line 28

def send_message(name, args_class, args = {})
  @oprot.write_message_begin(name, MessageTypes::CALL, @seqid)
  data = args_class.new
  args.each do |k, v|
    data.send("#{k.to_s}=", v)
  end
  begin
    data.write(@oprot)
  rescue StandardError => e
    @oprot.trans.close
    raise e
  end
  @oprot.write_message_end
  @oprot.trans.flush
end