From a84ea5e5485a09c9b9dcfc1214a90066a9a8a8f6 Mon Sep 17 00:00:00 2001 From: Teddy Date: Wed, 13 Sep 2023 14:17:39 +0200 Subject: [PATCH] fix: set engine for each project ID (#13176) --- .../src/metadata/ingestion/source/database/bigquery/metadata.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/ingestion/src/metadata/ingestion/source/database/bigquery/metadata.py b/ingestion/src/metadata/ingestion/source/database/bigquery/metadata.py index a5a764f38b7..db641067072 100644 --- a/ingestion/src/metadata/ingestion/source/database/bigquery/metadata.py +++ b/ingestion/src/metadata/ingestion/source/database/bigquery/metadata.py @@ -65,6 +65,7 @@ from metadata.ingestion.api.steps import InvalidSourceException from metadata.ingestion.lineage.models import ConnectionTypeDialectMapper from metadata.ingestion.lineage.sql_lineage import get_lineage_by_query from metadata.ingestion.models.ometa_classification import OMetaTagAndClassification +from metadata.ingestion.source.connections import get_connection from metadata.ingestion.source.database.bigquery.models import ( STORED_PROC_LANGUAGE_MAP, BigQueryStoredProcedure, @@ -397,6 +398,7 @@ class BigquerySource(CommonDbSourceService): ) self.client = get_bigquery_client(project_id=database_name, **kwargs) + self.engine = get_connection(self.service_connection) self.inspector = inspect(self.engine) def get_database_names(self) -> Iterable[str]: