fallback if credentials fail for metrics client

Author: daniel-sanche

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

@@ -17,6 +17,7 @@
 
 import abc
 import concurrent.futures
+import logging
 import os
 import random
 import time
@@ -69,6 +70,8 @@
     OperationType,
     tracked_retry,
 )
+
+_LOGGER = logging.getLogger(__name__)
 from google.cloud.bigtable.data._metrics.handlers.gcp_exporter import (
     BigtableMetricsExporter,
     GoogleCloudMetricsHandler,
@@ -275,16 +278,26 @@ def __init__(
                 client_version=self._client_version(),
             )
         else:
-            # create a metrics exporter using the same client configuration
-            exporter = BigtableMetricsExporter(
-                project_id=self.project,
-                credentials=credentials,
-                client_options=client_options,
-            )
-            self._metrics_handler = GoogleCloudMetricsHandler(
-                exporter=exporter,
-                client_version=self._client_version(),
-            )
+            try:
+                # create a metrics exporter using the same client configuration
+                exporter = BigtableMetricsExporter(
+                    project_id=self.project,
+                    credentials=credentials,
+                    client_options=client_options,
+                )
+                self._metrics_handler = GoogleCloudMetricsHandler(
+                    exporter=exporter,
+                    client_version=self._client_version(),
+                )
+            except Exception as e:
+                _LOGGER.warning(
+                    "Failed to initialize Google Cloud Metrics Exporter: %s. "
+                    "Falling back to local OpenTelemetry metrics handler.",
+                    e,
+                )
+                self._metrics_handler = OpenTelemetryMetricsHandler(
+                    client_version=self._client_version(),
+                )
         self._metrics = BigtableClientSideMetricsController(
             handlers=[self._metrics_handler]
         )

packages/google-cloud-bigtable/google/cloud/bigtable/data/_sync_autogen/client.py

@@ -20,6 +20,7 @@
 
 import abc
 import concurrent.futures
+import logging
 import os
 import random
 import time
@@ -63,6 +64,8 @@
     OperationType,
     tracked_retry,
 )
+
+_LOGGER = logging.getLogger(__name__)
 from google.cloud.bigtable.data._metrics.handlers.gcp_exporter import (
     BigtableMetricsExporter,
     GoogleCloudMetricsHandler,
@@ -206,15 +209,25 @@ def __init__(
                 client_version=self._client_version()
             )
         else:
-            exporter = BigtableMetricsExporter(
-                project_id=self.project,
-                credentials=credentials,
-                client_options=client_options,
-            )
-            self._metrics_handler = GoogleCloudMetricsHandler(
-                exporter=exporter,
-                client_version=self._client_version(),
-            )
+            try:
+                exporter = BigtableMetricsExporter(
+                    project_id=self.project,
+                    credentials=credentials,
+                    client_options=client_options,
+                )
+                self._metrics_handler = GoogleCloudMetricsHandler(
+                    exporter=exporter,
+                    client_version=self._client_version(),
+                )
+            except Exception as e:
+                _LOGGER.warning(
+                    "Failed to initialize Google Cloud Metrics Exporter: %s. "
+                    "Falling back to local OpenTelemetry metrics handler.",
+                    e,
+                )
+                self._metrics_handler = OpenTelemetryMetricsHandler(
+                    client_version=self._client_version(),
+                )
         self._metrics = BigtableClientSideMetricsController(
             handlers=[self._metrics_handler]
         )

packages/google-cloud-bigtable/tests/unit/data/_async/test_client.py

@@ -255,6 +255,24 @@ async def test_metrics_exporter_init_implicit_project(self):
         async with self._make_client(use_emulator=False) as client:
             assert client._metrics.handlers[0]._exporter.project_id == client.project
 
+    @CrossSync.pytest
+    @mock.patch(
+        "google.cloud.bigtable.data._async.client.BigtableMetricsExporter",
+        side_effect=Exception("Auth error"),
+    )
+    @mock.patch(
+        "google.cloud.bigtable.data._sync_autogen.client.BigtableMetricsExporter",
+        side_effect=Exception("Auth error"),
+    )
+    async def test_metrics_exporter_init_error_fallback(self, mock_sync, mock_async):
+        from google.cloud.bigtable.data._metrics.handlers.opentelemetry import (
+            OpenTelemetryMetricsHandler,
+        )
+
+        async with self._make_client(use_emulator=False) as client:
+            assert isinstance(client._metrics.handlers[0], OpenTelemetryMetricsHandler)
+            assert getattr(client._metrics.handlers[0], "_exporter", None) is None
+
     @CrossSync.pytest
     @mock.patch("google.cloud.bigtable.data._async.client.BigtableMetricsExporter")
     @mock.patch(

packages/google-cloud-bigtable/tests/unit/data/_sync_autogen/test_client.py

@@ -225,6 +225,23 @@ def test_metrics_exporter_init_implicit_project(self):
         with self._make_client(use_emulator=False) as client:
             assert client._metrics.handlers[0]._exporter.project_id == client.project
 
+    @mock.patch(
+        "google.cloud.bigtable.data._async.client.BigtableMetricsExporter",
+        side_effect=Exception("Auth error"),
+    )
+    @mock.patch(
+        "google.cloud.bigtable.data._sync_autogen.client.BigtableMetricsExporter",
+        side_effect=Exception("Auth error"),
+    )
+    def test_metrics_exporter_init_error_fallback(self, mock_sync, mock_async):
+        from google.cloud.bigtable.data._metrics.handlers.opentelemetry import (
+            OpenTelemetryMetricsHandler,
+        )
+
+        with self._make_client(use_emulator=False) as client:
+            assert isinstance(client._metrics.handlers[0], OpenTelemetryMetricsHandler)
+            assert getattr(client._metrics.handlers[0], "_exporter", None) is None
+
     @mock.patch("google.cloud.bigtable.data._async.client.BigtableMetricsExporter")
     @mock.patch(
         "google.cloud.bigtable.data._sync_autogen.client.BigtableMetricsExporter"