[DRAFT] feat: support authorized views

docs/async_data_client/async_data_authorized_view.rst

@@ -0,0 +1,11 @@
+Authorized View Async
+~~~~~~~~~~~~~~~~~~~~~
+
+    .. note::
+
+      It is generally not recommended to use the async client in an otherwise synchronous codebase. To make use of asyncio's
+      performance benefits, the codebase should be designed to be async from the ground up.
+
+.. autoclass:: google.cloud.bigtable.data._async.client.AuthorizedViewAsync
+  :members:
+  :inherited-members:

docs/async_data_client/async_data_table.rst

@@ -8,4 +8,4 @@ Table Async
 
 .. autoclass:: google.cloud.bigtable.data._async.client.TableAsync
   :members:
-  :show-inheritance:
+  :inherited-members:

docs/async_data_client/async_data_usage.rst

@@ -6,6 +6,7 @@ Async Data Client
 
   async_data_client
   async_data_table
+  async_data_authorized_view
   async_data_mutations_batcher
   async_data_read_rows_query
   async_data_row

google/cloud/bigtable/data/_async/_mutate_rows.py

@@ -15,10 +15,10 @@
 from __future__ import annotations
 
 from typing import Sequence, TYPE_CHECKING
-import functools
 
 from google.api_core import exceptions as core_exceptions
 from google.api_core import retry as retries
+import google.cloud.bigtable_v2.types.bigtable as types_pb
 import google.cloud.bigtable.data.exceptions as bt_exceptions
 from google.cloud.bigtable.data._helpers import _attempt_timeout_generator
 from google.cloud.bigtable.data._helpers import _retry_exception_factory
@@ -36,11 +36,17 @@
         from google.cloud.bigtable_v2.services.bigtable.async_client import (
             BigtableAsyncClient as GapicClientType,
         )
+        from google.cloud.bigtable_v2.services.bigtable.client import (  # type: ignore
+            _ApiSurfaceAsync as ApiSurfaceType,
+        )
         from google.cloud.bigtable.data._async.client import TableAsync as TableType
     else:
         from google.cloud.bigtable_v2.services.bigtable.client import (  # type: ignore
             BigtableClient as GapicClientType,
         )
+        from google.cloud.bigtable_v2.services.bigtable.client import (  # type: ignore
+            _ApiSurface as ApiSurfaceType,
+        )
         from google.cloud.bigtable.data._sync_autogen.client import Table as TableType  # type: ignore
 
 __CROSS_SYNC_OUTPUT__ = "google.cloud.bigtable.data._sync_autogen._mutate_rows"
@@ -70,7 +76,7 @@ class _MutateRowsOperationAsync:
     def __init__(
         self,
         gapic_client: GapicClientType,
-        table: TableType,
+        table: ApiSurfaceType,
         mutation_entries: list["RowMutationEntry"],
         operation_timeout: float,
         attempt_timeout: float | None,
@@ -84,13 +90,8 @@ def __init__(
                 f"{_MUTATE_ROWS_REQUEST_MUTATION_LIMIT} mutations across "
                 f"all entries. Found {total_mutations}."
             )
-        # create partial function to pass to trigger rpc call
-        self._gapic_fn = functools.partial(
-            gapic_client.mutate_rows,
-            table_name=table.table_name,
-            app_profile_id=table.app_profile_id,
-            retry=None,
-        )
+        self._table = table
+        self._gapic_fn = gapic_client.mutate_rows
         # create predicate for determining which errors are retryable
         self.is_retryable = retries.if_exception_type(
             # RPC level errors
@@ -173,8 +174,12 @@ async def _run_attempt(self):
         # make gapic request
         try:
             result_generator = await self._gapic_fn(
+                request=types_pb.MutateRowsRequest(
+                    entries=request_entries,
+                    app_profile_id=self._table.app_profile_id,
+                    **self._table._request_path,
+                ),
                 timeout=next(self.timeout_generator),
-                entries=request_entries,
                 retry=None,
             )
             async for result_list in result_generator:

google/cloud/bigtable/data/_async/_read_rows.py

@@ -37,9 +37,11 @@
 
 if TYPE_CHECKING:
     if CrossSync.is_async:
-        from google.cloud.bigtable.data._async.client import TableAsync as TableType
+        from google.cloud.bigtable.data._async.client import (
+            _ApiSurfaceAsync as ApiSurfaceType,
+        )
     else:
-        from google.cloud.bigtable.data._sync_autogen.client import Table as TableType  # type: ignore
+        from google.cloud.bigtable.data._sync_autogen.client import _ApiSurface as ApiSurfaceType  # type: ignore
 
 __CROSS_SYNC_OUTPUT__ = "google.cloud.bigtable.data._sync_autogen._read_rows"
 
@@ -78,7 +80,7 @@ class _ReadRowsOperationAsync:
     def __init__(
         self,
         query: ReadRowsQuery,
-        table: TableType,
+        table: ApiSurfaceType,
         operation_timeout: float,
         attempt_timeout: float,
         retryable_exceptions: Sequence[type[Exception]] = (),
@@ -90,7 +92,7 @@ def __init__(
         if isinstance(query, dict):
             self.request = ReadRowsRequestPB(
                 **query,
-                table_name=table.table_name,
+                **table._request_path,
                 app_profile_id=table.app_profile_id,
             )
         else: