add list projects timer details

datahub-project · Jul 26, 2024 · 6dcd898 · 6dcd898
1 parent 9e1b62b
commit 6dcd898
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/bigquery_v2/bigquery_queries.py b/metadata-ingestion/src/datahub/ingestion/source/bigquery_v2/bigquery_queries.py
@@ -1,5 +1,5 @@
 import logging
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from typing import Iterable, Optional
 
 from pydantic import Field
@@ -35,6 +35,9 @@
 class BigQueryQueriesSourceReport(SourceReport):
     window: Optional[BaseTimeWindowConfig] = None
     queries_extractor: Optional[BigQueryQueriesExtractorReport] = None
+    schema_api_perf: BigQuerySchemaApiPerfReport = field(
+        default_factory=BigQuerySchemaApiPerfReport
+    )
 
 
 class BigQueryQueriesSourceConfig(
@@ -58,9 +61,7 @@ def __init__(self, ctx: PipelineContext, config: BigQueryQueriesSourceConfig):
 
         self.queries_extractor = BigQueryQueriesExtractor(
             connection=self.connection,
-            schema_api=BigQuerySchemaApi(
-                BigQuerySchemaApiPerfReport(), self.connection
-            ),
+            schema_api=BigQuerySchemaApi(self.report.schema_api_perf, self.connection),
             config=self.config,
             structured_report=self.report,
             filters=self.filters,