Class: Prefab::CancellableInterceptor
- Inherits:
-
GRPC::ClientInterceptor
- Object
- GRPC::ClientInterceptor
- Prefab::CancellableInterceptor
- Defined in:
- lib/prefab/cancellable_interceptor.rb
Constant Summary collapse
- WAIT_SEC =
3
Instance Method Summary collapse
- #bidi_streamer(requests:, call:, method:, metadata:, &block) ⇒ Object
- #cancel ⇒ Object
- #client_streamer(requests:, call:, method:, metadata:, &block) ⇒ Object
-
#initialize(base_client) ⇒ CancellableInterceptor
constructor
A new instance of CancellableInterceptor.
- #request_response(request:, call:, method:, metadata:, &block) ⇒ Object
- #server_streamer(request:, call:, method:, metadata:, &block) ⇒ Object
- #shared(call) ⇒ Object
Constructor Details
#initialize(base_client) ⇒ CancellableInterceptor
Returns a new instance of CancellableInterceptor.
6 7 8 |
# File 'lib/prefab/cancellable_interceptor.rb', line 6 def initialize(base_client) @base_client = base_client end |
Instance Method Details
#bidi_streamer(requests:, call:, method:, metadata:, &block) ⇒ Object
39 40 41 |
# File 'lib/prefab/cancellable_interceptor.rb', line 39 def bidi_streamer(requests:, call:, method:, metadata:, &block) shared(call, &block) end |
#cancel ⇒ Object
10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
# File 'lib/prefab/cancellable_interceptor.rb', line 10 def cancel @call.instance_variable_get("@wrapped").instance_variable_get("@call").cancel i = 0 while (i < WAIT_SEC) do if @call.instance_variable_get("@wrapped").cancelled? @base_client.log_internal Logger::DEBUG, "Cancelled streaming." return else @base_client.log_internal Logger::DEBUG, "Unable to cancel streaming. Trying again" @call.instance_variable_get("@wrapped").instance_variable_get("@call").cancel i += 1 sleep(1) end end @base_client.log_internal Logger::INFO, "Unable to cancel streaming." end |
#client_streamer(requests:, call:, method:, metadata:, &block) ⇒ Object
31 32 33 |
# File 'lib/prefab/cancellable_interceptor.rb', line 31 def client_streamer(requests:, call:, method:, metadata:, &block) shared(call, &block) end |
#request_response(request:, call:, method:, metadata:, &block) ⇒ Object
27 28 29 |
# File 'lib/prefab/cancellable_interceptor.rb', line 27 def request_response(request:, call:, method:, metadata:, &block) shared(call, &block) end |
#server_streamer(request:, call:, method:, metadata:, &block) ⇒ Object
35 36 37 |
# File 'lib/prefab/cancellable_interceptor.rb', line 35 def server_streamer(request:, call:, method:, metadata:, &block) shared(call, &block) end |
#shared(call) ⇒ Object
43 44 45 46 |
# File 'lib/prefab/cancellable_interceptor.rb', line 43 def shared(call) @call = call yield end |