|
|
|
@ -26,7 +26,6 @@ |
|
|
|
|
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
|
|
|
|
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
|
|
|
|
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
|
|
|
|
|
|
|
|
|
"""Translates gRPC's client-side API into gRPC's client-side Beta API.""" |
|
|
|
|
|
|
|
|
|
import grpc |
|
|
|
@ -38,14 +37,14 @@ from grpc.framework.foundation import future |
|
|
|
|
from grpc.framework.interfaces.face import face |
|
|
|
|
|
|
|
|
|
_STATUS_CODE_TO_ABORTION_KIND_AND_ABORTION_ERROR_CLASS = { |
|
|
|
|
grpc.StatusCode.CANCELLED: ( |
|
|
|
|
face.Abortion.Kind.CANCELLED, face.CancellationError), |
|
|
|
|
grpc.StatusCode.UNKNOWN: ( |
|
|
|
|
face.Abortion.Kind.REMOTE_FAILURE, face.RemoteError), |
|
|
|
|
grpc.StatusCode.DEADLINE_EXCEEDED: ( |
|
|
|
|
face.Abortion.Kind.EXPIRED, face.ExpirationError), |
|
|
|
|
grpc.StatusCode.UNIMPLEMENTED: ( |
|
|
|
|
face.Abortion.Kind.LOCAL_FAILURE, face.LocalError), |
|
|
|
|
grpc.StatusCode.CANCELLED: (face.Abortion.Kind.CANCELLED, |
|
|
|
|
face.CancellationError), |
|
|
|
|
grpc.StatusCode.UNKNOWN: (face.Abortion.Kind.REMOTE_FAILURE, |
|
|
|
|
face.RemoteError), |
|
|
|
|
grpc.StatusCode.DEADLINE_EXCEEDED: (face.Abortion.Kind.EXPIRED, |
|
|
|
|
face.ExpirationError), |
|
|
|
|
grpc.StatusCode.UNIMPLEMENTED: (face.Abortion.Kind.LOCAL_FAILURE, |
|
|
|
|
face.LocalError), |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
@ -65,18 +64,19 @@ def _abortion(rpc_error_call): |
|
|
|
|
code = rpc_error_call.code() |
|
|
|
|
pair = _STATUS_CODE_TO_ABORTION_KIND_AND_ABORTION_ERROR_CLASS.get(code) |
|
|
|
|
error_kind = face.Abortion.Kind.LOCAL_FAILURE if pair is None else pair[0] |
|
|
|
|
return face.Abortion( |
|
|
|
|
error_kind, rpc_error_call.initial_metadata(), |
|
|
|
|
rpc_error_call.trailing_metadata(), code, rpc_error_call.details()) |
|
|
|
|
return face.Abortion(error_kind, |
|
|
|
|
rpc_error_call.initial_metadata(), |
|
|
|
|
rpc_error_call.trailing_metadata(), code, |
|
|
|
|
rpc_error_call.details()) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _abortion_error(rpc_error_call): |
|
|
|
|
code = rpc_error_call.code() |
|
|
|
|
pair = _STATUS_CODE_TO_ABORTION_KIND_AND_ABORTION_ERROR_CLASS.get(code) |
|
|
|
|
exception_class = face.AbortionError if pair is None else pair[1] |
|
|
|
|
return exception_class( |
|
|
|
|
rpc_error_call.initial_metadata(), rpc_error_call.trailing_metadata(), |
|
|
|
|
code, rpc_error_call.details()) |
|
|
|
|
return exception_class(rpc_error_call.initial_metadata(), |
|
|
|
|
rpc_error_call.trailing_metadata(), code, |
|
|
|
|
rpc_error_call.details()) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class _InvocationProtocolContext(interfaces.GRPCInvocationContext): |
|
|
|
@ -159,9 +159,11 @@ class _Rendezvous(future.Future, face.Call): |
|
|
|
|
return self._call.time_remaining() |
|
|
|
|
|
|
|
|
|
def add_abortion_callback(self, abortion_callback): |
|
|
|
|
|
|
|
|
|
def done_callback(): |
|
|
|
|
if self.code() is not grpc.StatusCode.OK: |
|
|
|
|
abortion_callback(_abortion(self._call)) |
|
|
|
|
|
|
|
|
|
registered = self._call.add_callback(done_callback) |
|
|
|
|
return None if registered else done_callback() |
|
|
|
|
|
|
|
|
@ -181,9 +183,9 @@ class _Rendezvous(future.Future, face.Call): |
|
|
|
|
return self._call.details() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _blocking_unary_unary( |
|
|
|
|
channel, group, method, timeout, with_call, protocol_options, metadata, |
|
|
|
|
metadata_transformer, request, request_serializer, response_deserializer): |
|
|
|
|
def _blocking_unary_unary(channel, group, method, timeout, with_call, |
|
|
|
|
protocol_options, metadata, metadata_transformer, |
|
|
|
|
request, request_serializer, response_deserializer): |
|
|
|
|
try: |
|
|
|
|
multi_callable = channel.unary_unary( |
|
|
|
|
_common.fully_qualified_method(group, method), |
|
|
|
@ -192,48 +194,56 @@ def _blocking_unary_unary( |
|
|
|
|
effective_metadata = _effective_metadata(metadata, metadata_transformer) |
|
|
|
|
if with_call: |
|
|
|
|
response, call = multi_callable.with_call( |
|
|
|
|
request, timeout=timeout, metadata=effective_metadata, |
|
|
|
|
request, |
|
|
|
|
timeout=timeout, |
|
|
|
|
metadata=effective_metadata, |
|
|
|
|
credentials=_credentials(protocol_options)) |
|
|
|
|
return response, _Rendezvous(None, None, call) |
|
|
|
|
else: |
|
|
|
|
return multi_callable( |
|
|
|
|
request, timeout=timeout, metadata=effective_metadata, |
|
|
|
|
request, |
|
|
|
|
timeout=timeout, |
|
|
|
|
metadata=effective_metadata, |
|
|
|
|
credentials=_credentials(protocol_options)) |
|
|
|
|
except grpc.RpcError as rpc_error_call: |
|
|
|
|
raise _abortion_error(rpc_error_call) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _future_unary_unary( |
|
|
|
|
channel, group, method, timeout, protocol_options, metadata, |
|
|
|
|
metadata_transformer, request, request_serializer, response_deserializer): |
|
|
|
|
def _future_unary_unary(channel, group, method, timeout, protocol_options, |
|
|
|
|
metadata, metadata_transformer, request, |
|
|
|
|
request_serializer, response_deserializer): |
|
|
|
|
multi_callable = channel.unary_unary( |
|
|
|
|
_common.fully_qualified_method(group, method), |
|
|
|
|
request_serializer=request_serializer, |
|
|
|
|
response_deserializer=response_deserializer) |
|
|
|
|
effective_metadata = _effective_metadata(metadata, metadata_transformer) |
|
|
|
|
response_future = multi_callable.future( |
|
|
|
|
request, timeout=timeout, metadata=effective_metadata, |
|
|
|
|
request, |
|
|
|
|
timeout=timeout, |
|
|
|
|
metadata=effective_metadata, |
|
|
|
|
credentials=_credentials(protocol_options)) |
|
|
|
|
return _Rendezvous(response_future, None, response_future) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _unary_stream( |
|
|
|
|
channel, group, method, timeout, protocol_options, metadata, |
|
|
|
|
metadata_transformer, request, request_serializer, response_deserializer): |
|
|
|
|
def _unary_stream(channel, group, method, timeout, protocol_options, metadata, |
|
|
|
|
metadata_transformer, request, request_serializer, |
|
|
|
|
response_deserializer): |
|
|
|
|
multi_callable = channel.unary_stream( |
|
|
|
|
_common.fully_qualified_method(group, method), |
|
|
|
|
request_serializer=request_serializer, |
|
|
|
|
response_deserializer=response_deserializer) |
|
|
|
|
effective_metadata = _effective_metadata(metadata, metadata_transformer) |
|
|
|
|
response_iterator = multi_callable( |
|
|
|
|
request, timeout=timeout, metadata=effective_metadata, |
|
|
|
|
request, |
|
|
|
|
timeout=timeout, |
|
|
|
|
metadata=effective_metadata, |
|
|
|
|
credentials=_credentials(protocol_options)) |
|
|
|
|
return _Rendezvous(None, response_iterator, response_iterator) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _blocking_stream_unary( |
|
|
|
|
channel, group, method, timeout, with_call, protocol_options, metadata, |
|
|
|
|
metadata_transformer, request_iterator, request_serializer, |
|
|
|
|
def _blocking_stream_unary(channel, group, method, timeout, with_call, |
|
|
|
|
protocol_options, metadata, metadata_transformer, |
|
|
|
|
request_iterator, request_serializer, |
|
|
|
|
response_deserializer): |
|
|
|
|
try: |
|
|
|
|
multi_callable = channel.stream_unary( |
|
|
|
@ -243,34 +253,38 @@ def _blocking_stream_unary( |
|
|
|
|
effective_metadata = _effective_metadata(metadata, metadata_transformer) |
|
|
|
|
if with_call: |
|
|
|
|
response, call = multi_callable.with_call( |
|
|
|
|
request_iterator, timeout=timeout, metadata=effective_metadata, |
|
|
|
|
request_iterator, |
|
|
|
|
timeout=timeout, |
|
|
|
|
metadata=effective_metadata, |
|
|
|
|
credentials=_credentials(protocol_options)) |
|
|
|
|
return response, _Rendezvous(None, None, call) |
|
|
|
|
else: |
|
|
|
|
return multi_callable( |
|
|
|
|
request_iterator, timeout=timeout, metadata=effective_metadata, |
|
|
|
|
request_iterator, |
|
|
|
|
timeout=timeout, |
|
|
|
|
metadata=effective_metadata, |
|
|
|
|
credentials=_credentials(protocol_options)) |
|
|
|
|
except grpc.RpcError as rpc_error_call: |
|
|
|
|
raise _abortion_error(rpc_error_call) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _future_stream_unary( |
|
|
|
|
channel, group, method, timeout, protocol_options, metadata, |
|
|
|
|
metadata_transformer, request_iterator, request_serializer, |
|
|
|
|
response_deserializer): |
|
|
|
|
def _future_stream_unary(channel, group, method, timeout, protocol_options, |
|
|
|
|
metadata, metadata_transformer, request_iterator, |
|
|
|
|
request_serializer, response_deserializer): |
|
|
|
|
multi_callable = channel.stream_unary( |
|
|
|
|
_common.fully_qualified_method(group, method), |
|
|
|
|
request_serializer=request_serializer, |
|
|
|
|
response_deserializer=response_deserializer) |
|
|
|
|
effective_metadata = _effective_metadata(metadata, metadata_transformer) |
|
|
|
|
response_future = multi_callable.future( |
|
|
|
|
request_iterator, timeout=timeout, metadata=effective_metadata, |
|
|
|
|
request_iterator, |
|
|
|
|
timeout=timeout, |
|
|
|
|
metadata=effective_metadata, |
|
|
|
|
credentials=_credentials(protocol_options)) |
|
|
|
|
return _Rendezvous(response_future, None, response_future) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _stream_stream( |
|
|
|
|
channel, group, method, timeout, protocol_options, metadata, |
|
|
|
|
def _stream_stream(channel, group, method, timeout, protocol_options, metadata, |
|
|
|
|
metadata_transformer, request_iterator, request_serializer, |
|
|
|
|
response_deserializer): |
|
|
|
|
multi_callable = channel.stream_stream( |
|
|
|
@ -279,16 +293,17 @@ def _stream_stream( |
|
|
|
|
response_deserializer=response_deserializer) |
|
|
|
|
effective_metadata = _effective_metadata(metadata, metadata_transformer) |
|
|
|
|
response_iterator = multi_callable( |
|
|
|
|
request_iterator, timeout=timeout, metadata=effective_metadata, |
|
|
|
|
request_iterator, |
|
|
|
|
timeout=timeout, |
|
|
|
|
metadata=effective_metadata, |
|
|
|
|
credentials=_credentials(protocol_options)) |
|
|
|
|
return _Rendezvous(None, response_iterator, response_iterator) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class _UnaryUnaryMultiCallable(face.UnaryUnaryMultiCallable): |
|
|
|
|
|
|
|
|
|
def __init__( |
|
|
|
|
self, channel, group, method, metadata_transformer, request_serializer, |
|
|
|
|
response_deserializer): |
|
|
|
|
def __init__(self, channel, group, method, metadata_transformer, |
|
|
|
|
request_serializer, response_deserializer): |
|
|
|
|
self._channel = channel |
|
|
|
|
self._group = group |
|
|
|
|
self._method = method |
|
|
|
@ -296,8 +311,11 @@ class _UnaryUnaryMultiCallable(face.UnaryUnaryMultiCallable): |
|
|
|
|
self._request_serializer = request_serializer |
|
|
|
|
self._response_deserializer = response_deserializer |
|
|
|
|
|
|
|
|
|
def __call__( |
|
|
|
|
self, request, timeout, metadata=None, with_call=False, |
|
|
|
|
def __call__(self, |
|
|
|
|
request, |
|
|
|
|
timeout, |
|
|
|
|
metadata=None, |
|
|
|
|
with_call=False, |
|
|
|
|
protocol_options=None): |
|
|
|
|
return _blocking_unary_unary( |
|
|
|
|
self._channel, self._group, self._method, timeout, with_call, |
|
|
|
@ -307,20 +325,23 @@ class _UnaryUnaryMultiCallable(face.UnaryUnaryMultiCallable): |
|
|
|
|
def future(self, request, timeout, metadata=None, protocol_options=None): |
|
|
|
|
return _future_unary_unary( |
|
|
|
|
self._channel, self._group, self._method, timeout, protocol_options, |
|
|
|
|
metadata, self._metadata_transformer, request, self._request_serializer, |
|
|
|
|
self._response_deserializer) |
|
|
|
|
metadata, self._metadata_transformer, request, |
|
|
|
|
self._request_serializer, self._response_deserializer) |
|
|
|
|
|
|
|
|
|
def event( |
|
|
|
|
self, request, receiver, abortion_callback, timeout, |
|
|
|
|
metadata=None, protocol_options=None): |
|
|
|
|
def event(self, |
|
|
|
|
request, |
|
|
|
|
receiver, |
|
|
|
|
abortion_callback, |
|
|
|
|
timeout, |
|
|
|
|
metadata=None, |
|
|
|
|
protocol_options=None): |
|
|
|
|
raise NotImplementedError() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class _UnaryStreamMultiCallable(face.UnaryStreamMultiCallable): |
|
|
|
|
|
|
|
|
|
def __init__( |
|
|
|
|
self, channel, group, method, metadata_transformer, request_serializer, |
|
|
|
|
response_deserializer): |
|
|
|
|
def __init__(self, channel, group, method, metadata_transformer, |
|
|
|
|
request_serializer, response_deserializer): |
|
|
|
|
self._channel = channel |
|
|
|
|
self._group = group |
|
|
|
|
self._method = method |
|
|
|
@ -331,20 +352,23 @@ class _UnaryStreamMultiCallable(face.UnaryStreamMultiCallable): |
|
|
|
|
def __call__(self, request, timeout, metadata=None, protocol_options=None): |
|
|
|
|
return _unary_stream( |
|
|
|
|
self._channel, self._group, self._method, timeout, protocol_options, |
|
|
|
|
metadata, self._metadata_transformer, request, self._request_serializer, |
|
|
|
|
self._response_deserializer) |
|
|
|
|
metadata, self._metadata_transformer, request, |
|
|
|
|
self._request_serializer, self._response_deserializer) |
|
|
|
|
|
|
|
|
|
def event( |
|
|
|
|
self, request, receiver, abortion_callback, timeout, |
|
|
|
|
metadata=None, protocol_options=None): |
|
|
|
|
def event(self, |
|
|
|
|
request, |
|
|
|
|
receiver, |
|
|
|
|
abortion_callback, |
|
|
|
|
timeout, |
|
|
|
|
metadata=None, |
|
|
|
|
protocol_options=None): |
|
|
|
|
raise NotImplementedError() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class _StreamUnaryMultiCallable(face.StreamUnaryMultiCallable): |
|
|
|
|
|
|
|
|
|
def __init__( |
|
|
|
|
self, channel, group, method, metadata_transformer, request_serializer, |
|
|
|
|
response_deserializer): |
|
|
|
|
def __init__(self, channel, group, method, metadata_transformer, |
|
|
|
|
request_serializer, response_deserializer): |
|
|
|
|
self._channel = channel |
|
|
|
|
self._group = group |
|
|
|
|
self._method = method |
|
|
|
@ -352,32 +376,41 @@ class _StreamUnaryMultiCallable(face.StreamUnaryMultiCallable): |
|
|
|
|
self._request_serializer = request_serializer |
|
|
|
|
self._response_deserializer = response_deserializer |
|
|
|
|
|
|
|
|
|
def __call__( |
|
|
|
|
self, request_iterator, timeout, metadata=None, with_call=False, |
|
|
|
|
def __call__(self, |
|
|
|
|
request_iterator, |
|
|
|
|
timeout, |
|
|
|
|
metadata=None, |
|
|
|
|
with_call=False, |
|
|
|
|
protocol_options=None): |
|
|
|
|
return _blocking_stream_unary( |
|
|
|
|
self._channel, self._group, self._method, timeout, with_call, |
|
|
|
|
protocol_options, metadata, self._metadata_transformer, |
|
|
|
|
request_iterator, self._request_serializer, self._response_deserializer) |
|
|
|
|
request_iterator, self._request_serializer, |
|
|
|
|
self._response_deserializer) |
|
|
|
|
|
|
|
|
|
def future( |
|
|
|
|
self, request_iterator, timeout, metadata=None, protocol_options=None): |
|
|
|
|
def future(self, |
|
|
|
|
request_iterator, |
|
|
|
|
timeout, |
|
|
|
|
metadata=None, |
|
|
|
|
protocol_options=None): |
|
|
|
|
return _future_stream_unary( |
|
|
|
|
self._channel, self._group, self._method, timeout, protocol_options, |
|
|
|
|
metadata, self._metadata_transformer, request_iterator, |
|
|
|
|
self._request_serializer, self._response_deserializer) |
|
|
|
|
|
|
|
|
|
def event( |
|
|
|
|
self, receiver, abortion_callback, timeout, metadata=None, |
|
|
|
|
def event(self, |
|
|
|
|
receiver, |
|
|
|
|
abortion_callback, |
|
|
|
|
timeout, |
|
|
|
|
metadata=None, |
|
|
|
|
protocol_options=None): |
|
|
|
|
raise NotImplementedError() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class _StreamStreamMultiCallable(face.StreamStreamMultiCallable): |
|
|
|
|
|
|
|
|
|
def __init__( |
|
|
|
|
self, channel, group, method, metadata_transformer, request_serializer, |
|
|
|
|
response_deserializer): |
|
|
|
|
def __init__(self, channel, group, method, metadata_transformer, |
|
|
|
|
request_serializer, response_deserializer): |
|
|
|
|
self._channel = channel |
|
|
|
|
self._group = group |
|
|
|
|
self._method = method |
|
|
|
@ -385,133 +418,226 @@ class _StreamStreamMultiCallable(face.StreamStreamMultiCallable): |
|
|
|
|
self._request_serializer = request_serializer |
|
|
|
|
self._response_deserializer = response_deserializer |
|
|
|
|
|
|
|
|
|
def __call__( |
|
|
|
|
self, request_iterator, timeout, metadata=None, protocol_options=None): |
|
|
|
|
def __call__(self, |
|
|
|
|
request_iterator, |
|
|
|
|
timeout, |
|
|
|
|
metadata=None, |
|
|
|
|
protocol_options=None): |
|
|
|
|
return _stream_stream( |
|
|
|
|
self._channel, self._group, self._method, timeout, protocol_options, |
|
|
|
|
metadata, self._metadata_transformer, request_iterator, |
|
|
|
|
self._request_serializer, self._response_deserializer) |
|
|
|
|
|
|
|
|
|
def event( |
|
|
|
|
self, receiver, abortion_callback, timeout, metadata=None, |
|
|
|
|
def event(self, |
|
|
|
|
receiver, |
|
|
|
|
abortion_callback, |
|
|
|
|
timeout, |
|
|
|
|
metadata=None, |
|
|
|
|
protocol_options=None): |
|
|
|
|
raise NotImplementedError() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class _GenericStub(face.GenericStub): |
|
|
|
|
|
|
|
|
|
def __init__( |
|
|
|
|
self, channel, metadata_transformer, request_serializers, |
|
|
|
|
def __init__(self, channel, metadata_transformer, request_serializers, |
|
|
|
|
response_deserializers): |
|
|
|
|
self._channel = channel |
|
|
|
|
self._metadata_transformer = metadata_transformer |
|
|
|
|
self._request_serializers = request_serializers or {} |
|
|
|
|
self._response_deserializers = response_deserializers or {} |
|
|
|
|
|
|
|
|
|
def blocking_unary_unary( |
|
|
|
|
self, group, method, request, timeout, metadata=None, |
|
|
|
|
with_call=None, protocol_options=None): |
|
|
|
|
request_serializer = self._request_serializers.get((group, method,)) |
|
|
|
|
response_deserializer = self._response_deserializers.get((group, method,)) |
|
|
|
|
return _blocking_unary_unary( |
|
|
|
|
self._channel, group, method, timeout, with_call, protocol_options, |
|
|
|
|
metadata, self._metadata_transformer, request, request_serializer, |
|
|
|
|
response_deserializer) |
|
|
|
|
def blocking_unary_unary(self, |
|
|
|
|
group, |
|
|
|
|
method, |
|
|
|
|
request, |
|
|
|
|
timeout, |
|
|
|
|
metadata=None, |
|
|
|
|
with_call=None, |
|
|
|
|
protocol_options=None): |
|
|
|
|
request_serializer = self._request_serializers.get(( |
|
|
|
|
group, |
|
|
|
|
method,)) |
|
|
|
|
response_deserializer = self._response_deserializers.get(( |
|
|
|
|
group, |
|
|
|
|
method,)) |
|
|
|
|
return _blocking_unary_unary(self._channel, group, method, timeout, |
|
|
|
|
with_call, protocol_options, metadata, |
|
|
|
|
self._metadata_transformer, request, |
|
|
|
|
request_serializer, response_deserializer) |
|
|
|
|
|
|
|
|
|
def future_unary_unary( |
|
|
|
|
self, group, method, request, timeout, metadata=None, |
|
|
|
|
def future_unary_unary(self, |
|
|
|
|
group, |
|
|
|
|
method, |
|
|
|
|
request, |
|
|
|
|
timeout, |
|
|
|
|
metadata=None, |
|
|
|
|
protocol_options=None): |
|
|
|
|
request_serializer = self._request_serializers.get((group, method,)) |
|
|
|
|
response_deserializer = self._response_deserializers.get((group, method,)) |
|
|
|
|
return _future_unary_unary( |
|
|
|
|
self._channel, group, method, timeout, protocol_options, metadata, |
|
|
|
|
self._metadata_transformer, request, request_serializer, |
|
|
|
|
response_deserializer) |
|
|
|
|
request_serializer = self._request_serializers.get(( |
|
|
|
|
group, |
|
|
|
|
method,)) |
|
|
|
|
response_deserializer = self._response_deserializers.get(( |
|
|
|
|
group, |
|
|
|
|
method,)) |
|
|
|
|
return _future_unary_unary(self._channel, group, method, timeout, |
|
|
|
|
protocol_options, metadata, |
|
|
|
|
self._metadata_transformer, request, |
|
|
|
|
request_serializer, response_deserializer) |
|
|
|
|
|
|
|
|
|
def inline_unary_stream( |
|
|
|
|
self, group, method, request, timeout, metadata=None, |
|
|
|
|
def inline_unary_stream(self, |
|
|
|
|
group, |
|
|
|
|
method, |
|
|
|
|
request, |
|
|
|
|
timeout, |
|
|
|
|
metadata=None, |
|
|
|
|
protocol_options=None): |
|
|
|
|
request_serializer = self._request_serializers.get((group, method,)) |
|
|
|
|
response_deserializer = self._response_deserializers.get((group, method,)) |
|
|
|
|
return _unary_stream( |
|
|
|
|
self._channel, group, method, timeout, protocol_options, metadata, |
|
|
|
|
self._metadata_transformer, request, request_serializer, |
|
|
|
|
response_deserializer) |
|
|
|
|
request_serializer = self._request_serializers.get(( |
|
|
|
|
group, |
|
|
|
|
method,)) |
|
|
|
|
response_deserializer = self._response_deserializers.get(( |
|
|
|
|
group, |
|
|
|
|
method,)) |
|
|
|
|
return _unary_stream(self._channel, group, method, timeout, |
|
|
|
|
protocol_options, metadata, |
|
|
|
|
self._metadata_transformer, request, |
|
|
|
|
request_serializer, response_deserializer) |
|
|
|
|
|
|
|
|
|
def blocking_stream_unary( |
|
|
|
|
self, group, method, request_iterator, timeout, metadata=None, |
|
|
|
|
with_call=None, protocol_options=None): |
|
|
|
|
request_serializer = self._request_serializers.get((group, method,)) |
|
|
|
|
response_deserializer = self._response_deserializers.get((group, method,)) |
|
|
|
|
def blocking_stream_unary(self, |
|
|
|
|
group, |
|
|
|
|
method, |
|
|
|
|
request_iterator, |
|
|
|
|
timeout, |
|
|
|
|
metadata=None, |
|
|
|
|
with_call=None, |
|
|
|
|
protocol_options=None): |
|
|
|
|
request_serializer = self._request_serializers.get(( |
|
|
|
|
group, |
|
|
|
|
method,)) |
|
|
|
|
response_deserializer = self._response_deserializers.get(( |
|
|
|
|
group, |
|
|
|
|
method,)) |
|
|
|
|
return _blocking_stream_unary( |
|
|
|
|
self._channel, group, method, timeout, with_call, protocol_options, |
|
|
|
|
metadata, self._metadata_transformer, request_iterator, |
|
|
|
|
request_serializer, response_deserializer) |
|
|
|
|
|
|
|
|
|
def future_stream_unary( |
|
|
|
|
self, group, method, request_iterator, timeout, metadata=None, |
|
|
|
|
def future_stream_unary(self, |
|
|
|
|
group, |
|
|
|
|
method, |
|
|
|
|
request_iterator, |
|
|
|
|
timeout, |
|
|
|
|
metadata=None, |
|
|
|
|
protocol_options=None): |
|
|
|
|
request_serializer = self._request_serializers.get((group, method,)) |
|
|
|
|
response_deserializer = self._response_deserializers.get((group, method,)) |
|
|
|
|
request_serializer = self._request_serializers.get(( |
|
|
|
|
group, |
|
|
|
|
method,)) |
|
|
|
|
response_deserializer = self._response_deserializers.get(( |
|
|
|
|
group, |
|
|
|
|
method,)) |
|
|
|
|
return _future_stream_unary( |
|
|
|
|
self._channel, group, method, timeout, protocol_options, metadata, |
|
|
|
|
self._metadata_transformer, request_iterator, request_serializer, |
|
|
|
|
response_deserializer) |
|
|
|
|
|
|
|
|
|
def inline_stream_stream( |
|
|
|
|
self, group, method, request_iterator, timeout, metadata=None, |
|
|
|
|
def inline_stream_stream(self, |
|
|
|
|
group, |
|
|
|
|
method, |
|
|
|
|
request_iterator, |
|
|
|
|
timeout, |
|
|
|
|
metadata=None, |
|
|
|
|
protocol_options=None): |
|
|
|
|
request_serializer = self._request_serializers.get((group, method,)) |
|
|
|
|
response_deserializer = self._response_deserializers.get((group, method,)) |
|
|
|
|
return _stream_stream( |
|
|
|
|
self._channel, group, method, timeout, protocol_options, metadata, |
|
|
|
|
self._metadata_transformer, request_iterator, request_serializer, |
|
|
|
|
response_deserializer) |
|
|
|
|
request_serializer = self._request_serializers.get(( |
|
|
|
|
group, |
|
|
|
|
method,)) |
|
|
|
|
response_deserializer = self._response_deserializers.get(( |
|
|
|
|
group, |
|
|
|
|
method,)) |
|
|
|
|
return _stream_stream(self._channel, group, method, timeout, |
|
|
|
|
protocol_options, metadata, |
|
|
|
|
self._metadata_transformer, request_iterator, |
|
|
|
|
request_serializer, response_deserializer) |
|
|
|
|
|
|
|
|
|
def event_unary_unary( |
|
|
|
|
self, group, method, request, receiver, abortion_callback, timeout, |
|
|
|
|
metadata=None, protocol_options=None): |
|
|
|
|
def event_unary_unary(self, |
|
|
|
|
group, |
|
|
|
|
method, |
|
|
|
|
request, |
|
|
|
|
receiver, |
|
|
|
|
abortion_callback, |
|
|
|
|
timeout, |
|
|
|
|
metadata=None, |
|
|
|
|
protocol_options=None): |
|
|
|
|
raise NotImplementedError() |
|
|
|
|
|
|
|
|
|
def event_unary_stream( |
|
|
|
|
self, group, method, request, receiver, abortion_callback, timeout, |
|
|
|
|
metadata=None, protocol_options=None): |
|
|
|
|
def event_unary_stream(self, |
|
|
|
|
group, |
|
|
|
|
method, |
|
|
|
|
request, |
|
|
|
|
receiver, |
|
|
|
|
abortion_callback, |
|
|
|
|
timeout, |
|
|
|
|
metadata=None, |
|
|
|
|
protocol_options=None): |
|
|
|
|
raise NotImplementedError() |
|
|
|
|
|
|
|
|
|
def event_stream_unary( |
|
|
|
|
self, group, method, receiver, abortion_callback, timeout, |
|
|
|
|
metadata=None, protocol_options=None): |
|
|
|
|
def event_stream_unary(self, |
|
|
|
|
group, |
|
|
|
|
method, |
|
|
|
|
receiver, |
|
|
|
|
abortion_callback, |
|
|
|
|
timeout, |
|
|
|
|
metadata=None, |
|
|
|
|
protocol_options=None): |
|
|
|
|
raise NotImplementedError() |
|
|
|
|
|
|
|
|
|
def event_stream_stream( |
|
|
|
|
self, group, method, receiver, abortion_callback, timeout, |
|
|
|
|
metadata=None, protocol_options=None): |
|
|
|
|
def event_stream_stream(self, |
|
|
|
|
group, |
|
|
|
|
method, |
|
|
|
|
receiver, |
|
|
|
|
abortion_callback, |
|
|
|
|
timeout, |
|
|
|
|
metadata=None, |
|
|
|
|
protocol_options=None): |
|
|
|
|
raise NotImplementedError() |
|
|
|
|
|
|
|
|
|
def unary_unary(self, group, method): |
|
|
|
|
request_serializer = self._request_serializers.get((group, method,)) |
|
|
|
|
response_deserializer = self._response_deserializers.get((group, method,)) |
|
|
|
|
request_serializer = self._request_serializers.get(( |
|
|
|
|
group, |
|
|
|
|
method,)) |
|
|
|
|
response_deserializer = self._response_deserializers.get(( |
|
|
|
|
group, |
|
|
|
|
method,)) |
|
|
|
|
return _UnaryUnaryMultiCallable( |
|
|
|
|
self._channel, group, method, self._metadata_transformer, |
|
|
|
|
request_serializer, response_deserializer) |
|
|
|
|
|
|
|
|
|
def unary_stream(self, group, method): |
|
|
|
|
request_serializer = self._request_serializers.get((group, method,)) |
|
|
|
|
response_deserializer = self._response_deserializers.get((group, method,)) |
|
|
|
|
request_serializer = self._request_serializers.get(( |
|
|
|
|
group, |
|
|
|
|
method,)) |
|
|
|
|
response_deserializer = self._response_deserializers.get(( |
|
|
|
|
group, |
|
|
|
|
method,)) |
|
|
|
|
return _UnaryStreamMultiCallable( |
|
|
|
|
self._channel, group, method, self._metadata_transformer, |
|
|
|
|
request_serializer, response_deserializer) |
|
|
|
|
|
|
|
|
|
def stream_unary(self, group, method): |
|
|
|
|
request_serializer = self._request_serializers.get((group, method,)) |
|
|
|
|
response_deserializer = self._response_deserializers.get((group, method,)) |
|
|
|
|
request_serializer = self._request_serializers.get(( |
|
|
|
|
group, |
|
|
|
|
method,)) |
|
|
|
|
response_deserializer = self._response_deserializers.get(( |
|
|
|
|
group, |
|
|
|
|
method,)) |
|
|
|
|
return _StreamUnaryMultiCallable( |
|
|
|
|
self._channel, group, method, self._metadata_transformer, |
|
|
|
|
request_serializer, response_deserializer) |
|
|
|
|
|
|
|
|
|
def stream_stream(self, group, method): |
|
|
|
|
request_serializer = self._request_serializers.get((group, method,)) |
|
|
|
|
response_deserializer = self._response_deserializers.get((group, method,)) |
|
|
|
|
request_serializer = self._request_serializers.get(( |
|
|
|
|
group, |
|
|
|
|
method,)) |
|
|
|
|
response_deserializer = self._response_deserializers.get(( |
|
|
|
|
group, |
|
|
|
|
method,)) |
|
|
|
|
return _StreamStreamMultiCallable( |
|
|
|
|
self._channel, group, method, self._metadata_transformer, |
|
|
|
|
request_serializer, response_deserializer) |
|
|
|
@ -541,7 +667,8 @@ class _DynamicStub(face.DynamicStub): |
|
|
|
|
elif method_cardinality is cardinality.Cardinality.STREAM_STREAM: |
|
|
|
|
return self._generic_stub.stream_stream(self._group, attr) |
|
|
|
|
else: |
|
|
|
|
raise AttributeError('_DynamicStub object has no attribute "%s"!' % attr) |
|
|
|
|
raise AttributeError('_DynamicStub object has no attribute "%s"!' % |
|
|
|
|
attr) |
|
|
|
|
|
|
|
|
|
def __enter__(self): |
|
|
|
|
return self |
|
|
|
@ -550,19 +677,14 @@ class _DynamicStub(face.DynamicStub): |
|
|
|
|
return False |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def generic_stub( |
|
|
|
|
channel, host, metadata_transformer, request_serializers, |
|
|
|
|
def generic_stub(channel, host, metadata_transformer, request_serializers, |
|
|
|
|
response_deserializers): |
|
|
|
|
return _GenericStub( |
|
|
|
|
channel, metadata_transformer, request_serializers, |
|
|
|
|
return _GenericStub(channel, metadata_transformer, request_serializers, |
|
|
|
|
response_deserializers) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def dynamic_stub( |
|
|
|
|
channel, service, cardinalities, host, metadata_transformer, |
|
|
|
|
def dynamic_stub(channel, service, cardinalities, host, metadata_transformer, |
|
|
|
|
request_serializers, response_deserializers): |
|
|
|
|
return _DynamicStub( |
|
|
|
|
_GenericStub( |
|
|
|
|
channel, metadata_transformer, request_serializers, |
|
|
|
|
response_deserializers), |
|
|
|
|
service, cardinalities) |
|
|
|
|
_GenericStub(channel, metadata_transformer, request_serializers, |
|
|
|
|
response_deserializers), service, cardinalities) |
|
|
|
|