feat: add pooled grpc transport (#748)

googleapis · Apr 24, 2023 · f9a1907 · f9a1907
1 parent c55099f
commit f9a1907
Show file tree

Hide file tree

Showing 10 changed files with 2,207 additions and 30 deletions.
diff --git a/.gitmodules b/.gitmodules
@@ -0,0 +1,3 @@
+[submodule "gapic-generator-fork"]
+	path = gapic-generator-fork
+	url = [email protected]:googleapis/gapic-generator-python.git
diff --git a/gapic-generator-fork b/gapic-generator-fork
diff --git a/google/cloud/bigtable/client.py b/google/cloud/bigtable/client.py
diff --git a/google/cloud/bigtable_v2/services/bigtable/client.py b/google/cloud/bigtable_v2/services/bigtable/client.py
@@ -53,6 +53,7 @@
 from .transports.base import BigtableTransport, DEFAULT_CLIENT_INFO
 from .transports.grpc import BigtableGrpcTransport
 from .transports.grpc_asyncio import BigtableGrpcAsyncIOTransport
+from .transports.pooled_grpc_asyncio import PooledBigtableGrpcAsyncIOTransport
 from .transports.rest import BigtableRestTransport
 
 
@@ -67,6 +68,7 @@ class BigtableClientMeta(type):
     _transport_registry = OrderedDict()  # type: Dict[str, Type[BigtableTransport]]
     _transport_registry["grpc"] = BigtableGrpcTransport
     _transport_registry["grpc_asyncio"] = BigtableGrpcAsyncIOTransport
+    _transport_registry["pooled_grpc_asyncio"] = PooledBigtableGrpcAsyncIOTransport
     _transport_registry["rest"] = BigtableRestTransport
 
     def get_transport_class(

diff --git a/google/cloud/bigtable_v2/services/bigtable/transports/__init__.py b/google/cloud/bigtable_v2/services/bigtable/transports/__init__.py
@@ -19,6 +19,7 @@
 from .base import BigtableTransport
 from .grpc import BigtableGrpcTransport
 from .grpc_asyncio import BigtableGrpcAsyncIOTransport
+from .pooled_grpc_asyncio import PooledBigtableGrpcAsyncIOTransport
 from .rest import BigtableRestTransport
 from .rest import BigtableRestInterceptor
 
@@ -27,12 +28,14 @@
 _transport_registry = OrderedDict()  # type: Dict[str, Type[BigtableTransport]]
 _transport_registry["grpc"] = BigtableGrpcTransport
 _transport_registry["grpc_asyncio"] = BigtableGrpcAsyncIOTransport
+_transport_registry["pooled_grpc_asyncio"] = PooledBigtableGrpcAsyncIOTransport
 _transport_registry["rest"] = BigtableRestTransport
 
 __all__ = (
     "BigtableTransport",
     "BigtableGrpcTransport",
     "BigtableGrpcAsyncIOTransport",
+    "PooledBigtableGrpcAsyncIOTransport",
     "BigtableRestTransport",
     "BigtableRestInterceptor",
 )