grpc 1.30.2-x86-linux
Sign up to get free protection for your applications and to get access to all the features.
Potentially problematic release.
This version of grpc might be problematic. Click here for more details.
- checksums.yaml +7 -0
- data/etc/roots.pem +4644 -0
- data/grpc_c.32.ruby +0 -0
- data/grpc_c.64.ruby +0 -0
- data/src/ruby/bin/math_client.rb +140 -0
- data/src/ruby/bin/math_pb.rb +34 -0
- data/src/ruby/bin/math_server.rb +191 -0
- data/src/ruby/bin/math_services_pb.rb +51 -0
- data/src/ruby/bin/noproto_client.rb +93 -0
- data/src/ruby/bin/noproto_server.rb +97 -0
- data/src/ruby/ext/grpc/ext-export.clang +1 -0
- data/src/ruby/ext/grpc/ext-export.gcc +6 -0
- data/src/ruby/ext/grpc/extconf.rb +107 -0
- data/src/ruby/ext/grpc/rb_byte_buffer.c +64 -0
- data/src/ruby/ext/grpc/rb_byte_buffer.h +35 -0
- data/src/ruby/ext/grpc/rb_call.c +1050 -0
- data/src/ruby/ext/grpc/rb_call.h +53 -0
- data/src/ruby/ext/grpc/rb_call_credentials.c +297 -0
- data/src/ruby/ext/grpc/rb_call_credentials.h +31 -0
- data/src/ruby/ext/grpc/rb_channel.c +835 -0
- data/src/ruby/ext/grpc/rb_channel.h +34 -0
- data/src/ruby/ext/grpc/rb_channel_args.c +155 -0
- data/src/ruby/ext/grpc/rb_channel_args.h +38 -0
- data/src/ruby/ext/grpc/rb_channel_credentials.c +267 -0
- data/src/ruby/ext/grpc/rb_channel_credentials.h +32 -0
- data/src/ruby/ext/grpc/rb_completion_queue.c +100 -0
- data/src/ruby/ext/grpc/rb_completion_queue.h +36 -0
- data/src/ruby/ext/grpc/rb_compression_options.c +470 -0
- data/src/ruby/ext/grpc/rb_compression_options.h +29 -0
- data/src/ruby/ext/grpc/rb_enable_cpp.cc +22 -0
- data/src/ruby/ext/grpc/rb_event_thread.c +143 -0
- data/src/ruby/ext/grpc/rb_event_thread.h +21 -0
- data/src/ruby/ext/grpc/rb_grpc.c +328 -0
- data/src/ruby/ext/grpc/rb_grpc.h +76 -0
- data/src/ruby/ext/grpc/rb_grpc_imports.generated.c +573 -0
- data/src/ruby/ext/grpc/rb_grpc_imports.generated.h +865 -0
- data/src/ruby/ext/grpc/rb_loader.c +57 -0
- data/src/ruby/ext/grpc/rb_loader.h +25 -0
- data/src/ruby/ext/grpc/rb_server.c +372 -0
- data/src/ruby/ext/grpc/rb_server.h +32 -0
- data/src/ruby/ext/grpc/rb_server_credentials.c +243 -0
- data/src/ruby/ext/grpc/rb_server_credentials.h +32 -0
- data/src/ruby/lib/grpc.rb +37 -0
- data/src/ruby/lib/grpc/2.3/grpc_c.so +0 -0
- data/src/ruby/lib/grpc/2.4/grpc_c.so +0 -0
- data/src/ruby/lib/grpc/2.5/grpc_c.so +0 -0
- data/src/ruby/lib/grpc/2.6/grpc_c.so +0 -0
- data/src/ruby/lib/grpc/2.7/grpc_c.so +0 -0
- data/src/ruby/lib/grpc/core/status_codes.rb +135 -0
- data/src/ruby/lib/grpc/core/time_consts.rb +56 -0
- data/src/ruby/lib/grpc/errors.rb +277 -0
- data/src/ruby/lib/grpc/generic/active_call.rb +669 -0
- data/src/ruby/lib/grpc/generic/bidi_call.rb +233 -0
- data/src/ruby/lib/grpc/generic/client_stub.rb +501 -0
- data/src/ruby/lib/grpc/generic/interceptor_registry.rb +53 -0
- data/src/ruby/lib/grpc/generic/interceptors.rb +186 -0
- data/src/ruby/lib/grpc/generic/rpc_desc.rb +204 -0
- data/src/ruby/lib/grpc/generic/rpc_server.rb +551 -0
- data/src/ruby/lib/grpc/generic/service.rb +211 -0
- data/src/ruby/lib/grpc/google_rpc_status_utils.rb +40 -0
- data/src/ruby/lib/grpc/grpc.rb +24 -0
- data/src/ruby/lib/grpc/logconfig.rb +44 -0
- data/src/ruby/lib/grpc/notifier.rb +45 -0
- data/src/ruby/lib/grpc/structs.rb +15 -0
- data/src/ruby/lib/grpc/version.rb +18 -0
- data/src/ruby/pb/README.md +42 -0
- data/src/ruby/pb/generate_proto_ruby.sh +51 -0
- data/src/ruby/pb/grpc/health/checker.rb +75 -0
- data/src/ruby/pb/grpc/health/v1/health_pb.rb +31 -0
- data/src/ruby/pb/grpc/health/v1/health_services_pb.rb +62 -0
- data/src/ruby/pb/grpc/testing/duplicate/echo_duplicate_services_pb.rb +44 -0
- data/src/ruby/pb/grpc/testing/metrics_pb.rb +28 -0
- data/src/ruby/pb/grpc/testing/metrics_services_pb.rb +49 -0
- data/src/ruby/pb/src/proto/grpc/testing/empty_pb.rb +17 -0
- data/src/ruby/pb/src/proto/grpc/testing/messages_pb.rb +105 -0
- data/src/ruby/pb/src/proto/grpc/testing/test_pb.rb +16 -0
- data/src/ruby/pb/src/proto/grpc/testing/test_services_pb.rb +118 -0
- data/src/ruby/pb/test/client.rb +769 -0
- data/src/ruby/pb/test/server.rb +252 -0
- data/src/ruby/pb/test/xds_client.rb +213 -0
- data/src/ruby/spec/call_credentials_spec.rb +42 -0
- data/src/ruby/spec/call_spec.rb +180 -0
- data/src/ruby/spec/channel_connection_spec.rb +126 -0
- data/src/ruby/spec/channel_credentials_spec.rb +82 -0
- data/src/ruby/spec/channel_spec.rb +234 -0
- data/src/ruby/spec/client_auth_spec.rb +126 -0
- data/src/ruby/spec/client_server_spec.rb +664 -0
- data/src/ruby/spec/compression_options_spec.rb +149 -0
- data/src/ruby/spec/debug_message_spec.rb +134 -0
- data/src/ruby/spec/error_sanity_spec.rb +49 -0
- data/src/ruby/spec/errors_spec.rb +142 -0
- data/src/ruby/spec/generic/active_call_spec.rb +672 -0
- data/src/ruby/spec/generic/client_interceptors_spec.rb +153 -0
- data/src/ruby/spec/generic/client_stub_spec.rb +1083 -0
- data/src/ruby/spec/generic/interceptor_registry_spec.rb +65 -0
- data/src/ruby/spec/generic/rpc_desc_spec.rb +374 -0
- data/src/ruby/spec/generic/rpc_server_pool_spec.rb +127 -0
- data/src/ruby/spec/generic/rpc_server_spec.rb +748 -0
- data/src/ruby/spec/generic/server_interceptors_spec.rb +218 -0
- data/src/ruby/spec/generic/service_spec.rb +263 -0
- data/src/ruby/spec/google_rpc_status_utils_spec.rb +282 -0
- data/src/ruby/spec/pb/codegen/grpc/testing/package_options.proto +28 -0
- data/src/ruby/spec/pb/codegen/grpc/testing/package_options_import.proto +22 -0
- data/src/ruby/spec/pb/codegen/grpc/testing/package_options_import2.proto +23 -0
- data/src/ruby/spec/pb/codegen/grpc/testing/package_options_ruby_style.proto +41 -0
- data/src/ruby/spec/pb/codegen/package_option_spec.rb +82 -0
- data/src/ruby/spec/pb/duplicate/codegen_spec.rb +57 -0
- data/src/ruby/spec/pb/health/checker_spec.rb +236 -0
- data/src/ruby/spec/server_credentials_spec.rb +79 -0
- data/src/ruby/spec/server_spec.rb +209 -0
- data/src/ruby/spec/spec_helper.rb +61 -0
- data/src/ruby/spec/support/helpers.rb +107 -0
- data/src/ruby/spec/support/services.rb +160 -0
- data/src/ruby/spec/testdata/README +1 -0
- data/src/ruby/spec/testdata/ca.pem +20 -0
- data/src/ruby/spec/testdata/client.key +28 -0
- data/src/ruby/spec/testdata/client.pem +20 -0
- data/src/ruby/spec/testdata/server1.key +28 -0
- data/src/ruby/spec/testdata/server1.pem +22 -0
- data/src/ruby/spec/time_consts_spec.rb +74 -0
- metadata +394 -0
@@ -0,0 +1,56 @@
|
|
1
|
+
# Copyright 2015 gRPC authors.
|
2
|
+
#
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
4
|
+
# you may not use this file except in compliance with the License.
|
5
|
+
# You may obtain a copy of the License at
|
6
|
+
#
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
8
|
+
#
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12
|
+
# See the License for the specific language governing permissions and
|
13
|
+
# limitations under the License.
|
14
|
+
|
15
|
+
require_relative '../grpc'
|
16
|
+
|
17
|
+
# GRPC contains the General RPC module.
|
18
|
+
module GRPC
|
19
|
+
module Core
|
20
|
+
# TimeConsts is a module from the C extension.
|
21
|
+
#
|
22
|
+
# Here it's re-opened to add a utility func.
|
23
|
+
module TimeConsts
|
24
|
+
# Converts a time delta to an absolute deadline.
|
25
|
+
#
|
26
|
+
# Assumes timeish is a relative time, and converts its to an absolute,
|
27
|
+
# with following exceptions:
|
28
|
+
#
|
29
|
+
# * if timish is one of the TimeConsts.TimeSpec constants the value is
|
30
|
+
# preserved.
|
31
|
+
# * timish < 0 => TimeConsts.INFINITE_FUTURE
|
32
|
+
# * timish == 0 => TimeConsts.ZERO
|
33
|
+
#
|
34
|
+
# @param timeish [Number|TimeSpec]
|
35
|
+
# @return [Number|TimeSpec]
|
36
|
+
def from_relative_time(timeish)
|
37
|
+
if timeish.is_a? TimeSpec
|
38
|
+
timeish
|
39
|
+
elsif timeish.nil?
|
40
|
+
TimeConsts::ZERO
|
41
|
+
elsif !timeish.is_a? Numeric
|
42
|
+
fail(TypeError,
|
43
|
+
"Cannot make an absolute deadline from #{timeish.inspect}")
|
44
|
+
elsif timeish < 0
|
45
|
+
TimeConsts::INFINITE_FUTURE
|
46
|
+
elsif timeish.zero?
|
47
|
+
TimeConsts::ZERO
|
48
|
+
else
|
49
|
+
Time.now + timeish
|
50
|
+
end
|
51
|
+
end
|
52
|
+
|
53
|
+
module_function :from_relative_time
|
54
|
+
end
|
55
|
+
end
|
56
|
+
end
|
@@ -0,0 +1,277 @@
|
|
1
|
+
# Copyright 2015 gRPC authors.
|
2
|
+
#
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
4
|
+
# you may not use this file except in compliance with the License.
|
5
|
+
# You may obtain a copy of the License at
|
6
|
+
#
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
8
|
+
#
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12
|
+
# See the License for the specific language governing permissions and
|
13
|
+
# limitations under the License.
|
14
|
+
|
15
|
+
require_relative './structs'
|
16
|
+
require_relative './core/status_codes'
|
17
|
+
require_relative './google_rpc_status_utils'
|
18
|
+
|
19
|
+
# GRPC contains the General RPC module.
|
20
|
+
module GRPC
|
21
|
+
# BadStatus is an exception class that indicates that an error occurred at
|
22
|
+
# either end of a GRPC connection. When raised, it indicates that a status
|
23
|
+
# error should be returned to the other end of a GRPC connection; when
|
24
|
+
# caught it means that this end received a status error.
|
25
|
+
#
|
26
|
+
# There is also subclass of BadStatus in this module for each GRPC status.
|
27
|
+
# E.g., the GRPC::Cancelled class corresponds to status CANCELLED.
|
28
|
+
#
|
29
|
+
# See
|
30
|
+
# https://github.com/grpc/grpc/blob/master/include/grpc/impl/codegen/status.h
|
31
|
+
# for detailed descriptions of each status code.
|
32
|
+
class BadStatus < StandardError
|
33
|
+
attr_reader :code, :details, :metadata, :debug_error_string
|
34
|
+
|
35
|
+
include GRPC::Core::StatusCodes
|
36
|
+
|
37
|
+
# @param code [Numeric] the status code
|
38
|
+
# @param details [String] the details of the exception
|
39
|
+
# @param metadata [Hash] the error's metadata
|
40
|
+
def initialize(code,
|
41
|
+
details = 'unknown cause',
|
42
|
+
metadata = {},
|
43
|
+
debug_error_string = nil)
|
44
|
+
exception_message = "#{code}:#{details}"
|
45
|
+
if debug_error_string
|
46
|
+
exception_message += ". debug_error_string:#{debug_error_string}"
|
47
|
+
end
|
48
|
+
super(exception_message)
|
49
|
+
@code = code
|
50
|
+
@details = details
|
51
|
+
@metadata = metadata
|
52
|
+
@debug_error_string = debug_error_string
|
53
|
+
end
|
54
|
+
|
55
|
+
# Converts the exception to a {Struct::Status} for use in the networking
|
56
|
+
# wrapper layer.
|
57
|
+
#
|
58
|
+
# @return [Struct::Status] with the same code and details
|
59
|
+
def to_status
|
60
|
+
Struct::Status.new(code, details, metadata, debug_error_string)
|
61
|
+
end
|
62
|
+
|
63
|
+
# Converts the exception to a deserialized {Google::Rpc::Status} object.
|
64
|
+
# Returns `nil` if the `grpc-status-details-bin` trailer could not be
|
65
|
+
# converted to a {Google::Rpc::Status} due to the server not providing
|
66
|
+
# the necessary trailers.
|
67
|
+
#
|
68
|
+
# @return [Google::Rpc::Status, nil]
|
69
|
+
def to_rpc_status
|
70
|
+
GoogleRpcStatusUtils.extract_google_rpc_status(to_status)
|
71
|
+
rescue Google::Protobuf::ParseError => parse_error
|
72
|
+
GRPC.logger.warn('parse error: to_rpc_status failed')
|
73
|
+
GRPC.logger.warn(parse_error)
|
74
|
+
nil
|
75
|
+
end
|
76
|
+
|
77
|
+
def self.new_status_exception(code,
|
78
|
+
details = 'unknown cause',
|
79
|
+
metadata = {},
|
80
|
+
debug_error_string = nil)
|
81
|
+
codes = {}
|
82
|
+
codes[OK] = Ok
|
83
|
+
codes[CANCELLED] = Cancelled
|
84
|
+
codes[UNKNOWN] = Unknown
|
85
|
+
codes[INVALID_ARGUMENT] = InvalidArgument
|
86
|
+
codes[DEADLINE_EXCEEDED] = DeadlineExceeded
|
87
|
+
codes[NOT_FOUND] = NotFound
|
88
|
+
codes[ALREADY_EXISTS] = AlreadyExists
|
89
|
+
codes[PERMISSION_DENIED] = PermissionDenied
|
90
|
+
codes[UNAUTHENTICATED] = Unauthenticated
|
91
|
+
codes[RESOURCE_EXHAUSTED] = ResourceExhausted
|
92
|
+
codes[FAILED_PRECONDITION] = FailedPrecondition
|
93
|
+
codes[ABORTED] = Aborted
|
94
|
+
codes[OUT_OF_RANGE] = OutOfRange
|
95
|
+
codes[UNIMPLEMENTED] = Unimplemented
|
96
|
+
codes[INTERNAL] = Internal
|
97
|
+
codes[UNAVAILABLE] = Unavailable
|
98
|
+
codes[DATA_LOSS] = DataLoss
|
99
|
+
|
100
|
+
if codes[code].nil?
|
101
|
+
BadStatus.new(code, details, metadata, debug_error_string)
|
102
|
+
else
|
103
|
+
codes[code].new(details, metadata, debug_error_string)
|
104
|
+
end
|
105
|
+
end
|
106
|
+
end
|
107
|
+
|
108
|
+
# GRPC status code corresponding to status OK
|
109
|
+
class Ok < BadStatus
|
110
|
+
def initialize(details = 'unknown cause',
|
111
|
+
metadata = {},
|
112
|
+
debug_error_string = nil)
|
113
|
+
super(Core::StatusCodes::OK,
|
114
|
+
details, metadata, debug_error_string)
|
115
|
+
end
|
116
|
+
end
|
117
|
+
|
118
|
+
# GRPC status code corresponding to status CANCELLED
|
119
|
+
class Cancelled < BadStatus
|
120
|
+
def initialize(details = 'unknown cause',
|
121
|
+
metadata = {},
|
122
|
+
debug_error_string = nil)
|
123
|
+
super(Core::StatusCodes::CANCELLED,
|
124
|
+
details, metadata, debug_error_string)
|
125
|
+
end
|
126
|
+
end
|
127
|
+
|
128
|
+
# GRPC status code corresponding to status UNKNOWN
|
129
|
+
class Unknown < BadStatus
|
130
|
+
def initialize(details = 'unknown cause',
|
131
|
+
metadata = {},
|
132
|
+
debug_error_string = nil)
|
133
|
+
super(Core::StatusCodes::UNKNOWN,
|
134
|
+
details, metadata, debug_error_string)
|
135
|
+
end
|
136
|
+
end
|
137
|
+
|
138
|
+
# GRPC status code corresponding to status INVALID_ARGUMENT
|
139
|
+
class InvalidArgument < BadStatus
|
140
|
+
def initialize(details = 'unknown cause',
|
141
|
+
metadata = {},
|
142
|
+
debug_error_string = nil)
|
143
|
+
super(Core::StatusCodes::INVALID_ARGUMENT,
|
144
|
+
details, metadata, debug_error_string)
|
145
|
+
end
|
146
|
+
end
|
147
|
+
|
148
|
+
# GRPC status code corresponding to status DEADLINE_EXCEEDED
|
149
|
+
class DeadlineExceeded < BadStatus
|
150
|
+
def initialize(details = 'unknown cause',
|
151
|
+
metadata = {},
|
152
|
+
debug_error_string = nil)
|
153
|
+
super(Core::StatusCodes::DEADLINE_EXCEEDED,
|
154
|
+
details, metadata, debug_error_string)
|
155
|
+
end
|
156
|
+
end
|
157
|
+
|
158
|
+
# GRPC status code corresponding to status NOT_FOUND
|
159
|
+
class NotFound < BadStatus
|
160
|
+
def initialize(details = 'unknown cause',
|
161
|
+
metadata = {},
|
162
|
+
debug_error_string = nil)
|
163
|
+
super(Core::StatusCodes::NOT_FOUND,
|
164
|
+
details, metadata, debug_error_string)
|
165
|
+
end
|
166
|
+
end
|
167
|
+
|
168
|
+
# GRPC status code corresponding to status ALREADY_EXISTS
|
169
|
+
class AlreadyExists < BadStatus
|
170
|
+
def initialize(details = 'unknown cause',
|
171
|
+
metadata = {},
|
172
|
+
debug_error_string = nil)
|
173
|
+
super(Core::StatusCodes::ALREADY_EXISTS,
|
174
|
+
details, metadata, debug_error_string)
|
175
|
+
end
|
176
|
+
end
|
177
|
+
|
178
|
+
# GRPC status code corresponding to status PERMISSION_DENIED
|
179
|
+
class PermissionDenied < BadStatus
|
180
|
+
def initialize(details = 'unknown cause',
|
181
|
+
metadata = {},
|
182
|
+
debug_error_string = nil)
|
183
|
+
super(Core::StatusCodes::PERMISSION_DENIED,
|
184
|
+
details, metadata, debug_error_string)
|
185
|
+
end
|
186
|
+
end
|
187
|
+
|
188
|
+
# GRPC status code corresponding to status UNAUTHENTICATED
|
189
|
+
class Unauthenticated < BadStatus
|
190
|
+
def initialize(details = 'unknown cause',
|
191
|
+
metadata = {},
|
192
|
+
debug_error_string = nil)
|
193
|
+
super(Core::StatusCodes::UNAUTHENTICATED,
|
194
|
+
details, metadata, debug_error_string)
|
195
|
+
end
|
196
|
+
end
|
197
|
+
|
198
|
+
# GRPC status code corresponding to status RESOURCE_EXHAUSTED
|
199
|
+
class ResourceExhausted < BadStatus
|
200
|
+
def initialize(details = 'unknown cause',
|
201
|
+
metadata = {},
|
202
|
+
debug_error_string = nil)
|
203
|
+
super(Core::StatusCodes::RESOURCE_EXHAUSTED,
|
204
|
+
details, metadata, debug_error_string)
|
205
|
+
end
|
206
|
+
end
|
207
|
+
|
208
|
+
# GRPC status code corresponding to status FAILED_PRECONDITION
|
209
|
+
class FailedPrecondition < BadStatus
|
210
|
+
def initialize(details = 'unknown cause',
|
211
|
+
metadata = {},
|
212
|
+
debug_error_string = nil)
|
213
|
+
super(Core::StatusCodes::FAILED_PRECONDITION,
|
214
|
+
details, metadata, debug_error_string)
|
215
|
+
end
|
216
|
+
end
|
217
|
+
|
218
|
+
# GRPC status code corresponding to status ABORTED
|
219
|
+
class Aborted < BadStatus
|
220
|
+
def initialize(details = 'unknown cause',
|
221
|
+
metadata = {},
|
222
|
+
debug_error_string = nil)
|
223
|
+
super(Core::StatusCodes::ABORTED,
|
224
|
+
details, metadata, debug_error_string)
|
225
|
+
end
|
226
|
+
end
|
227
|
+
|
228
|
+
# GRPC status code corresponding to status OUT_OF_RANGE
|
229
|
+
class OutOfRange < BadStatus
|
230
|
+
def initialize(details = 'unknown cause',
|
231
|
+
metadata = {},
|
232
|
+
debug_error_string = nil)
|
233
|
+
super(Core::StatusCodes::OUT_OF_RANGE,
|
234
|
+
details, metadata, debug_error_string)
|
235
|
+
end
|
236
|
+
end
|
237
|
+
|
238
|
+
# GRPC status code corresponding to status UNIMPLEMENTED
|
239
|
+
class Unimplemented < BadStatus
|
240
|
+
def initialize(details = 'unknown cause',
|
241
|
+
metadata = {},
|
242
|
+
debug_error_string = nil)
|
243
|
+
super(Core::StatusCodes::UNIMPLEMENTED,
|
244
|
+
details, metadata, debug_error_string)
|
245
|
+
end
|
246
|
+
end
|
247
|
+
|
248
|
+
# GRPC status code corresponding to status INTERNAL
|
249
|
+
class Internal < BadStatus
|
250
|
+
def initialize(details = 'unknown cause',
|
251
|
+
metadata = {},
|
252
|
+
debug_error_string = nil)
|
253
|
+
super(Core::StatusCodes::INTERNAL,
|
254
|
+
details, metadata, debug_error_string)
|
255
|
+
end
|
256
|
+
end
|
257
|
+
|
258
|
+
# GRPC status code corresponding to status UNAVAILABLE
|
259
|
+
class Unavailable < BadStatus
|
260
|
+
def initialize(details = 'unknown cause',
|
261
|
+
metadata = {},
|
262
|
+
debug_error_string = nil)
|
263
|
+
super(Core::StatusCodes::UNAVAILABLE,
|
264
|
+
details, metadata, debug_error_string)
|
265
|
+
end
|
266
|
+
end
|
267
|
+
|
268
|
+
# GRPC status code corresponding to status DATA_LOSS
|
269
|
+
class DataLoss < BadStatus
|
270
|
+
def initialize(details = 'unknown cause',
|
271
|
+
metadata = {},
|
272
|
+
debug_error_string = nil)
|
273
|
+
super(Core::StatusCodes::DATA_LOSS,
|
274
|
+
details, metadata, debug_error_string)
|
275
|
+
end
|
276
|
+
end
|
277
|
+
end
|
@@ -0,0 +1,669 @@
|
|
1
|
+
# Copyright 2015 gRPC authors.
|
2
|
+
#
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
4
|
+
# you may not use this file except in compliance with the License.
|
5
|
+
# You may obtain a copy of the License at
|
6
|
+
#
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
8
|
+
#
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12
|
+
# See the License for the specific language governing permissions and
|
13
|
+
# limitations under the License.
|
14
|
+
|
15
|
+
require 'forwardable'
|
16
|
+
require 'weakref'
|
17
|
+
require_relative 'bidi_call'
|
18
|
+
|
19
|
+
class Struct
|
20
|
+
# BatchResult is the struct returned by calls to call#start_batch.
|
21
|
+
class BatchResult
|
22
|
+
# check_status returns the status, raising an error if the status
|
23
|
+
# is non-nil and not OK.
|
24
|
+
def check_status
|
25
|
+
return nil if status.nil?
|
26
|
+
if status.code != GRPC::Core::StatusCodes::OK
|
27
|
+
GRPC.logger.debug("Failing with status #{status}")
|
28
|
+
# raise BadStatus, propagating the metadata if present.
|
29
|
+
fail GRPC::BadStatus.new_status_exception(
|
30
|
+
status.code, status.details, status.metadata,
|
31
|
+
status.debug_error_string)
|
32
|
+
end
|
33
|
+
status
|
34
|
+
end
|
35
|
+
end
|
36
|
+
end
|
37
|
+
|
38
|
+
# GRPC contains the General RPC module.
|
39
|
+
module GRPC
|
40
|
+
# The ActiveCall class provides simple methods for sending marshallable
|
41
|
+
# data to a call
|
42
|
+
class ActiveCall # rubocop:disable Metrics/ClassLength
|
43
|
+
include Core::TimeConsts
|
44
|
+
include Core::CallOps
|
45
|
+
extend Forwardable
|
46
|
+
attr_reader :deadline, :metadata_sent, :metadata_to_send, :peer, :peer_cert
|
47
|
+
def_delegators :@call, :cancel, :metadata, :write_flag, :write_flag=,
|
48
|
+
:trailing_metadata, :status
|
49
|
+
|
50
|
+
# client_invoke begins a client invocation.
|
51
|
+
#
|
52
|
+
# Flow Control note: this blocks until flow control accepts that client
|
53
|
+
# request can go ahead.
|
54
|
+
#
|
55
|
+
# deadline is the absolute deadline for the call.
|
56
|
+
#
|
57
|
+
# == Keyword Arguments ==
|
58
|
+
# any keyword arguments are treated as metadata to be sent to the server
|
59
|
+
# if a keyword value is a list, multiple metadata for it's key are sent
|
60
|
+
#
|
61
|
+
# @param call [Call] a call on which to start and invocation
|
62
|
+
# @param metadata [Hash] the metadata
|
63
|
+
def self.client_invoke(call, metadata = {})
|
64
|
+
fail(TypeError, '!Core::Call') unless call.is_a? Core::Call
|
65
|
+
call.run_batch(SEND_INITIAL_METADATA => metadata)
|
66
|
+
end
|
67
|
+
|
68
|
+
# Creates an ActiveCall.
|
69
|
+
#
|
70
|
+
# ActiveCall should only be created after a call is accepted. That
|
71
|
+
# means different things on a client and a server. On the client, the
|
72
|
+
# call is accepted after calling call.invoke. On the server, this is
|
73
|
+
# after call.accept.
|
74
|
+
#
|
75
|
+
# #initialize cannot determine if the call is accepted or not; so if a
|
76
|
+
# call that's not accepted is used here, the error won't be visible until
|
77
|
+
# the ActiveCall methods are called.
|
78
|
+
#
|
79
|
+
# deadline is the absolute deadline for the call.
|
80
|
+
#
|
81
|
+
# @param call [Call] the call used by the ActiveCall
|
82
|
+
# @param marshal [Function] f(obj)->string that marshal requests
|
83
|
+
# @param unmarshal [Function] f(string)->obj that unmarshals responses
|
84
|
+
# @param deadline [Fixnum] the deadline for the call to complete
|
85
|
+
# @param started [true|false] indicates that metadata was sent
|
86
|
+
# @param metadata_received [true|false] indicates if metadata has already
|
87
|
+
# been received. Should always be true for server calls
|
88
|
+
def initialize(call, marshal, unmarshal, deadline, started: true,
|
89
|
+
metadata_received: false, metadata_to_send: nil)
|
90
|
+
fail(TypeError, '!Core::Call') unless call.is_a? Core::Call
|
91
|
+
@call = call
|
92
|
+
@deadline = deadline
|
93
|
+
@marshal = marshal
|
94
|
+
@unmarshal = unmarshal
|
95
|
+
@metadata_received = metadata_received
|
96
|
+
@metadata_sent = started
|
97
|
+
@op_notifier = nil
|
98
|
+
|
99
|
+
fail(ArgumentError, 'Already sent md') if started && metadata_to_send
|
100
|
+
@metadata_to_send = metadata_to_send || {} unless started
|
101
|
+
@send_initial_md_mutex = Mutex.new
|
102
|
+
|
103
|
+
@output_stream_done = false
|
104
|
+
@input_stream_done = false
|
105
|
+
@call_finished = false
|
106
|
+
@call_finished_mu = Mutex.new
|
107
|
+
|
108
|
+
@client_call_executed = false
|
109
|
+
@client_call_executed_mu = Mutex.new
|
110
|
+
|
111
|
+
# set the peer now so that the accessor can still function
|
112
|
+
# after the server closes the call
|
113
|
+
@peer = call.peer
|
114
|
+
end
|
115
|
+
|
116
|
+
# Sends the initial metadata that has yet to be sent.
|
117
|
+
# Does nothing if metadata has already been sent for this call.
|
118
|
+
def send_initial_metadata(new_metadata = {})
|
119
|
+
@send_initial_md_mutex.synchronize do
|
120
|
+
return if @metadata_sent
|
121
|
+
@metadata_to_send.merge!(new_metadata)
|
122
|
+
ActiveCall.client_invoke(@call, @metadata_to_send)
|
123
|
+
@metadata_sent = true
|
124
|
+
end
|
125
|
+
end
|
126
|
+
|
127
|
+
# output_metadata are provides access to hash that can be used to
|
128
|
+
# save metadata to be sent as trailer
|
129
|
+
def output_metadata
|
130
|
+
@output_metadata ||= {}
|
131
|
+
end
|
132
|
+
|
133
|
+
# cancelled indicates if the call was cancelled
|
134
|
+
def cancelled?
|
135
|
+
!@call.status.nil? && @call.status.code == Core::StatusCodes::CANCELLED
|
136
|
+
end
|
137
|
+
|
138
|
+
# multi_req_view provides a restricted view of this ActiveCall for use
|
139
|
+
# in a server client-streaming handler.
|
140
|
+
def multi_req_view
|
141
|
+
MultiReqView.new(self)
|
142
|
+
end
|
143
|
+
|
144
|
+
# single_req_view provides a restricted view of this ActiveCall for use in
|
145
|
+
# a server request-response handler.
|
146
|
+
def single_req_view
|
147
|
+
SingleReqView.new(self)
|
148
|
+
end
|
149
|
+
|
150
|
+
# operation provides a restricted view of this ActiveCall for use as
|
151
|
+
# a Operation.
|
152
|
+
def operation
|
153
|
+
@op_notifier = Notifier.new
|
154
|
+
Operation.new(self)
|
155
|
+
end
|
156
|
+
|
157
|
+
##
|
158
|
+
# Returns a restricted view of this ActiveCall for use in interceptors
|
159
|
+
#
|
160
|
+
# @return [InterceptableView]
|
161
|
+
#
|
162
|
+
def interceptable
|
163
|
+
InterceptableView.new(self)
|
164
|
+
end
|
165
|
+
|
166
|
+
def receive_and_check_status
|
167
|
+
batch_result = @call.run_batch(RECV_STATUS_ON_CLIENT => nil)
|
168
|
+
set_input_stream_done
|
169
|
+
attach_status_results_and_complete_call(batch_result)
|
170
|
+
end
|
171
|
+
|
172
|
+
def attach_status_results_and_complete_call(recv_status_batch_result)
|
173
|
+
unless recv_status_batch_result.status.nil?
|
174
|
+
@call.trailing_metadata = recv_status_batch_result.status.metadata
|
175
|
+
end
|
176
|
+
@call.status = recv_status_batch_result.status
|
177
|
+
|
178
|
+
# The RECV_STATUS in run_batch always succeeds
|
179
|
+
# Check the status for a bad status or failed run batch
|
180
|
+
recv_status_batch_result.check_status
|
181
|
+
end
|
182
|
+
|
183
|
+
# remote_send sends a request to the remote endpoint.
|
184
|
+
#
|
185
|
+
# It blocks until the remote endpoint accepts the message.
|
186
|
+
#
|
187
|
+
# @param req [Object, String] the object to send or it's marshal form.
|
188
|
+
# @param marshalled [false, true] indicates if the object is already
|
189
|
+
# marshalled.
|
190
|
+
def remote_send(req, marshalled = false)
|
191
|
+
send_initial_metadata
|
192
|
+
GRPC.logger.debug("sending #{req}, marshalled? #{marshalled}")
|
193
|
+
payload = marshalled ? req : @marshal.call(req)
|
194
|
+
@call.run_batch(SEND_MESSAGE => payload)
|
195
|
+
end
|
196
|
+
|
197
|
+
# send_status sends a status to the remote endpoint.
|
198
|
+
#
|
199
|
+
# @param code [int] the status code to send
|
200
|
+
# @param details [String] details
|
201
|
+
# @param assert_finished [true, false] when true(default), waits for
|
202
|
+
# FINISHED.
|
203
|
+
# @param metadata [Hash] metadata to send to the server. If a value is a
|
204
|
+
# list, mulitple metadata for its key are sent
|
205
|
+
def send_status(code = OK, details = '', assert_finished = false,
|
206
|
+
metadata: {})
|
207
|
+
send_initial_metadata
|
208
|
+
ops = {
|
209
|
+
SEND_STATUS_FROM_SERVER => Struct::Status.new(code, details, metadata)
|
210
|
+
}
|
211
|
+
ops[RECV_CLOSE_ON_SERVER] = nil if assert_finished
|
212
|
+
@call.run_batch(ops)
|
213
|
+
set_output_stream_done
|
214
|
+
|
215
|
+
nil
|
216
|
+
end
|
217
|
+
|
218
|
+
# Intended for use on server-side calls when a single request from
|
219
|
+
# the client is expected (i.e., unary and server-streaming RPC types).
|
220
|
+
def read_unary_request
|
221
|
+
req = remote_read
|
222
|
+
set_input_stream_done
|
223
|
+
req
|
224
|
+
end
|
225
|
+
|
226
|
+
def server_unary_response(req, trailing_metadata: {},
|
227
|
+
code: Core::StatusCodes::OK, details: 'OK')
|
228
|
+
ops = {}
|
229
|
+
@send_initial_md_mutex.synchronize do
|
230
|
+
ops[SEND_INITIAL_METADATA] = @metadata_to_send unless @metadata_sent
|
231
|
+
@metadata_sent = true
|
232
|
+
end
|
233
|
+
|
234
|
+
payload = @marshal.call(req)
|
235
|
+
ops[SEND_MESSAGE] = payload
|
236
|
+
ops[SEND_STATUS_FROM_SERVER] = Struct::Status.new(
|
237
|
+
code, details, trailing_metadata)
|
238
|
+
ops[RECV_CLOSE_ON_SERVER] = nil
|
239
|
+
|
240
|
+
@call.run_batch(ops)
|
241
|
+
set_output_stream_done
|
242
|
+
end
|
243
|
+
|
244
|
+
# remote_read reads a response from the remote endpoint.
|
245
|
+
#
|
246
|
+
# It blocks until the remote endpoint replies with a message or status.
|
247
|
+
# On receiving a message, it returns the response after unmarshalling it.
|
248
|
+
# On receiving a status, it returns nil if the status is OK, otherwise
|
249
|
+
# raising BadStatus
|
250
|
+
def remote_read
|
251
|
+
ops = { RECV_MESSAGE => nil }
|
252
|
+
ops[RECV_INITIAL_METADATA] = nil unless @metadata_received
|
253
|
+
batch_result = @call.run_batch(ops)
|
254
|
+
unless @metadata_received
|
255
|
+
@call.metadata = batch_result.metadata
|
256
|
+
@metadata_received = true
|
257
|
+
end
|
258
|
+
get_message_from_batch_result(batch_result)
|
259
|
+
end
|
260
|
+
|
261
|
+
def get_message_from_batch_result(recv_message_batch_result)
|
262
|
+
unless recv_message_batch_result.nil? ||
|
263
|
+
recv_message_batch_result.message.nil?
|
264
|
+
return @unmarshal.call(recv_message_batch_result.message)
|
265
|
+
end
|
266
|
+
GRPC.logger.debug('found nil; the final response has been sent')
|
267
|
+
nil
|
268
|
+
end
|
269
|
+
|
270
|
+
# each_remote_read passes each response to the given block or returns an
|
271
|
+
# enumerator the responses if no block is given.
|
272
|
+
# Used to generate the request enumerable for
|
273
|
+
# server-side client-streaming RPC's.
|
274
|
+
#
|
275
|
+
# == Enumerator ==
|
276
|
+
#
|
277
|
+
# * #next blocks until the remote endpoint sends a READ or FINISHED
|
278
|
+
# * for each read, enumerator#next yields the response
|
279
|
+
# * on status
|
280
|
+
# * if it's is OK, enumerator#next raises StopException
|
281
|
+
# * if is not OK, enumerator#next raises RuntimeException
|
282
|
+
#
|
283
|
+
# == Block ==
|
284
|
+
#
|
285
|
+
# * if provided it is executed for each response
|
286
|
+
# * the call blocks until no more responses are provided
|
287
|
+
#
|
288
|
+
# @return [Enumerator] if no block was given
|
289
|
+
def each_remote_read
|
290
|
+
return enum_for(:each_remote_read) unless block_given?
|
291
|
+
begin
|
292
|
+
loop do
|
293
|
+
resp = remote_read
|
294
|
+
break if resp.nil? # the last response was received
|
295
|
+
yield resp
|
296
|
+
end
|
297
|
+
ensure
|
298
|
+
set_input_stream_done
|
299
|
+
end
|
300
|
+
end
|
301
|
+
|
302
|
+
# each_remote_read_then_finish passes each response to the given block or
|
303
|
+
# returns an enumerator of the responses if no block is given.
|
304
|
+
#
|
305
|
+
# It is like each_remote_read, but it blocks on finishing on detecting
|
306
|
+
# the final message.
|
307
|
+
#
|
308
|
+
# == Enumerator ==
|
309
|
+
#
|
310
|
+
# * #next blocks until the remote endpoint sends a READ or FINISHED
|
311
|
+
# * for each read, enumerator#next yields the response
|
312
|
+
# * on status
|
313
|
+
# * if it's is OK, enumerator#next raises StopException
|
314
|
+
# * if is not OK, enumerator#next raises RuntimeException
|
315
|
+
#
|
316
|
+
# == Block ==
|
317
|
+
#
|
318
|
+
# * if provided it is executed for each response
|
319
|
+
# * the call blocks until no more responses are provided
|
320
|
+
#
|
321
|
+
# @return [Enumerator] if no block was given
|
322
|
+
def each_remote_read_then_finish
|
323
|
+
return enum_for(:each_remote_read_then_finish) unless block_given?
|
324
|
+
loop do
|
325
|
+
resp =
|
326
|
+
begin
|
327
|
+
remote_read
|
328
|
+
rescue GRPC::Core::CallError => e
|
329
|
+
GRPC.logger.warn("In each_remote_read_then_finish: #{e}")
|
330
|
+
nil
|
331
|
+
end
|
332
|
+
|
333
|
+
break if resp.nil? # the last response was received
|
334
|
+
yield resp
|
335
|
+
end
|
336
|
+
|
337
|
+
receive_and_check_status
|
338
|
+
ensure
|
339
|
+
set_input_stream_done
|
340
|
+
end
|
341
|
+
|
342
|
+
# request_response sends a request to a GRPC server, and returns the
|
343
|
+
# response.
|
344
|
+
#
|
345
|
+
# @param req [Object] the request sent to the server
|
346
|
+
# @param metadata [Hash] metadata to be sent to the server. If a value is
|
347
|
+
# a list, multiple metadata for its key are sent
|
348
|
+
# @return [Object] the response received from the server
|
349
|
+
def request_response(req, metadata: {})
|
350
|
+
raise_error_if_already_executed
|
351
|
+
ops = {
|
352
|
+
SEND_MESSAGE => @marshal.call(req),
|
353
|
+
SEND_CLOSE_FROM_CLIENT => nil,
|
354
|
+
RECV_INITIAL_METADATA => nil,
|
355
|
+
RECV_MESSAGE => nil,
|
356
|
+
RECV_STATUS_ON_CLIENT => nil
|
357
|
+
}
|
358
|
+
@send_initial_md_mutex.synchronize do
|
359
|
+
# Metadata might have already been sent if this is an operation view
|
360
|
+
unless @metadata_sent
|
361
|
+
ops[SEND_INITIAL_METADATA] = @metadata_to_send.merge!(metadata)
|
362
|
+
end
|
363
|
+
@metadata_sent = true
|
364
|
+
end
|
365
|
+
|
366
|
+
begin
|
367
|
+
batch_result = @call.run_batch(ops)
|
368
|
+
# no need to check for cancellation after a CallError because this
|
369
|
+
# batch contains a RECV_STATUS op
|
370
|
+
ensure
|
371
|
+
set_input_stream_done
|
372
|
+
set_output_stream_done
|
373
|
+
end
|
374
|
+
|
375
|
+
@call.metadata = batch_result.metadata
|
376
|
+
attach_status_results_and_complete_call(batch_result)
|
377
|
+
get_message_from_batch_result(batch_result)
|
378
|
+
end
|
379
|
+
|
380
|
+
# client_streamer sends a stream of requests to a GRPC server, and
|
381
|
+
# returns a single response.
|
382
|
+
#
|
383
|
+
# requests provides an 'iterable' of Requests. I.e. it follows Ruby's
|
384
|
+
# #each enumeration protocol. In the simplest case, requests will be an
|
385
|
+
# array of marshallable objects; in typical case it will be an Enumerable
|
386
|
+
# that allows dynamic construction of the marshallable objects.
|
387
|
+
#
|
388
|
+
# @param requests [Object] an Enumerable of requests to send
|
389
|
+
# @param metadata [Hash] metadata to be sent to the server. If a value is
|
390
|
+
# a list, multiple metadata for its key are sent
|
391
|
+
# @return [Object] the response received from the server
|
392
|
+
def client_streamer(requests, metadata: {})
|
393
|
+
raise_error_if_already_executed
|
394
|
+
begin
|
395
|
+
send_initial_metadata(metadata)
|
396
|
+
requests.each { |r| @call.run_batch(SEND_MESSAGE => @marshal.call(r)) }
|
397
|
+
rescue GRPC::Core::CallError => e
|
398
|
+
receive_and_check_status # check for Cancelled
|
399
|
+
raise e
|
400
|
+
rescue => e
|
401
|
+
set_input_stream_done
|
402
|
+
raise e
|
403
|
+
ensure
|
404
|
+
set_output_stream_done
|
405
|
+
end
|
406
|
+
|
407
|
+
batch_result = @call.run_batch(
|
408
|
+
SEND_CLOSE_FROM_CLIENT => nil,
|
409
|
+
RECV_INITIAL_METADATA => nil,
|
410
|
+
RECV_MESSAGE => nil,
|
411
|
+
RECV_STATUS_ON_CLIENT => nil
|
412
|
+
)
|
413
|
+
|
414
|
+
set_input_stream_done
|
415
|
+
|
416
|
+
@call.metadata = batch_result.metadata
|
417
|
+
attach_status_results_and_complete_call(batch_result)
|
418
|
+
get_message_from_batch_result(batch_result)
|
419
|
+
end
|
420
|
+
|
421
|
+
# server_streamer sends one request to the GRPC server, which yields a
|
422
|
+
# stream of responses.
|
423
|
+
#
|
424
|
+
# responses provides an enumerator over the streamed responses, i.e. it
|
425
|
+
# follows Ruby's #each iteration protocol. The enumerator blocks while
|
426
|
+
# waiting for each response, stops when the server signals that no
|
427
|
+
# further responses will be supplied. If the implicit block is provided,
|
428
|
+
# it is executed with each response as the argument and no result is
|
429
|
+
# returned.
|
430
|
+
#
|
431
|
+
# @param req [Object] the request sent to the server
|
432
|
+
# @param metadata [Hash] metadata to be sent to the server. If a value is
|
433
|
+
# a list, multiple metadata for its key are sent
|
434
|
+
# @return [Enumerator|nil] a response Enumerator
|
435
|
+
def server_streamer(req, metadata: {})
|
436
|
+
raise_error_if_already_executed
|
437
|
+
ops = {
|
438
|
+
SEND_MESSAGE => @marshal.call(req),
|
439
|
+
SEND_CLOSE_FROM_CLIENT => nil
|
440
|
+
}
|
441
|
+
@send_initial_md_mutex.synchronize do
|
442
|
+
# Metadata might have already been sent if this is an operation view
|
443
|
+
unless @metadata_sent
|
444
|
+
ops[SEND_INITIAL_METADATA] = @metadata_to_send.merge!(metadata)
|
445
|
+
end
|
446
|
+
@metadata_sent = true
|
447
|
+
end
|
448
|
+
|
449
|
+
begin
|
450
|
+
@call.run_batch(ops)
|
451
|
+
rescue GRPC::Core::CallError => e
|
452
|
+
receive_and_check_status # checks for Cancelled
|
453
|
+
raise e
|
454
|
+
rescue => e
|
455
|
+
set_input_stream_done
|
456
|
+
raise e
|
457
|
+
ensure
|
458
|
+
set_output_stream_done
|
459
|
+
end
|
460
|
+
|
461
|
+
replies = enum_for(:each_remote_read_then_finish)
|
462
|
+
return replies unless block_given?
|
463
|
+
replies.each { |r| yield r }
|
464
|
+
end
|
465
|
+
|
466
|
+
# bidi_streamer sends a stream of requests to the GRPC server, and yields
|
467
|
+
# a stream of responses.
|
468
|
+
#
|
469
|
+
# This method takes an Enumerable of requests, and returns and enumerable
|
470
|
+
# of responses.
|
471
|
+
#
|
472
|
+
# == requests ==
|
473
|
+
#
|
474
|
+
# requests provides an 'iterable' of Requests. I.e. it follows Ruby's
|
475
|
+
# #each enumeration protocol. In the simplest case, requests will be an
|
476
|
+
# array of marshallable objects; in typical case it will be an
|
477
|
+
# Enumerable that allows dynamic construction of the marshallable
|
478
|
+
# objects.
|
479
|
+
#
|
480
|
+
# == responses ==
|
481
|
+
#
|
482
|
+
# This is an enumerator of responses. I.e, its #next method blocks
|
483
|
+
# waiting for the next response. Also, if at any point the block needs
|
484
|
+
# to consume all the remaining responses, this can be done using #each or
|
485
|
+
# #collect. Calling #each or #collect should only be done if
|
486
|
+
# the_call#writes_done has been called, otherwise the block will loop
|
487
|
+
# forever.
|
488
|
+
#
|
489
|
+
# @param requests [Object] an Enumerable of requests to send
|
490
|
+
# @param metadata [Hash] metadata to be sent to the server. If a value is
|
491
|
+
# a list, multiple metadata for its key are sent
|
492
|
+
# @return [Enumerator, nil] a response Enumerator
|
493
|
+
def bidi_streamer(requests, metadata: {}, &blk)
|
494
|
+
raise_error_if_already_executed
|
495
|
+
# Metadata might have already been sent if this is an operation view
|
496
|
+
begin
|
497
|
+
send_initial_metadata(metadata)
|
498
|
+
rescue GRPC::Core::CallError => e
|
499
|
+
batch_result = @call.run_batch(RECV_STATUS_ON_CLIENT => nil)
|
500
|
+
set_input_stream_done
|
501
|
+
set_output_stream_done
|
502
|
+
attach_status_results_and_complete_call(batch_result)
|
503
|
+
raise e
|
504
|
+
rescue => e
|
505
|
+
set_input_stream_done
|
506
|
+
set_output_stream_done
|
507
|
+
raise e
|
508
|
+
end
|
509
|
+
|
510
|
+
bd = BidiCall.new(@call,
|
511
|
+
@marshal,
|
512
|
+
@unmarshal,
|
513
|
+
metadata_received: @metadata_received)
|
514
|
+
|
515
|
+
bd.run_on_client(requests,
|
516
|
+
proc { set_input_stream_done },
|
517
|
+
proc { set_output_stream_done },
|
518
|
+
&blk)
|
519
|
+
end
|
520
|
+
|
521
|
+
# run_server_bidi orchestrates a BiDi stream processing on a server.
|
522
|
+
#
|
523
|
+
# N.B. gen_each_reply is a func(Enumerable<Requests>)
|
524
|
+
#
|
525
|
+
# It takes an enumerable of requests as an arg, in case there is a
|
526
|
+
# relationship between the stream of requests and the stream of replies.
|
527
|
+
#
|
528
|
+
# This does not mean that must necessarily be one. E.g, the replies
|
529
|
+
# produced by gen_each_reply could ignore the received_msgs
|
530
|
+
#
|
531
|
+
# @param mth [Proc] generates the BiDi stream replies
|
532
|
+
# @param interception_ctx [InterceptionContext]
|
533
|
+
#
|
534
|
+
def run_server_bidi(mth, interception_ctx)
|
535
|
+
view = multi_req_view
|
536
|
+
bidi_call = BidiCall.new(
|
537
|
+
@call,
|
538
|
+
@marshal,
|
539
|
+
@unmarshal,
|
540
|
+
metadata_received: @metadata_received,
|
541
|
+
req_view: view
|
542
|
+
)
|
543
|
+
requests = bidi_call.read_next_loop(proc { set_input_stream_done }, false)
|
544
|
+
interception_ctx.intercept!(
|
545
|
+
:bidi_streamer,
|
546
|
+
call: view,
|
547
|
+
method: mth,
|
548
|
+
requests: requests
|
549
|
+
) do
|
550
|
+
bidi_call.run_on_server(mth, requests)
|
551
|
+
end
|
552
|
+
end
|
553
|
+
|
554
|
+
# Waits till an operation completes
|
555
|
+
def wait
|
556
|
+
return if @op_notifier.nil?
|
557
|
+
GRPC.logger.debug("active_call.wait: on #{@op_notifier}")
|
558
|
+
@op_notifier.wait
|
559
|
+
end
|
560
|
+
|
561
|
+
# Signals that an operation is done.
|
562
|
+
# Only relevant on the client-side (this is a no-op on the server-side)
|
563
|
+
def op_is_done
|
564
|
+
return if @op_notifier.nil?
|
565
|
+
@op_notifier.notify(self)
|
566
|
+
end
|
567
|
+
|
568
|
+
# Add to the metadata that will be sent from the server.
|
569
|
+
# Fails if metadata has already been sent.
|
570
|
+
# Unused by client calls.
|
571
|
+
def merge_metadata_to_send(new_metadata = {})
|
572
|
+
@send_initial_md_mutex.synchronize do
|
573
|
+
fail('cant change metadata after already sent') if @metadata_sent
|
574
|
+
@metadata_to_send.merge!(new_metadata)
|
575
|
+
end
|
576
|
+
end
|
577
|
+
|
578
|
+
def attach_peer_cert(peer_cert)
|
579
|
+
@peer_cert = peer_cert
|
580
|
+
end
|
581
|
+
|
582
|
+
private
|
583
|
+
|
584
|
+
# To be called once the "input stream" has been completelly
|
585
|
+
# read through (i.e, done reading from client or received status)
|
586
|
+
# note this is idempotent
|
587
|
+
def set_input_stream_done
|
588
|
+
@call_finished_mu.synchronize do
|
589
|
+
@input_stream_done = true
|
590
|
+
maybe_finish_and_close_call_locked
|
591
|
+
end
|
592
|
+
end
|
593
|
+
|
594
|
+
# To be called once the "output stream" has been completelly
|
595
|
+
# sent through (i.e, done sending from client or sent status)
|
596
|
+
# note this is idempotent
|
597
|
+
def set_output_stream_done
|
598
|
+
@call_finished_mu.synchronize do
|
599
|
+
@output_stream_done = true
|
600
|
+
maybe_finish_and_close_call_locked
|
601
|
+
end
|
602
|
+
end
|
603
|
+
|
604
|
+
def maybe_finish_and_close_call_locked
|
605
|
+
return unless @output_stream_done && @input_stream_done
|
606
|
+
return if @call_finished
|
607
|
+
@call_finished = true
|
608
|
+
op_is_done
|
609
|
+
@call.close
|
610
|
+
end
|
611
|
+
|
612
|
+
# Starts the call if not already started
|
613
|
+
# @param metadata [Hash] metadata to be sent to the server. If a value is
|
614
|
+
# a list, multiple metadata for its key are sent
|
615
|
+
def start_call(metadata = {})
|
616
|
+
merge_metadata_to_send(metadata) && send_initial_metadata
|
617
|
+
end
|
618
|
+
|
619
|
+
def raise_error_if_already_executed
|
620
|
+
@client_call_executed_mu.synchronize do
|
621
|
+
if @client_call_executed
|
622
|
+
fail GRPC::Core::CallError, 'attempting to re-run a call'
|
623
|
+
end
|
624
|
+
@client_call_executed = true
|
625
|
+
end
|
626
|
+
end
|
627
|
+
|
628
|
+
def self.view_class(*visible_methods)
|
629
|
+
Class.new do
|
630
|
+
extend ::Forwardable
|
631
|
+
def_delegators :@wrapped, *visible_methods
|
632
|
+
|
633
|
+
# @param wrapped [ActiveCall] the call whose methods are shielded
|
634
|
+
def initialize(wrapped)
|
635
|
+
@wrapped = wrapped
|
636
|
+
end
|
637
|
+
end
|
638
|
+
end
|
639
|
+
|
640
|
+
# SingleReqView limits access to an ActiveCall's methods for use in server
|
641
|
+
# handlers that receive just one request.
|
642
|
+
SingleReqView = view_class(:cancelled?, :deadline, :metadata,
|
643
|
+
:output_metadata, :peer, :peer_cert,
|
644
|
+
:send_initial_metadata,
|
645
|
+
:metadata_to_send,
|
646
|
+
:merge_metadata_to_send,
|
647
|
+
:metadata_sent)
|
648
|
+
|
649
|
+
# MultiReqView limits access to an ActiveCall's methods for use in
|
650
|
+
# server client_streamer handlers.
|
651
|
+
MultiReqView = view_class(:cancelled?, :deadline,
|
652
|
+
:each_remote_read, :metadata, :output_metadata,
|
653
|
+
:peer, :peer_cert,
|
654
|
+
:send_initial_metadata,
|
655
|
+
:metadata_to_send,
|
656
|
+
:merge_metadata_to_send,
|
657
|
+
:metadata_sent)
|
658
|
+
|
659
|
+
# Operation limits access to an ActiveCall's methods for use as
|
660
|
+
# a Operation on the client.
|
661
|
+
Operation = view_class(:cancel, :cancelled?, :deadline, :execute,
|
662
|
+
:metadata, :status, :start_call, :wait, :write_flag,
|
663
|
+
:write_flag=, :trailing_metadata)
|
664
|
+
|
665
|
+
# InterceptableView further limits access to an ActiveCall's methods
|
666
|
+
# for use in interceptors on the client, exposing only the deadline
|
667
|
+
InterceptableView = view_class(:deadline)
|
668
|
+
end
|
669
|
+
end
|