fix: set engine for each project ID (#13176)

This commit is contained in:
Teddy 2023-09-13 14:17:39 +02:00 committed by GitHub
parent 791b2f4c38
commit a84ea5e548
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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]: