Class: Couchbase::Protostellar::Client Private

Inherits:
Object
  • Object
show all
Defined in:
lib/couchbase/protostellar/client.rb

Overview

This class is part of a private API. You should avoid using this class if possible, as it may be removed or be changed in the future.

API:

  • private

Instance Method Summary collapse

Constructor Details

#initialize(host:, credentials:, channel_args:, call_metadata:, timeouts:) ⇒ Client

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

Returns a new instance of Client.

API:

  • private



34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
 
# File 'lib/couchbase/protostellar/client.rb', line 34

def initialize(host:, credentials:, channel_args:, call_metadata:, timeouts:)
  @channel = GRPC::Core::Channel.new(host, channel_args, credentials)
  @call_metadata = call_metadata
  @timeouts = timeouts

  @stubs = {
    routing: Generated::Routing::V1::RoutingService::Stub.new(host, credentials, channel_override: @channel),
    kv: Generated::KV::V1::KvService::Stub.new(host, credentials, channel_override: @channel),
    query: Generated::Query::V1::QueryService::Stub.new(host, credentials, channel_override: @channel),
    search: Generated::Search::V1::SearchService::Stub.new(host, credentials, channel_override: @channel),
    analytics: Generated::Analytics::V1::AnalyticsService::Stub.new(host, credentials, channel_override: @channel),
    view: Generated::View::V1::ViewService::Stub.new(host, credentials, channel_override: @channel),
    bucket_admin: Generated::Admin::Bucket::V1::BucketAdminService::Stub.new(host, credentials, channel_override: @channel),
    collection_admin: Generated::Admin::Collection::V1::CollectionAdminService::Stub.new(host, credentials,
                                                                                         channel_override: @channel),
    query_admin: Generated::Admin::Query::V1::QueryAdminService::Stub.new(host, credentials, channel_override: @channel),
    search_admin: Generated::Admin::Search::V1::SearchAdminService::Stub.new(host, credentials, channel_override: @channel),
  }
end

Instance Method Details

#close ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

API:

  • private



54
55
56
 
# File 'lib/couchbase/protostellar/client.rb', line 54

def close
  @channel.close
end

#send_request(request) ⇒ Object

This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.

API:

  • private



58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
 
# File 'lib/couchbase/protostellar/client.rb', line 58

def send_request(request)
  request.set_timeout_from_defaults(@timeouts)
  loop do
    resp = @stubs[request.service].public_send(request.rpc, request.proto_request, deadline: request.deadline,
                                                                                   metadata: @call_metadata)
    return resp unless resp.respond_to?(:next)

    # Streaming RPC - wrap it in an enumerator that handles any mid-stream errors
    return Enumerator.new do |y|
      loop do
        y << resp.next
      rescue GRPC::BadStatus => e
        request_behaviour = ErrorHandling.handle_grpc_error(e, request)
        raise request_behaviour.error unless request_behaviour.error.nil?

        unless request_behaviour.retry_duration.nil?
          raise Couchbase::Error::RequestCanceled.new("Error encountered mid-stream",
                                                      request.error_context)
        end

        next
      end
    end

  # Simple RPC - just return it
  rescue GRPC::BadStatus => e
    request_behaviour = ErrorHandling.handle_grpc_error(e, request)

    unless request_behaviour.error.nil? ^ request_behaviour.retry_duration.nil?
      raise Couchbase::Error::CouchbaseError, "Either the error or the retry duration can be set"
    end
    raise request_behaviour.error unless request_behaviour.error.nil?

    unless request_behaviour.retry_duration.nil?
      sleep(0.001 * request_behaviour.retry_duration)
      next
    end
  end
end