|
@@ -9,13 +9,14 @@ import threading
|
|
import grpc
|
|
import grpc
|
|
from typing import Any, AnyStr, Callable, Iterator, Optional, Sequence, Tuple, TypeVar, Union
|
|
from typing import Any, AnyStr, Callable, Iterator, Optional, Sequence, Tuple, TypeVar, Union
|
|
|
|
|
|
-
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
_EVICTION_PERIOD_KEY = "GRPC_PYTHON_MANAGED_CHANNEL_EVICTION_SECONDS"
|
|
_EVICTION_PERIOD_KEY = "GRPC_PYTHON_MANAGED_CHANNEL_EVICTION_SECONDS"
|
|
if _EVICTION_PERIOD_KEY in os.environ:
|
|
if _EVICTION_PERIOD_KEY in os.environ:
|
|
- _EVICTION_PERIOD = datetime.timedelta(seconds=float(os.environ[_EVICTION_PERIOD_KEY]))
|
|
|
|
- _LOGGER.info(f"Setting managed channel eviction period to {_EVICTION_PERIOD}")
|
|
|
|
|
|
+ _EVICTION_PERIOD = datetime.timedelta(
|
|
|
|
+ seconds=float(os.environ[_EVICTION_PERIOD_KEY]))
|
|
|
|
+ _LOGGER.info(
|
|
|
|
+ f"Setting managed channel eviction period to {_EVICTION_PERIOD}")
|
|
else:
|
|
else:
|
|
_EVICTION_PERIOD = datetime.timedelta(minutes=10)
|
|
_EVICTION_PERIOD = datetime.timedelta(minutes=10)
|
|
|
|
|
|
@@ -24,39 +25,40 @@ if _MAXIMUM_CHANNELS_KEY in os.environ:
|
|
_MAXIMUM_CHANNELS = int(os.environ[_MAXIMUM_CHANNELS_KEY])
|
|
_MAXIMUM_CHANNELS = int(os.environ[_MAXIMUM_CHANNELS_KEY])
|
|
_LOGGER.info(f"Setting maximum managed channels to {_MAXIMUM_CHANNELS}")
|
|
_LOGGER.info(f"Setting maximum managed channels to {_MAXIMUM_CHANNELS}")
|
|
else:
|
|
else:
|
|
- _MAXIMUM_CHANNELS = 2 ** 8
|
|
|
|
|
|
+ _MAXIMUM_CHANNELS = 2**8
|
|
|
|
+
|
|
|
|
|
|
-def _create_channel(target: str,
|
|
|
|
- options: Sequence[Tuple[str, str]],
|
|
|
|
- channel_credentials: Optional[grpc.ChannelCredentials],
|
|
|
|
|
|
+def _create_channel(target: str, options: Sequence[Tuple[str, str]],
|
|
|
|
+ channel_credentials: grpc.ChannelCredentials,
|
|
compression: Optional[grpc.Compression]) -> grpc.Channel:
|
|
compression: Optional[grpc.Compression]) -> grpc.Channel:
|
|
- if channel_credentials is None:
|
|
|
|
|
|
+ if channel_credentials._credentials is grpc._insecure_channel_credentials:
|
|
_LOGGER.info(f"Creating insecure channel with options '{options}' " +
|
|
_LOGGER.info(f"Creating insecure channel with options '{options}' " +
|
|
- f"and compression '{compression}'")
|
|
|
|
|
|
+ f"and compression '{compression}'")
|
|
return grpc.insecure_channel(target,
|
|
return grpc.insecure_channel(target,
|
|
options=options,
|
|
options=options,
|
|
compression=compression)
|
|
compression=compression)
|
|
else:
|
|
else:
|
|
- _LOGGER.info(f"Creating secure channel with credentials '{channel_credentials}', " +
|
|
|
|
- f"options '{options}' and compression '{compression}'")
|
|
|
|
|
|
+ _LOGGER.info(
|
|
|
|
+ f"Creating secure channel with credentials '{channel_credentials}', "
|
|
|
|
+ + f"options '{options}' and compression '{compression}'")
|
|
return grpc.secure_channel(target,
|
|
return grpc.secure_channel(target,
|
|
credentials=channel_credentials,
|
|
credentials=channel_credentials,
|
|
options=options,
|
|
options=options,
|
|
compression=compression)
|
|
compression=compression)
|
|
|
|
|
|
|
|
+
|
|
class ChannelCache:
|
|
class ChannelCache:
|
|
_singleton = None
|
|
_singleton = None
|
|
_lock = threading.RLock()
|
|
_lock = threading.RLock()
|
|
_condition = threading.Condition(lock=_lock)
|
|
_condition = threading.Condition(lock=_lock)
|
|
_eviction_ready = threading.Event()
|
|
_eviction_ready = threading.Event()
|
|
|
|
|
|
-
|
|
|
|
def __init__(self):
|
|
def __init__(self):
|
|
self._mapping = collections.OrderedDict()
|
|
self._mapping = collections.OrderedDict()
|
|
- self._eviction_thread = threading.Thread(target=ChannelCache._perform_evictions, daemon=True)
|
|
|
|
|
|
+ self._eviction_thread = threading.Thread(
|
|
|
|
+ target=ChannelCache._perform_evictions, daemon=True)
|
|
self._eviction_thread.start()
|
|
self._eviction_thread.start()
|
|
|
|
|
|
-
|
|
|
|
@staticmethod
|
|
@staticmethod
|
|
def get():
|
|
def get():
|
|
with ChannelCache._lock:
|
|
with ChannelCache._lock:
|
|
@@ -72,7 +74,6 @@ class ChannelCache:
|
|
channel.close()
|
|
channel.close()
|
|
del channel
|
|
del channel
|
|
|
|
|
|
-
|
|
|
|
# TODO: Refactor. Way too deeply nested.
|
|
# TODO: Refactor. Way too deeply nested.
|
|
@staticmethod
|
|
@staticmethod
|
|
def _perform_evictions():
|
|
def _perform_evictions():
|
|
@@ -86,7 +87,8 @@ class ChannelCache:
|
|
ChannelCache._singleton._evict_locked(key)
|
|
ChannelCache._singleton._evict_locked(key)
|
|
# And immediately reevaluate.
|
|
# And immediately reevaluate.
|
|
else:
|
|
else:
|
|
- key, (channel, eviction_time) = next(iter(ChannelCache._singleton._mapping.items()))
|
|
|
|
|
|
+ key, (channel, eviction_time) = next(
|
|
|
|
+ iter(ChannelCache._singleton._mapping.items()))
|
|
now = datetime.datetime.now()
|
|
now = datetime.datetime.now()
|
|
if eviction_time <= now:
|
|
if eviction_time <= now:
|
|
ChannelCache._singleton._evict_locked(key)
|
|
ChannelCache._singleton._evict_locked(key)
|
|
@@ -95,12 +97,8 @@ class ChannelCache:
|
|
time_to_eviction = (eviction_time - now).total_seconds()
|
|
time_to_eviction = (eviction_time - now).total_seconds()
|
|
ChannelCache._condition.wait(timeout=time_to_eviction)
|
|
ChannelCache._condition.wait(timeout=time_to_eviction)
|
|
|
|
|
|
-
|
|
|
|
-
|
|
|
|
- def get_channel(self,
|
|
|
|
- target: str,
|
|
|
|
- options: Sequence[Tuple[str, str]],
|
|
|
|
- channel_credentials: Optional[grpc.ChannelCredentials],
|
|
|
|
|
|
+ def get_channel(self, target: str, options: Sequence[Tuple[str, str]],
|
|
|
|
+ channel_credentials: grpc.ChannelCredentials,
|
|
compression: Optional[grpc.Compression]) -> grpc.Channel:
|
|
compression: Optional[grpc.Compression]) -> grpc.Channel:
|
|
key = (target, options, channel_credentials, compression)
|
|
key = (target, options, channel_credentials, compression)
|
|
with self._lock:
|
|
with self._lock:
|
|
@@ -109,12 +107,16 @@ class ChannelCache:
|
|
if channel_data is not None:
|
|
if channel_data is not None:
|
|
channel = channel_data[0]
|
|
channel = channel_data[0]
|
|
self._mapping.pop(key)
|
|
self._mapping.pop(key)
|
|
- self._mapping[key] = (channel, datetime.datetime.now() + _EVICTION_PERIOD)
|
|
|
|
|
|
+ self._mapping[key] = (channel, datetime.datetime.now() +
|
|
|
|
+ _EVICTION_PERIOD)
|
|
return channel
|
|
return channel
|
|
else:
|
|
else:
|
|
- channel = _create_channel(target, options, channel_credentials, compression)
|
|
|
|
- self._mapping[key] = (channel, datetime.datetime.now() + _EVICTION_PERIOD)
|
|
|
|
- if len(self._mapping) == 1 or len(self._mapping) >= _MAXIMUM_CHANNELS:
|
|
|
|
|
|
+ channel = _create_channel(target, options, channel_credentials,
|
|
|
|
+ compression)
|
|
|
|
+ self._mapping[key] = (channel, datetime.datetime.now() +
|
|
|
|
+ _EVICTION_PERIOD)
|
|
|
|
+ if len(self._mapping) == 1 or len(
|
|
|
|
+ self._mapping) >= _MAXIMUM_CHANNELS:
|
|
self._condition.notify()
|
|
self._condition.notify()
|
|
return channel
|
|
return channel
|
|
|
|
|
|
@@ -122,22 +124,38 @@ class ChannelCache:
|
|
with self._lock:
|
|
with self._lock:
|
|
return len(self._mapping)
|
|
return len(self._mapping)
|
|
|
|
|
|
|
|
+
|
|
RequestType = TypeVar('RequestType')
|
|
RequestType = TypeVar('RequestType')
|
|
ResponseType = TypeVar('ResponseType')
|
|
ResponseType = TypeVar('ResponseType')
|
|
|
|
|
|
-def unary_unary(request: RequestType,
|
|
|
|
- target: str,
|
|
|
|
- method: str,
|
|
|
|
- request_serializer: Optional[Callable[[Any], bytes]] = None,
|
|
|
|
- request_deserializer: Optional[Callable[[bytes], Any]] = None,
|
|
|
|
- options: Sequence[Tuple[AnyStr, AnyStr]] = (),
|
|
|
|
- # TODO: Somehow make insecure_channel opt-in, not the default.
|
|
|
|
- channel_credentials: Optional[grpc.ChannelCredentials] = None,
|
|
|
|
- call_credentials: Optional[grpc.CallCredentials] = None,
|
|
|
|
- compression: Optional[grpc.Compression] = None,
|
|
|
|
- wait_for_ready: Optional[bool] = None,
|
|
|
|
- timeout: Optional[float] = None,
|
|
|
|
- metadata: Optional[Sequence[Tuple[str, Union[str, bytes]]]] = None) -> ResponseType:
|
|
|
|
|
|
+
|
|
|
|
+# TODO(rbellevi): Consider a credential type that has the
|
|
|
|
+# following functionality matrix:
|
|
|
|
+#
|
|
|
|
+# +----------+-------+--------+
|
|
|
|
+# | | local | remote |
|
|
|
|
+# |----------+-------+--------+
|
|
|
|
+# | secure | o | o |
|
|
|
|
+# | insecure | o | x |
|
|
|
|
+# +----------+-------+--------+
|
|
|
|
+#
|
|
|
|
+# Make this the default option.
|
|
|
|
+
|
|
|
|
+# TODO: Make LocalChannelCredentials the default.
|
|
|
|
+def unary_unary(
|
|
|
|
+ request: RequestType,
|
|
|
|
+ target: str,
|
|
|
|
+ method: str,
|
|
|
|
+ request_serializer: Optional[Callable[[Any], bytes]] = None,
|
|
|
|
+ request_deserializer: Optional[Callable[[bytes], Any]] = None,
|
|
|
|
+ options: Sequence[Tuple[AnyStr, AnyStr]] = (),
|
|
|
|
+ channel_credentials: Optional[grpc.ChannelCredentials] = None,
|
|
|
|
+ call_credentials: Optional[grpc.CallCredentials] = None,
|
|
|
|
+ compression: Optional[grpc.Compression] = None,
|
|
|
|
+ wait_for_ready: Optional[bool] = None,
|
|
|
|
+ timeout: Optional[float] = None,
|
|
|
|
+ metadata: Optional[Sequence[Tuple[str, Union[str, bytes]]]] = None
|
|
|
|
+) -> ResponseType:
|
|
"""Invokes a unary-unary RPC without an explicitly specified channel.
|
|
"""Invokes a unary-unary RPC without an explicitly specified channel.
|
|
|
|
|
|
This is backed by a per-process cache of channels. Channels are evicted
|
|
This is backed by a per-process cache of channels. Channels are evicted
|
|
@@ -147,7 +165,7 @@ def unary_unary(request: RequestType,
|
|
The default eviction period is 10 minutes. One may set the environment
|
|
The default eviction period is 10 minutes. One may set the environment
|
|
variable "GRPC_PYTHON_MANAGED_CHANNEL_EVICTION_SECONDS" to configure this.
|
|
variable "GRPC_PYTHON_MANAGED_CHANNEL_EVICTION_SECONDS" to configure this.
|
|
|
|
|
|
- The default maximum maximum number of channels is 256. One may set the
|
|
|
|
|
|
+ The default maximum number of channels is 256. One may set the
|
|
environment variable "GRPC_PYTHON_MANAGED_CHANNEL_MAXIMUM" to configure
|
|
environment variable "GRPC_PYTHON_MANAGED_CHANNEL_MAXIMUM" to configure
|
|
this.
|
|
this.
|
|
|
|
|
|
@@ -162,7 +180,8 @@ def unary_unary(request: RequestType,
|
|
options: An optional list of key-value pairs (channel args in gRPC Core
|
|
options: An optional list of key-value pairs (channel args in gRPC Core
|
|
runtime) to configure the channel.
|
|
runtime) to configure the channel.
|
|
channel_credentials: A credential applied to the whole channel, e.g. the
|
|
channel_credentials: A credential applied to the whole channel, e.g. the
|
|
- return value of grpc.ssl_channel_credentials().
|
|
|
|
|
|
+ return value of grpc.ssl_channel_credentials() or
|
|
|
|
+ grpc.insecure_channel_credentials().
|
|
call_credentials: A call credential applied to each call individually,
|
|
call_credentials: A call credential applied to each call individually,
|
|
e.g. the output of grpc.metadata_call_credentials() or
|
|
e.g. the output of grpc.metadata_call_credentials() or
|
|
grpc.access_token_call_credentials().
|
|
grpc.access_token_call_credentials().
|
|
@@ -180,8 +199,11 @@ def unary_unary(request: RequestType,
|
|
Returns:
|
|
Returns:
|
|
The response to the RPC.
|
|
The response to the RPC.
|
|
"""
|
|
"""
|
|
- channel = ChannelCache.get().get_channel(target, options, channel_credentials, compression)
|
|
|
|
- multicallable = channel.unary_unary(method, request_serializer, request_deserializer)
|
|
|
|
|
|
+ channel_credentials = channel_credentials or grpc.local_channel_credentials()
|
|
|
|
+ channel = ChannelCache.get().get_channel(target, options,
|
|
|
|
+ channel_credentials, compression)
|
|
|
|
+ multicallable = channel.unary_unary(method, request_serializer,
|
|
|
|
+ request_deserializer)
|
|
return multicallable(request,
|
|
return multicallable(request,
|
|
metadata=metadata,
|
|
metadata=metadata,
|
|
wait_for_ready=wait_for_ready,
|
|
wait_for_ready=wait_for_ready,
|
|
@@ -189,18 +211,20 @@ def unary_unary(request: RequestType,
|
|
timeout=timeout)
|
|
timeout=timeout)
|
|
|
|
|
|
|
|
|
|
-def unary_stream(request: RequestType,
|
|
|
|
- target: str,
|
|
|
|
- method: str,
|
|
|
|
- request_serializer: Optional[Callable[[Any], bytes]] = None,
|
|
|
|
- request_deserializer: Optional[Callable[[bytes], Any]] = None,
|
|
|
|
- options: Sequence[Tuple[AnyStr, AnyStr]] = (),
|
|
|
|
- channel_credentials: Optional[grpc.ChannelCredentials] = None,
|
|
|
|
- call_credentials: Optional[grpc.CallCredentials] = None,
|
|
|
|
- compression: Optional[grpc.Compression] = None,
|
|
|
|
- wait_for_ready: Optional[bool] = None,
|
|
|
|
- timeout: Optional[float] = None,
|
|
|
|
- metadata: Optional[Sequence[Tuple[str, Union[str, bytes]]]] = None) -> Iterator[ResponseType]:
|
|
|
|
|
|
+def unary_stream(
|
|
|
|
+ request: RequestType,
|
|
|
|
+ target: str,
|
|
|
|
+ method: str,
|
|
|
|
+ request_serializer: Optional[Callable[[Any], bytes]] = None,
|
|
|
|
+ request_deserializer: Optional[Callable[[bytes], Any]] = None,
|
|
|
|
+ options: Sequence[Tuple[AnyStr, AnyStr]] = (),
|
|
|
|
+ channel_credentials: Optional[grpc.ChannelCredentials] = None,
|
|
|
|
+ call_credentials: Optional[grpc.CallCredentials] = None,
|
|
|
|
+ compression: Optional[grpc.Compression] = None,
|
|
|
|
+ wait_for_ready: Optional[bool] = None,
|
|
|
|
+ timeout: Optional[float] = None,
|
|
|
|
+ metadata: Optional[Sequence[Tuple[str, Union[str, bytes]]]] = None
|
|
|
|
+) -> Iterator[ResponseType]:
|
|
"""Invokes a unary-stream RPC without an explicitly specified channel.
|
|
"""Invokes a unary-stream RPC without an explicitly specified channel.
|
|
|
|
|
|
This is backed by a per-process cache of channels. Channels are evicted
|
|
This is backed by a per-process cache of channels. Channels are evicted
|
|
@@ -210,7 +234,7 @@ def unary_stream(request: RequestType,
|
|
The default eviction period is 10 minutes. One may set the environment
|
|
The default eviction period is 10 minutes. One may set the environment
|
|
variable "GRPC_PYTHON_MANAGED_CHANNEL_EVICTION_SECONDS" to configure this.
|
|
variable "GRPC_PYTHON_MANAGED_CHANNEL_EVICTION_SECONDS" to configure this.
|
|
|
|
|
|
- The default maximum maximum number of channels is 256. One may set the
|
|
|
|
|
|
+ The default maximum number of channels is 256. One may set the
|
|
environment variable "GRPC_PYTHON_MANAGED_CHANNEL_MAXIMUM" to configure
|
|
environment variable "GRPC_PYTHON_MANAGED_CHANNEL_MAXIMUM" to configure
|
|
this.
|
|
this.
|
|
|
|
|
|
@@ -243,8 +267,11 @@ def unary_stream(request: RequestType,
|
|
Returns:
|
|
Returns:
|
|
An iterator of responses.
|
|
An iterator of responses.
|
|
"""
|
|
"""
|
|
- channel = ChannelCache.get().get_channel(target, options, channel_credentials, compression)
|
|
|
|
- multicallable = channel.unary_stream(method, request_serializer, request_deserializer)
|
|
|
|
|
|
+ channel_credentials = channel_credentials or grpc.local_channel_credentials()
|
|
|
|
+ channel = ChannelCache.get().get_channel(target, options,
|
|
|
|
+ channel_credentials, compression)
|
|
|
|
+ multicallable = channel.unary_stream(method, request_serializer,
|
|
|
|
+ request_deserializer)
|
|
return multicallable(request,
|
|
return multicallable(request,
|
|
metadata=metadata,
|
|
metadata=metadata,
|
|
wait_for_ready=wait_for_ready,
|
|
wait_for_ready=wait_for_ready,
|
|
@@ -252,18 +279,20 @@ def unary_stream(request: RequestType,
|
|
timeout=timeout)
|
|
timeout=timeout)
|
|
|
|
|
|
|
|
|
|
-def stream_unary(request_iterator: Iterator[RequestType],
|
|
|
|
- target: str,
|
|
|
|
- method: str,
|
|
|
|
- request_serializer: Optional[Callable[[Any], bytes]] = None,
|
|
|
|
- request_deserializer: Optional[Callable[[bytes], Any]] = None,
|
|
|
|
- options: Sequence[Tuple[AnyStr, AnyStr]] = (),
|
|
|
|
- channel_credentials: Optional[grpc.ChannelCredentials] = None,
|
|
|
|
- call_credentials: Optional[grpc.CallCredentials] = None,
|
|
|
|
- compression: Optional[grpc.Compression] = None,
|
|
|
|
- wait_for_ready: Optional[bool] = None,
|
|
|
|
- timeout: Optional[float] = None,
|
|
|
|
- metadata: Optional[Sequence[Tuple[str, Union[str, bytes]]]] = None) -> ResponseType:
|
|
|
|
|
|
+def stream_unary(
|
|
|
|
+ request_iterator: Iterator[RequestType],
|
|
|
|
+ target: str,
|
|
|
|
+ method: str,
|
|
|
|
+ request_serializer: Optional[Callable[[Any], bytes]] = None,
|
|
|
|
+ request_deserializer: Optional[Callable[[bytes], Any]] = None,
|
|
|
|
+ options: Sequence[Tuple[AnyStr, AnyStr]] = (),
|
|
|
|
+ channel_credentials: Optional[grpc.ChannelCredentials] = None,
|
|
|
|
+ call_credentials: Optional[grpc.CallCredentials] = None,
|
|
|
|
+ compression: Optional[grpc.Compression] = None,
|
|
|
|
+ wait_for_ready: Optional[bool] = None,
|
|
|
|
+ timeout: Optional[float] = None,
|
|
|
|
+ metadata: Optional[Sequence[Tuple[str, Union[str, bytes]]]] = None
|
|
|
|
+) -> ResponseType:
|
|
"""Invokes a stream-unary RPC without an explicitly specified channel.
|
|
"""Invokes a stream-unary RPC without an explicitly specified channel.
|
|
|
|
|
|
This is backed by a per-process cache of channels. Channels are evicted
|
|
This is backed by a per-process cache of channels. Channels are evicted
|
|
@@ -273,7 +302,7 @@ def stream_unary(request_iterator: Iterator[RequestType],
|
|
The default eviction period is 10 minutes. One may set the environment
|
|
The default eviction period is 10 minutes. One may set the environment
|
|
variable "GRPC_PYTHON_MANAGED_CHANNEL_EVICTION_SECONDS" to configure this.
|
|
variable "GRPC_PYTHON_MANAGED_CHANNEL_EVICTION_SECONDS" to configure this.
|
|
|
|
|
|
- The default maximum maximum number of channels is 256. One may set the
|
|
|
|
|
|
+ The default maximum number of channels is 256. One may set the
|
|
environment variable "GRPC_PYTHON_MANAGED_CHANNEL_MAXIMUM" to configure
|
|
environment variable "GRPC_PYTHON_MANAGED_CHANNEL_MAXIMUM" to configure
|
|
this.
|
|
this.
|
|
|
|
|
|
@@ -306,8 +335,11 @@ def stream_unary(request_iterator: Iterator[RequestType],
|
|
Returns:
|
|
Returns:
|
|
The response to the RPC.
|
|
The response to the RPC.
|
|
"""
|
|
"""
|
|
- channel = ChannelCache.get().get_channel(target, options, channel_credentials, compression)
|
|
|
|
- multicallable = channel.stream_unary(method, request_serializer, request_deserializer)
|
|
|
|
|
|
+ channel_credentials = channel_credentials or grpc.local_channel_credentials()
|
|
|
|
+ channel = ChannelCache.get().get_channel(target, options,
|
|
|
|
+ channel_credentials, compression)
|
|
|
|
+ multicallable = channel.stream_unary(method, request_serializer,
|
|
|
|
+ request_deserializer)
|
|
return multicallable(request_iterator,
|
|
return multicallable(request_iterator,
|
|
metadata=metadata,
|
|
metadata=metadata,
|
|
wait_for_ready=wait_for_ready,
|
|
wait_for_ready=wait_for_ready,
|
|
@@ -315,18 +347,20 @@ def stream_unary(request_iterator: Iterator[RequestType],
|
|
timeout=timeout)
|
|
timeout=timeout)
|
|
|
|
|
|
|
|
|
|
-def stream_stream(request_iterator: Iterator[RequestType],
|
|
|
|
- target: str,
|
|
|
|
- method: str,
|
|
|
|
- request_serializer: Optional[Callable[[Any], bytes]] = None,
|
|
|
|
- request_deserializer: Optional[Callable[[bytes], Any]] = None,
|
|
|
|
- options: Sequence[Tuple[AnyStr, AnyStr]] = (),
|
|
|
|
- channel_credentials: Optional[grpc.ChannelCredentials] = None,
|
|
|
|
- call_credentials: Optional[grpc.CallCredentials] = None,
|
|
|
|
- compression: Optional[grpc.Compression] = None,
|
|
|
|
- wait_for_ready: Optional[bool] = None,
|
|
|
|
- timeout: Optional[float] = None,
|
|
|
|
- metadata: Optional[Sequence[Tuple[str, Union[str, bytes]]]] = None) -> Iterator[ResponseType]:
|
|
|
|
|
|
+def stream_stream(
|
|
|
|
+ request_iterator: Iterator[RequestType],
|
|
|
|
+ target: str,
|
|
|
|
+ method: str,
|
|
|
|
+ request_serializer: Optional[Callable[[Any], bytes]] = None,
|
|
|
|
+ request_deserializer: Optional[Callable[[bytes], Any]] = None,
|
|
|
|
+ options: Sequence[Tuple[AnyStr, AnyStr]] = (),
|
|
|
|
+ channel_credentials: Optional[grpc.ChannelCredentials] = None,
|
|
|
|
+ call_credentials: Optional[grpc.CallCredentials] = None,
|
|
|
|
+ compression: Optional[grpc.Compression] = None,
|
|
|
|
+ wait_for_ready: Optional[bool] = None,
|
|
|
|
+ timeout: Optional[float] = None,
|
|
|
|
+ metadata: Optional[Sequence[Tuple[str, Union[str, bytes]]]] = None
|
|
|
|
+) -> Iterator[ResponseType]:
|
|
"""Invokes a stream-stream RPC without an explicitly specified channel.
|
|
"""Invokes a stream-stream RPC without an explicitly specified channel.
|
|
|
|
|
|
This is backed by a per-process cache of channels. Channels are evicted
|
|
This is backed by a per-process cache of channels. Channels are evicted
|
|
@@ -336,7 +370,7 @@ def stream_stream(request_iterator: Iterator[RequestType],
|
|
The default eviction period is 10 minutes. One may set the environment
|
|
The default eviction period is 10 minutes. One may set the environment
|
|
variable "GRPC_PYTHON_MANAGED_CHANNEL_EVICTION_SECONDS" to configure this.
|
|
variable "GRPC_PYTHON_MANAGED_CHANNEL_EVICTION_SECONDS" to configure this.
|
|
|
|
|
|
- The default maximum maximum number of channels is 256. One may set the
|
|
|
|
|
|
+ The default maximum number of channels is 256. One may set the
|
|
environment variable "GRPC_PYTHON_MANAGED_CHANNEL_MAXIMUM" to configure
|
|
environment variable "GRPC_PYTHON_MANAGED_CHANNEL_MAXIMUM" to configure
|
|
this.
|
|
this.
|
|
|
|
|
|
@@ -369,8 +403,11 @@ def stream_stream(request_iterator: Iterator[RequestType],
|
|
Returns:
|
|
Returns:
|
|
An iterator of responses.
|
|
An iterator of responses.
|
|
"""
|
|
"""
|
|
- channel = ChannelCache.get().get_channel(target, options, channel_credentials, compression)
|
|
|
|
- multicallable = channel.stream_stream(method, request_serializer, request_deserializer)
|
|
|
|
|
|
+ channel_credentials = channel_credentials or grpc.local_channel_credentials()
|
|
|
|
+ channel = ChannelCache.get().get_channel(target, options,
|
|
|
|
+ channel_credentials, compression)
|
|
|
|
+ multicallable = channel.stream_stream(method, request_serializer,
|
|
|
|
+ request_deserializer)
|
|
return multicallable(request_iterator,
|
|
return multicallable(request_iterator,
|
|
metadata=metadata,
|
|
metadata=metadata,
|
|
wait_for_ready=wait_for_ready,
|
|
wait_for_ready=wait_for_ready,
|