Contents

# frozen_string_literal: true

require 'grpc_kit/rpcs'
require 'grpc_kit/calls/client_request_response'

module GrpcKit
  module Rpcs::Client
    class RequestResponse < GrpcKit::Rpcs::ClientRpc
      def invoke(stream, request, metadata: {}, timeout: nil)
        call = GrpcKit::Calls::Client::RequestResponse.new(
          metadata: metadata,
          config: @config,
          timeout: timeout,
          stream: stream,
        )

        # TODO: DRY
        if @config.interceptor && timeout
          @config.interceptor.intercept(request, call, call.metadata) do |r, c, _|
            Timeout.timeout(timeout.to_f, GrpcKit::Errors::DeadlineExceeded) do
              call.send_msg(request, timeout: timeout.to_s, last: true)
              call.recv(last: true)
            end
          end
        elsif @config.interceptor && !timeout
          @config.interceptor.intercept(request, call, call.metadata) do |r, c, _|
            call.send_msg(request, last: true)
            call.recv(last: true)
          end
        elsif !@config.interceptor && timeout
          Timeout.timeout(timeout.to_f, GrpcKit::Errors::DeadlineExceeded) do
            call.send_msg(request, timeout: timeout.to_s, last: true)
            call.recv(last: true)
          end
        else
          call.send_msg(request, last: true)
          call.recv(last: true)
        end
      end
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
grpc_kit-0.1.8	lib/grpc_kit/rpcs/client_request_response.rb
grpc_kit-0.1.7	lib/grpc_kit/rpcs/client_request_response.rb
grpc_kit-0.1.6	lib/grpc_kit/rpcs/client_request_response.rb
grpc_kit-0.1.5	lib/grpc_kit/rpcs/client_request_response.rb