grpc 1.30.2-universal-darwin
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.bundle +0 -0
- data/src/ruby/lib/grpc/2.4/grpc_c.bundle +0 -0
- data/src/ruby/lib/grpc/2.5/grpc_c.bundle +0 -0
- data/src/ruby/lib/grpc/2.6/grpc_c.bundle +0 -0
- data/src/ruby/lib/grpc/2.7/grpc_c.bundle +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,233 @@
|
|
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_relative '../grpc'
|
17
|
+
|
18
|
+
# GRPC contains the General RPC module.
|
19
|
+
module GRPC
|
20
|
+
# The BiDiCall class orchestrates execution of a BiDi stream on a client or
|
21
|
+
# server.
|
22
|
+
class BidiCall
|
23
|
+
include Core::CallOps
|
24
|
+
include Core::StatusCodes
|
25
|
+
include Core::TimeConsts
|
26
|
+
|
27
|
+
# Creates a BidiCall.
|
28
|
+
#
|
29
|
+
# BidiCall should only be created after a call is accepted. That means
|
30
|
+
# different things on a client and a server. On the client, the call is
|
31
|
+
# accepted after call.invoke. On the server, this is after call.accept.
|
32
|
+
#
|
33
|
+
# #initialize cannot determine if the call is accepted or not; so if a
|
34
|
+
# call that's not accepted is used here, the error won't be visible until
|
35
|
+
# the BidiCall#run is called.
|
36
|
+
#
|
37
|
+
# deadline is the absolute deadline for the call.
|
38
|
+
#
|
39
|
+
# @param call [Call] the call used by the ActiveCall
|
40
|
+
# @param marshal [Function] f(obj)->string that marshal requests
|
41
|
+
# @param unmarshal [Function] f(string)->obj that unmarshals responses
|
42
|
+
# @param metadata_received [true|false] indicates if metadata has already
|
43
|
+
# been received. Should always be true for server calls
|
44
|
+
def initialize(call, marshal, unmarshal, metadata_received: false,
|
45
|
+
req_view: nil)
|
46
|
+
fail(ArgumentError, 'not a call') unless call.is_a? Core::Call
|
47
|
+
@call = call
|
48
|
+
@marshal = marshal
|
49
|
+
@op_notifier = nil # signals completion on clients
|
50
|
+
@unmarshal = unmarshal
|
51
|
+
@metadata_received = metadata_received
|
52
|
+
@reads_complete = false
|
53
|
+
@writes_complete = false
|
54
|
+
@complete = false
|
55
|
+
@done_mutex = Mutex.new
|
56
|
+
@req_view = req_view
|
57
|
+
end
|
58
|
+
|
59
|
+
# Begins orchestration of the Bidi stream for a client sending requests.
|
60
|
+
#
|
61
|
+
# The method either returns an Enumerator of the responses, or accepts a
|
62
|
+
# block that can be invoked with each response.
|
63
|
+
#
|
64
|
+
# @param requests the Enumerable of requests to send
|
65
|
+
# @param set_input_stream_done [Proc] called back when we're done
|
66
|
+
# reading the input stream
|
67
|
+
# @param set_output_stream_done [Proc] called back when we're done
|
68
|
+
# sending data on the output stream
|
69
|
+
# @return an Enumerator of requests to yield
|
70
|
+
def run_on_client(requests,
|
71
|
+
set_input_stream_done,
|
72
|
+
set_output_stream_done,
|
73
|
+
&blk)
|
74
|
+
@enq_th = Thread.new do
|
75
|
+
write_loop(requests, set_output_stream_done: set_output_stream_done)
|
76
|
+
end
|
77
|
+
read_loop(set_input_stream_done, &blk)
|
78
|
+
end
|
79
|
+
|
80
|
+
# Begins orchestration of the Bidi stream for a server generating replies.
|
81
|
+
#
|
82
|
+
# N.B. gen_each_reply is a func(Enumerable<Requests>)
|
83
|
+
#
|
84
|
+
# It takes an enumerable of requests as an arg, in case there is a
|
85
|
+
# relationship between the stream of requests and the stream of replies.
|
86
|
+
#
|
87
|
+
# This does not mean that must necessarily be one. E.g, the replies
|
88
|
+
# produced by gen_each_reply could ignore the received_msgs
|
89
|
+
#
|
90
|
+
# @param [Proc] gen_each_reply generates the BiDi stream replies.
|
91
|
+
# @param [Enumerable] requests The enumerable of requests to run
|
92
|
+
def run_on_server(gen_each_reply, requests)
|
93
|
+
replies = nil
|
94
|
+
|
95
|
+
# Pass in the optional call object parameter if possible
|
96
|
+
if gen_each_reply.arity == 1
|
97
|
+
replies = gen_each_reply.call(requests)
|
98
|
+
elsif gen_each_reply.arity == 2
|
99
|
+
replies = gen_each_reply.call(requests, @req_view)
|
100
|
+
else
|
101
|
+
fail 'Illegal arity of reply generator'
|
102
|
+
end
|
103
|
+
|
104
|
+
write_loop(replies, is_client: false)
|
105
|
+
end
|
106
|
+
|
107
|
+
##
|
108
|
+
# Read the next stream iteration
|
109
|
+
#
|
110
|
+
# @param [Proc] finalize_stream callback to call when the reads have been
|
111
|
+
# completely read through.
|
112
|
+
# @param [Boolean] is_client If this is a client or server request
|
113
|
+
#
|
114
|
+
def read_next_loop(finalize_stream, is_client = false)
|
115
|
+
read_loop(finalize_stream, is_client: is_client)
|
116
|
+
end
|
117
|
+
|
118
|
+
private
|
119
|
+
|
120
|
+
END_OF_READS = :end_of_reads
|
121
|
+
END_OF_WRITES = :end_of_writes
|
122
|
+
|
123
|
+
# performs a read using @call.run_batch, ensures metadata is set up
|
124
|
+
def read_using_run_batch
|
125
|
+
ops = { RECV_MESSAGE => nil }
|
126
|
+
ops[RECV_INITIAL_METADATA] = nil unless @metadata_received
|
127
|
+
begin
|
128
|
+
batch_result = @call.run_batch(ops)
|
129
|
+
unless @metadata_received
|
130
|
+
@call.metadata = batch_result.metadata
|
131
|
+
@metadata_received = true
|
132
|
+
end
|
133
|
+
batch_result
|
134
|
+
rescue GRPC::Core::CallError => e
|
135
|
+
GRPC.logger.warn('bidi call: read_using_run_batch failed')
|
136
|
+
GRPC.logger.warn(e)
|
137
|
+
nil
|
138
|
+
end
|
139
|
+
end
|
140
|
+
|
141
|
+
# set_output_stream_done is relevant on client-side
|
142
|
+
def write_loop(requests, is_client: true, set_output_stream_done: nil)
|
143
|
+
GRPC.logger.debug('bidi-write-loop: starting')
|
144
|
+
count = 0
|
145
|
+
requests.each do |req|
|
146
|
+
GRPC.logger.debug("bidi-write-loop: #{count}")
|
147
|
+
count += 1
|
148
|
+
payload = @marshal.call(req)
|
149
|
+
# Fails if status already received
|
150
|
+
begin
|
151
|
+
@req_view.send_initial_metadata unless @req_view.nil?
|
152
|
+
@call.run_batch(SEND_MESSAGE => payload)
|
153
|
+
rescue GRPC::Core::CallError => e
|
154
|
+
# This is almost definitely caused by a status arriving while still
|
155
|
+
# writing. Don't re-throw the error
|
156
|
+
GRPC.logger.warn('bidi-write-loop: ended with error')
|
157
|
+
GRPC.logger.warn(e)
|
158
|
+
break
|
159
|
+
end
|
160
|
+
end
|
161
|
+
GRPC.logger.debug("bidi-write-loop: #{count} writes done")
|
162
|
+
if is_client
|
163
|
+
GRPC.logger.debug("bidi-write-loop: client sent #{count}, waiting")
|
164
|
+
begin
|
165
|
+
@call.run_batch(SEND_CLOSE_FROM_CLIENT => nil)
|
166
|
+
rescue GRPC::Core::CallError => e
|
167
|
+
GRPC.logger.warn('bidi-write-loop: send close failed')
|
168
|
+
GRPC.logger.warn(e)
|
169
|
+
end
|
170
|
+
GRPC.logger.debug('bidi-write-loop: done')
|
171
|
+
end
|
172
|
+
GRPC.logger.debug('bidi-write-loop: finished')
|
173
|
+
rescue StandardError => e
|
174
|
+
GRPC.logger.warn('bidi-write-loop: failed')
|
175
|
+
GRPC.logger.warn(e)
|
176
|
+
if is_client
|
177
|
+
@call.cancel_with_status(GRPC::Core::StatusCodes::UNKNOWN,
|
178
|
+
"GRPC bidi call error: #{e.inspect}")
|
179
|
+
else
|
180
|
+
raise e
|
181
|
+
end
|
182
|
+
ensure
|
183
|
+
set_output_stream_done.call if is_client
|
184
|
+
end
|
185
|
+
|
186
|
+
# Provides an enumerator that yields results of remote reads
|
187
|
+
def read_loop(set_input_stream_done, is_client: true)
|
188
|
+
return enum_for(:read_loop,
|
189
|
+
set_input_stream_done,
|
190
|
+
is_client: is_client) unless block_given?
|
191
|
+
GRPC.logger.debug('bidi-read-loop: starting')
|
192
|
+
begin
|
193
|
+
count = 0
|
194
|
+
# queue the initial read before beginning the loop
|
195
|
+
loop do
|
196
|
+
GRPC.logger.debug("bidi-read-loop: #{count}")
|
197
|
+
count += 1
|
198
|
+
batch_result = read_using_run_batch
|
199
|
+
|
200
|
+
# handle the next message
|
201
|
+
if batch_result.nil? || batch_result.message.nil?
|
202
|
+
GRPC.logger.debug("bidi-read-loop: null batch #{batch_result}")
|
203
|
+
|
204
|
+
if is_client
|
205
|
+
batch_result = @call.run_batch(RECV_STATUS_ON_CLIENT => nil)
|
206
|
+
@call.status = batch_result.status
|
207
|
+
@call.trailing_metadata = @call.status.metadata if @call.status
|
208
|
+
GRPC.logger.debug("bidi-read-loop: done status #{@call.status}")
|
209
|
+
batch_result.check_status
|
210
|
+
end
|
211
|
+
|
212
|
+
GRPC.logger.debug('bidi-read-loop: done reading!')
|
213
|
+
break
|
214
|
+
end
|
215
|
+
|
216
|
+
res = @unmarshal.call(batch_result.message)
|
217
|
+
yield res
|
218
|
+
end
|
219
|
+
rescue StandardError => e
|
220
|
+
GRPC.logger.warn('bidi: read-loop failed')
|
221
|
+
GRPC.logger.warn(e)
|
222
|
+
raise e
|
223
|
+
ensure
|
224
|
+
set_input_stream_done.call
|
225
|
+
end
|
226
|
+
GRPC.logger.debug('bidi-read-loop: finished')
|
227
|
+
# Make sure that the write loop is done before finishing the call.
|
228
|
+
# Note that blocking is ok at this point because we've already received
|
229
|
+
# a status
|
230
|
+
@enq_th.join if is_client
|
231
|
+
end
|
232
|
+
end
|
233
|
+
end
|
@@ -0,0 +1,501 @@
|
|
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 'active_call'
|
16
|
+
require_relative '../version'
|
17
|
+
|
18
|
+
# GRPC contains the General RPC module.
|
19
|
+
module GRPC
|
20
|
+
# rubocop:disable Metrics/ParameterLists
|
21
|
+
|
22
|
+
# ClientStub represents a client connection to a gRPC server, and can be used
|
23
|
+
# to send requests.
|
24
|
+
class ClientStub
|
25
|
+
include Core::StatusCodes
|
26
|
+
include Core::TimeConsts
|
27
|
+
|
28
|
+
# Default timeout is infinity.
|
29
|
+
DEFAULT_TIMEOUT = INFINITE_FUTURE
|
30
|
+
|
31
|
+
# setup_channel is used by #initialize to constuct a channel from its
|
32
|
+
# arguments.
|
33
|
+
def self.setup_channel(alt_chan, host, creds, channel_args = {})
|
34
|
+
unless alt_chan.nil?
|
35
|
+
fail(TypeError, '!Channel') unless alt_chan.is_a?(Core::Channel)
|
36
|
+
return alt_chan
|
37
|
+
end
|
38
|
+
if channel_args['grpc.primary_user_agent'].nil?
|
39
|
+
channel_args['grpc.primary_user_agent'] = ''
|
40
|
+
else
|
41
|
+
channel_args['grpc.primary_user_agent'] += ' '
|
42
|
+
end
|
43
|
+
channel_args['grpc.primary_user_agent'] += "grpc-ruby/#{VERSION}"
|
44
|
+
unless creds.is_a?(Core::ChannelCredentials) || creds.is_a?(Symbol)
|
45
|
+
fail(TypeError, '!ChannelCredentials or Symbol')
|
46
|
+
end
|
47
|
+
Core::Channel.new(host, channel_args, creds)
|
48
|
+
end
|
49
|
+
|
50
|
+
# Allows users of the stub to modify the propagate mask.
|
51
|
+
#
|
52
|
+
# This is an advanced feature for use when making calls to another gRPC
|
53
|
+
# server whilst running in the handler of an existing one.
|
54
|
+
attr_writer :propagate_mask
|
55
|
+
|
56
|
+
# Creates a new ClientStub.
|
57
|
+
#
|
58
|
+
# Minimally, a stub is created with the just the host of the gRPC service
|
59
|
+
# it wishes to access, e.g.,
|
60
|
+
#
|
61
|
+
# my_stub = ClientStub.new(example.host.com:50505,
|
62
|
+
# :this_channel_is_insecure)
|
63
|
+
#
|
64
|
+
# If a channel_override argument is passed, it will be used as the
|
65
|
+
# underlying channel. Otherwise, the channel_args argument will be used
|
66
|
+
# to construct a new underlying channel.
|
67
|
+
#
|
68
|
+
# There are some specific keyword args that are not used to configure the
|
69
|
+
# channel:
|
70
|
+
#
|
71
|
+
# - :channel_override
|
72
|
+
# when present, this must be a pre-created GRPC::Core::Channel. If it's
|
73
|
+
# present the host and arbitrary keyword arg areignored, and the RPC
|
74
|
+
# connection uses this channel.
|
75
|
+
#
|
76
|
+
# - :timeout
|
77
|
+
# when present, this is the default timeout used for calls
|
78
|
+
#
|
79
|
+
# @param host [String] the host the stub connects to
|
80
|
+
# @param creds [Core::ChannelCredentials|Symbol] the channel credentials, or
|
81
|
+
# :this_channel_is_insecure, which explicitly indicates that the client
|
82
|
+
# should be created with an insecure connection. Note: this argument is
|
83
|
+
# ignored if the channel_override argument is provided.
|
84
|
+
# @param channel_override [Core::Channel] a pre-created channel
|
85
|
+
# @param timeout [Number] the default timeout to use in requests
|
86
|
+
# @param propagate_mask [Number] A bitwise combination of flags in
|
87
|
+
# GRPC::Core::PropagateMasks. Indicates how data should be propagated
|
88
|
+
# from parent server calls to child client calls if this client is being
|
89
|
+
# used within a gRPC server.
|
90
|
+
# @param channel_args [Hash] the channel arguments. Note: this argument is
|
91
|
+
# ignored if the channel_override argument is provided.
|
92
|
+
# @param interceptors [Array<GRPC::ClientInterceptor>] An array of
|
93
|
+
# GRPC::ClientInterceptor objects that will be used for
|
94
|
+
# intercepting calls before they are executed
|
95
|
+
# Interceptors are an EXPERIMENTAL API.
|
96
|
+
def initialize(host, creds,
|
97
|
+
channel_override: nil,
|
98
|
+
timeout: nil,
|
99
|
+
propagate_mask: nil,
|
100
|
+
channel_args: {},
|
101
|
+
interceptors: [])
|
102
|
+
@ch = ClientStub.setup_channel(channel_override, host, creds,
|
103
|
+
channel_args)
|
104
|
+
alt_host = channel_args[Core::Channel::SSL_TARGET]
|
105
|
+
@host = alt_host.nil? ? host : alt_host
|
106
|
+
@propagate_mask = propagate_mask
|
107
|
+
@timeout = timeout.nil? ? DEFAULT_TIMEOUT : timeout
|
108
|
+
@interceptors = InterceptorRegistry.new(interceptors)
|
109
|
+
end
|
110
|
+
|
111
|
+
# request_response sends a request to a GRPC server, and returns the
|
112
|
+
# response.
|
113
|
+
#
|
114
|
+
# == Flow Control ==
|
115
|
+
# This is a blocking call.
|
116
|
+
#
|
117
|
+
# * it does not return until a response is received.
|
118
|
+
#
|
119
|
+
# * the requests is sent only when GRPC core's flow control allows it to
|
120
|
+
# be sent.
|
121
|
+
#
|
122
|
+
# == Errors ==
|
123
|
+
# An RuntimeError is raised if
|
124
|
+
#
|
125
|
+
# * the server responds with a non-OK status
|
126
|
+
#
|
127
|
+
# * the deadline is exceeded
|
128
|
+
#
|
129
|
+
# == Return Value ==
|
130
|
+
#
|
131
|
+
# If return_op is false, the call returns the response
|
132
|
+
#
|
133
|
+
# If return_op is true, the call returns an Operation, calling execute
|
134
|
+
# on the Operation returns the response.
|
135
|
+
#
|
136
|
+
# @param method [String] the RPC method to call on the GRPC server
|
137
|
+
# @param req [Object] the request sent to the server
|
138
|
+
# @param marshal [Function] f(obj)->string that marshals requests
|
139
|
+
# @param unmarshal [Function] f(string)->obj that unmarshals responses
|
140
|
+
# @param deadline [Time] (optional) the time the request should complete
|
141
|
+
# @param return_op [true|false] return an Operation if true
|
142
|
+
# @param parent [Core::Call] a prior call whose reserved metadata
|
143
|
+
# will be propagated by this one.
|
144
|
+
# @param credentials [Core::CallCredentials] credentials to use when making
|
145
|
+
# the call
|
146
|
+
# @param metadata [Hash] metadata to be sent to the server
|
147
|
+
# @return [Object] the response received from the server
|
148
|
+
def request_response(method, req, marshal, unmarshal,
|
149
|
+
deadline: nil,
|
150
|
+
return_op: false,
|
151
|
+
parent: nil,
|
152
|
+
credentials: nil,
|
153
|
+
metadata: {})
|
154
|
+
c = new_active_call(method, marshal, unmarshal,
|
155
|
+
deadline: deadline,
|
156
|
+
parent: parent,
|
157
|
+
credentials: credentials)
|
158
|
+
interception_context = @interceptors.build_context
|
159
|
+
intercept_args = {
|
160
|
+
method: method,
|
161
|
+
request: req,
|
162
|
+
call: c.interceptable,
|
163
|
+
metadata: metadata
|
164
|
+
}
|
165
|
+
if return_op
|
166
|
+
# return the operation view of the active_call; define #execute as a
|
167
|
+
# new method for this instance that invokes #request_response.
|
168
|
+
c.merge_metadata_to_send(metadata)
|
169
|
+
op = c.operation
|
170
|
+
op.define_singleton_method(:execute) do
|
171
|
+
interception_context.intercept!(:request_response, intercept_args) do
|
172
|
+
c.request_response(req, metadata: metadata)
|
173
|
+
end
|
174
|
+
end
|
175
|
+
op
|
176
|
+
else
|
177
|
+
interception_context.intercept!(:request_response, intercept_args) do
|
178
|
+
c.request_response(req, metadata: metadata)
|
179
|
+
end
|
180
|
+
end
|
181
|
+
end
|
182
|
+
|
183
|
+
# client_streamer sends a stream of requests to a GRPC server, and
|
184
|
+
# returns a single response.
|
185
|
+
#
|
186
|
+
# requests provides an 'iterable' of Requests. I.e. it follows Ruby's
|
187
|
+
# #each enumeration protocol. In the simplest case, requests will be an
|
188
|
+
# array of marshallable objects; in typical case it will be an Enumerable
|
189
|
+
# that allows dynamic construction of the marshallable objects.
|
190
|
+
#
|
191
|
+
# == Flow Control ==
|
192
|
+
# This is a blocking call.
|
193
|
+
#
|
194
|
+
# * it does not return until a response is received.
|
195
|
+
#
|
196
|
+
# * each requests is sent only when GRPC core's flow control allows it to
|
197
|
+
# be sent.
|
198
|
+
#
|
199
|
+
# == Errors ==
|
200
|
+
# An RuntimeError is raised if
|
201
|
+
#
|
202
|
+
# * the server responds with a non-OK status
|
203
|
+
#
|
204
|
+
# * the deadline is exceeded
|
205
|
+
#
|
206
|
+
# == Return Value ==
|
207
|
+
#
|
208
|
+
# If return_op is false, the call consumes the requests and returns
|
209
|
+
# the response.
|
210
|
+
#
|
211
|
+
# If return_op is true, the call returns the response.
|
212
|
+
#
|
213
|
+
# @param method [String] the RPC method to call on the GRPC server
|
214
|
+
# @param requests [Object] an Enumerable of requests to send
|
215
|
+
# @param marshal [Function] f(obj)->string that marshals requests
|
216
|
+
# @param unmarshal [Function] f(string)->obj that unmarshals responses
|
217
|
+
# @param deadline [Time] (optional) the time the request should complete
|
218
|
+
# @param return_op [true|false] return an Operation if true
|
219
|
+
# @param parent [Core::Call] a prior call whose reserved metadata
|
220
|
+
# will be propagated by this one.
|
221
|
+
# @param credentials [Core::CallCredentials] credentials to use when making
|
222
|
+
# the call
|
223
|
+
# @param metadata [Hash] metadata to be sent to the server
|
224
|
+
# @return [Object|Operation] the response received from the server
|
225
|
+
def client_streamer(method, requests, marshal, unmarshal,
|
226
|
+
deadline: nil,
|
227
|
+
return_op: false,
|
228
|
+
parent: nil,
|
229
|
+
credentials: nil,
|
230
|
+
metadata: {})
|
231
|
+
c = new_active_call(method, marshal, unmarshal,
|
232
|
+
deadline: deadline,
|
233
|
+
parent: parent,
|
234
|
+
credentials: credentials)
|
235
|
+
interception_context = @interceptors.build_context
|
236
|
+
intercept_args = {
|
237
|
+
method: method,
|
238
|
+
requests: requests,
|
239
|
+
call: c.interceptable,
|
240
|
+
metadata: metadata
|
241
|
+
}
|
242
|
+
if return_op
|
243
|
+
# return the operation view of the active_call; define #execute as a
|
244
|
+
# new method for this instance that invokes #client_streamer.
|
245
|
+
c.merge_metadata_to_send(metadata)
|
246
|
+
op = c.operation
|
247
|
+
op.define_singleton_method(:execute) do
|
248
|
+
interception_context.intercept!(:client_streamer, intercept_args) do
|
249
|
+
c.client_streamer(requests)
|
250
|
+
end
|
251
|
+
end
|
252
|
+
op
|
253
|
+
else
|
254
|
+
interception_context.intercept!(:client_streamer, intercept_args) do
|
255
|
+
c.client_streamer(requests, metadata: metadata)
|
256
|
+
end
|
257
|
+
end
|
258
|
+
end
|
259
|
+
|
260
|
+
# server_streamer sends one request to the GRPC server, which yields a
|
261
|
+
# stream of responses.
|
262
|
+
#
|
263
|
+
# responses provides an enumerator over the streamed responses, i.e. it
|
264
|
+
# follows Ruby's #each iteration protocol. The enumerator blocks while
|
265
|
+
# waiting for each response, stops when the server signals that no
|
266
|
+
# further responses will be supplied. If the implicit block is provided,
|
267
|
+
# it is executed with each response as the argument and no result is
|
268
|
+
# returned.
|
269
|
+
#
|
270
|
+
# == Flow Control ==
|
271
|
+
# This is a blocking call.
|
272
|
+
#
|
273
|
+
# * the request is sent only when GRPC core's flow control allows it to
|
274
|
+
# be sent.
|
275
|
+
#
|
276
|
+
# * the request will not complete until the server sends the final
|
277
|
+
# response followed by a status message.
|
278
|
+
#
|
279
|
+
# == Errors ==
|
280
|
+
# An RuntimeError is raised if
|
281
|
+
#
|
282
|
+
# * the server responds with a non-OK status when any response is
|
283
|
+
# * retrieved
|
284
|
+
#
|
285
|
+
# * the deadline is exceeded
|
286
|
+
#
|
287
|
+
# == Return Value ==
|
288
|
+
#
|
289
|
+
# if the return_op is false, the return value is an Enumerator of the
|
290
|
+
# results, unless a block is provided, in which case the block is
|
291
|
+
# executed with each response.
|
292
|
+
#
|
293
|
+
# if return_op is true, the function returns an Operation whose #execute
|
294
|
+
# method runs server streamer call. Again, Operation#execute either
|
295
|
+
# calls the given block with each response or returns an Enumerator of the
|
296
|
+
# responses.
|
297
|
+
#
|
298
|
+
# == Keyword Args ==
|
299
|
+
#
|
300
|
+
# Unspecified keyword arguments are treated as metadata to be sent to the
|
301
|
+
# server.
|
302
|
+
#
|
303
|
+
# @param method [String] the RPC method to call on the GRPC server
|
304
|
+
# @param req [Object] the request sent to the server
|
305
|
+
# @param marshal [Function] f(obj)->string that marshals requests
|
306
|
+
# @param unmarshal [Function] f(string)->obj that unmarshals responses
|
307
|
+
# @param deadline [Time] (optional) the time the request should complete
|
308
|
+
# @param return_op [true|false]return an Operation if true
|
309
|
+
# @param parent [Core::Call] a prior call whose reserved metadata
|
310
|
+
# will be propagated by this one.
|
311
|
+
# @param credentials [Core::CallCredentials] credentials to use when making
|
312
|
+
# the call
|
313
|
+
# @param metadata [Hash] metadata to be sent to the server
|
314
|
+
# @param blk [Block] when provided, is executed for each response
|
315
|
+
# @return [Enumerator|Operation|nil] as discussed above
|
316
|
+
def server_streamer(method, req, marshal, unmarshal,
|
317
|
+
deadline: nil,
|
318
|
+
return_op: false,
|
319
|
+
parent: nil,
|
320
|
+
credentials: nil,
|
321
|
+
metadata: {},
|
322
|
+
&blk)
|
323
|
+
c = new_active_call(method, marshal, unmarshal,
|
324
|
+
deadline: deadline,
|
325
|
+
parent: parent,
|
326
|
+
credentials: credentials)
|
327
|
+
interception_context = @interceptors.build_context
|
328
|
+
intercept_args = {
|
329
|
+
method: method,
|
330
|
+
request: req,
|
331
|
+
call: c.interceptable,
|
332
|
+
metadata: metadata
|
333
|
+
}
|
334
|
+
if return_op
|
335
|
+
# return the operation view of the active_call; define #execute
|
336
|
+
# as a new method for this instance that invokes #server_streamer
|
337
|
+
c.merge_metadata_to_send(metadata)
|
338
|
+
op = c.operation
|
339
|
+
op.define_singleton_method(:execute) do
|
340
|
+
interception_context.intercept!(:server_streamer, intercept_args) do
|
341
|
+
c.server_streamer(req, &blk)
|
342
|
+
end
|
343
|
+
end
|
344
|
+
op
|
345
|
+
else
|
346
|
+
interception_context.intercept!(:server_streamer, intercept_args) do
|
347
|
+
c.server_streamer(req, metadata: metadata, &blk)
|
348
|
+
end
|
349
|
+
end
|
350
|
+
end
|
351
|
+
|
352
|
+
# bidi_streamer sends a stream of requests to the GRPC server, and yields
|
353
|
+
# a stream of responses.
|
354
|
+
#
|
355
|
+
# This method takes an Enumerable of requests, and returns and enumerable
|
356
|
+
# of responses.
|
357
|
+
#
|
358
|
+
# == requests ==
|
359
|
+
#
|
360
|
+
# requests provides an 'iterable' of Requests. I.e. it follows Ruby's
|
361
|
+
# #each enumeration protocol. In the simplest case, requests will be an
|
362
|
+
# array of marshallable objects; in typical case it will be an
|
363
|
+
# Enumerable that allows dynamic construction of the marshallable
|
364
|
+
# objects.
|
365
|
+
#
|
366
|
+
# == responses ==
|
367
|
+
#
|
368
|
+
# This is an enumerator of responses. I.e, its #next method blocks
|
369
|
+
# waiting for the next response. Also, if at any point the block needs
|
370
|
+
# to consume all the remaining responses, this can be done using #each or
|
371
|
+
# #collect. Calling #each or #collect should only be done if
|
372
|
+
# the_call#writes_done has been called, otherwise the block will loop
|
373
|
+
# forever.
|
374
|
+
#
|
375
|
+
# == Flow Control ==
|
376
|
+
# This is a blocking call.
|
377
|
+
#
|
378
|
+
# * the call completes when the next call to provided block returns
|
379
|
+
# false
|
380
|
+
#
|
381
|
+
# * the execution block parameters are two objects for sending and
|
382
|
+
# receiving responses, each of which blocks waiting for flow control.
|
383
|
+
# E.g, calles to bidi_call#remote_send will wait until flow control
|
384
|
+
# allows another write before returning; and obviously calls to
|
385
|
+
# responses#next block until the next response is available.
|
386
|
+
#
|
387
|
+
# == Termination ==
|
388
|
+
#
|
389
|
+
# As well as sending and receiving messages, the block passed to the
|
390
|
+
# function is also responsible for:
|
391
|
+
#
|
392
|
+
# * calling bidi_call#writes_done to indicate no further reqs will be
|
393
|
+
# sent.
|
394
|
+
#
|
395
|
+
# * returning false if once the bidi stream is functionally completed.
|
396
|
+
#
|
397
|
+
# Note that response#next will indicate that there are no further
|
398
|
+
# responses by throwing StopIteration, but can only happen either
|
399
|
+
# if bidi_call#writes_done is called.
|
400
|
+
#
|
401
|
+
# To properly terminate the RPC, the responses should be completely iterated
|
402
|
+
# through; one way to do this is to loop on responses#next until no further
|
403
|
+
# responses are available.
|
404
|
+
#
|
405
|
+
# == Errors ==
|
406
|
+
# An RuntimeError is raised if
|
407
|
+
#
|
408
|
+
# * the server responds with a non-OK status when any response is
|
409
|
+
# * retrieved
|
410
|
+
#
|
411
|
+
# * the deadline is exceeded
|
412
|
+
#
|
413
|
+
#
|
414
|
+
# == Return Value ==
|
415
|
+
#
|
416
|
+
# if the return_op is false, the return value is an Enumerator of the
|
417
|
+
# results, unless a block is provided, in which case the block is
|
418
|
+
# executed with each response.
|
419
|
+
#
|
420
|
+
# if return_op is true, the function returns an Operation whose #execute
|
421
|
+
# method runs the Bidi call. Again, Operation#execute either calls a
|
422
|
+
# given block with each response or returns an Enumerator of the
|
423
|
+
# responses.
|
424
|
+
#
|
425
|
+
# @param method [String] the RPC method to call on the GRPC server
|
426
|
+
# @param requests [Object] an Enumerable of requests to send
|
427
|
+
# @param marshal [Function] f(obj)->string that marshals requests
|
428
|
+
# @param unmarshal [Function] f(string)->obj that unmarshals responses
|
429
|
+
# @param deadline [Time] (optional) the time the request should complete
|
430
|
+
# @param return_op [true|false] return an Operation if true
|
431
|
+
# @param parent [Core::Call] a prior call whose reserved metadata
|
432
|
+
# will be propagated by this one.
|
433
|
+
# @param credentials [Core::CallCredentials] credentials to use when making
|
434
|
+
# the call
|
435
|
+
# @param metadata [Hash] metadata to be sent to the server
|
436
|
+
# @param blk [Block] when provided, is executed for each response
|
437
|
+
# @return [Enumerator|nil|Operation] as discussed above
|
438
|
+
def bidi_streamer(method, requests, marshal, unmarshal,
|
439
|
+
deadline: nil,
|
440
|
+
return_op: false,
|
441
|
+
parent: nil,
|
442
|
+
credentials: nil,
|
443
|
+
metadata: {},
|
444
|
+
&blk)
|
445
|
+
c = new_active_call(method, marshal, unmarshal,
|
446
|
+
deadline: deadline,
|
447
|
+
parent: parent,
|
448
|
+
credentials: credentials)
|
449
|
+
interception_context = @interceptors.build_context
|
450
|
+
intercept_args = {
|
451
|
+
method: method,
|
452
|
+
requests: requests,
|
453
|
+
call: c.interceptable,
|
454
|
+
metadata: metadata
|
455
|
+
}
|
456
|
+
if return_op
|
457
|
+
# return the operation view of the active_call; define #execute
|
458
|
+
# as a new method for this instance that invokes #bidi_streamer
|
459
|
+
c.merge_metadata_to_send(metadata)
|
460
|
+
op = c.operation
|
461
|
+
op.define_singleton_method(:execute) do
|
462
|
+
interception_context.intercept!(:bidi_streamer, intercept_args) do
|
463
|
+
c.bidi_streamer(requests, &blk)
|
464
|
+
end
|
465
|
+
end
|
466
|
+
op
|
467
|
+
else
|
468
|
+
interception_context.intercept!(:bidi_streamer, intercept_args) do
|
469
|
+
c.bidi_streamer(requests, metadata: metadata, &blk)
|
470
|
+
end
|
471
|
+
end
|
472
|
+
end
|
473
|
+
|
474
|
+
private
|
475
|
+
|
476
|
+
# Creates a new active stub
|
477
|
+
#
|
478
|
+
# @param method [string] the method being called.
|
479
|
+
# @param marshal [Function] f(obj)->string that marshals requests
|
480
|
+
# @param unmarshal [Function] f(string)->obj that unmarshals responses
|
481
|
+
# @param parent [Grpc::Call] a parent call, available when calls are
|
482
|
+
# made from server
|
483
|
+
# @param credentials [Core::CallCredentials] credentials to use when making
|
484
|
+
# the call
|
485
|
+
def new_active_call(method, marshal, unmarshal,
|
486
|
+
deadline: nil,
|
487
|
+
parent: nil,
|
488
|
+
credentials: nil)
|
489
|
+
deadline = from_relative_time(@timeout) if deadline.nil?
|
490
|
+
# Provide each new client call with its own completion queue
|
491
|
+
call = @ch.create_call(parent, # parent call
|
492
|
+
@propagate_mask, # propagation options
|
493
|
+
method,
|
494
|
+
nil, # host use nil,
|
495
|
+
deadline)
|
496
|
+
call.set_credentials! credentials unless credentials.nil?
|
497
|
+
ActiveCall.new(call, marshal, unmarshal, deadline,
|
498
|
+
started: false)
|
499
|
+
end
|
500
|
+
end
|
501
|
+
end
|