1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
# frozen_string_literal: true
module HTTPX
module Plugins
module GRPC
# Encoding module for GRPC responses
#
# Can encode and decode grpc messages.
module Message
module_function
# decodes a unary grpc response
def unary(response)
verify_status(response)
decoder = Transcoder::GRPCEncoding.decode(response)
decoder.call(response.to_s)
end
# lazy decodes a grpc stream response
def stream(response, &block)
return enum_for(__method__, response) unless block
decoder = Transcoder::GRPCEncoding.decode(response)
response.each do |frame|
decoder.call(frame, &block)
end
verify_status(response)
end
def cancel(request)
request.emit(:refuse, :client_cancellation)
end
# interprets the grpc call trailing metadata, and raises an
# exception in case of error code
def verify_status(response)
# return standard errors if need be
response.raise_for_status
status = Integer(response.headers["grpc-status"])
message = response.headers["grpc-message"]
return if status.zero?
response.close
raise GRPCError.new(status, message, response.trailing_metadata)
end
end
end
end
end
|
