Class: GrpcKit::Stream::ClientStream

Inherits:
Object
  • Object
show all
Defined in:
lib/grpc_kit/stream/client_stream.rb

Defined Under Namespace

Classes: Status

Instance Method Summary collapse

Constructor Details

#initialize(transport, config, authority:, timeout: nil) ⇒ ClientStream

Returns a new instance of ClientStream.



11
12
13
14
15
16
17
18
19
 
# File 'lib/grpc_kit/stream/client_stream.rb', line 11

def initialize(transport, config, authority:, timeout: nil)
  @transport = transport
  @config = config

  @authority = authority
  @timeout = timeout

  @started = false
end

Instance Method Details

#close_and_recv ⇒ Object 



53
54
55
56
57
58
59
60
61
 
# File 'lib/grpc_kit/stream/client_stream.rb', line 53

def close_and_recv
  validate_if_request_start!

  @transport.close_and_flush

  data = []
  loop { data.push(do_recv) }
  data
end

#each ⇒ Object 



41
42
43
44
45
 
# File 'lib/grpc_kit/stream/client_stream.rb', line 41

def each
  validate_if_request_start!

  loop { yield(do_recv) }
end

#recv_msg(last: false) ⇒ Object 



47
48
49
50
51
 
# File 'lib/grpc_kit/stream/client_stream.rb', line 47

def recv_msg(last: false)
  validate_if_request_start!

  do_recv(last: last)
end

#send_msg(data, metadata: {}, timeout: nil, last: false) ⇒ Object 



21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
 
# File 'lib/grpc_kit/stream/client_stream.rb', line 21

def send_msg(data, metadata: {}, timeout: nil, last: false)
  buf =
    begin
      @config.protobuf.encode(data)
    rescue ArgumentError => e
      raise GrpcKit::Errors::Internal, "Error while encoding in client: #{e}"
    end

  limit_size = @config.max_send_message_size
  if limit_size && buf.bytesize > limit_size
    raise GrpcKit::Errors::ResourceExhausted, "Sending message is too large: send=#{req.bytesize}, max=#{limit_size}"
  end

  if @started
    @transport.write_data(buf, last: last)
  else
    start_request(buf, metadata: metadata, last: last)
  end
end