use default backoff instance

Author: daniel-sanche

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

@@ -102,11 +102,10 @@ def __init__(
             # Entry level errors
             bt_exceptions._MutateRowsIncomplete,
         )
-        sleep_generator = TrackedBackoffGenerator(0.01, 2, 60)
         self._operation = lambda: CrossSync.retry_target(
             self._run_attempt,
             self.is_retryable,
-            sleep_generator,
+            metric.backoff_generator,
             operation_timeout,
             exception_factory=_retry_exception_factory,
         )
@@ -118,7 +117,6 @@ def __init__(
         self.remaining_indices = list(range(len(self.mutations)))
         self.errors: dict[int, list[Exception]] = {}
         # set up metrics
-        metric.backoff_generator = sleep_generator
         self._operation_metric = metric
 
     @CrossSync.convert

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

@@ -118,9 +118,6 @@ def start_operation(self) -> CrossSync.Iterable[Row]:
         Yields:
             Row: The next row in the stream
         """
-        self._operation_metric.backoff_generator = TrackedBackoffGenerator(
-            0.01, 60, multiplier=2
-        )
         return CrossSync.retry_target_stream(
             self._read_rows_attempt,
             self._predicate,

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

@@ -1345,11 +1345,9 @@ async def sample_row_keys(
         retryable_excs = _get_retryable_errors(retryable_errors, self)
         predicate = retries.if_exception_type(*retryable_excs)
 
-        sleep_generator = TrackedBackoffGenerator(0.01, 2, 60)
-
         async with self._metrics.create_operation(
-            OperationType.SAMPLE_ROW_KEYS, backoff_generator=sleep_generator
-        ):
+            OperationType.SAMPLE_ROW_KEYS
+        ) as operation_metric:
 
             @CrossSync.convert
             async def execute_rpc():
@@ -1365,7 +1363,7 @@ async def execute_rpc():
             return await CrossSync.retry_target(
                 execute_rpc,
                 predicate,
-                sleep_generator,
+                operation_metric.backoff_generator,
                 operation_timeout,
                 exception_factory=_retry_exception_factory,
             )
@@ -1479,10 +1477,9 @@ async def mutate_row(
             # mutations should not be retried
             predicate = retries.if_exception_type()
 
-        sleep_generator = TrackedBackoffGenerator(0.01, 2, 60)
         async with self._metrics.create_operation(
-            OperationType.MUTATE_ROW, backoff_generator=sleep_generator
-        ):
+            OperationType.MUTATE_ROW
+        ) as operation_metric:
             target = partial(
                 self.client._gapic_client.mutate_row,
                 request=MutateRowRequest(
@@ -1499,7 +1496,7 @@ async def mutate_row(
             return await CrossSync.retry_target(
                 target,
                 predicate,
-                sleep_generator,
+                operation_metric.backoff_generator,
                 operation_timeout,
                 exception_factory=_retry_exception_factory,
             )

google/cloud/bigtable/data/_metrics/data_model.py

@@ -429,4 +429,4 @@ async def __aexit__(self, exc_type, exc_val, exc_tb):
         if exc_val is None:
             self.end_with_success()
         else:
-            self.end_with_status(exc_val)
+            self.end_with_status(exc_val)