diff --git a/ingestion/src/metadata/ingestion/source/clickhouse_usage.py b/ingestion/src/metadata/ingestion/source/clickhouse_usage.py index 2b70f9bd382..38a63df6f08 100644 --- a/ingestion/src/metadata/ingestion/source/clickhouse_usage.py +++ b/ingestion/src/metadata/ingestion/source/clickhouse_usage.py @@ -43,7 +43,7 @@ class ClickhouseUsageSource(Source[TableQuery]): start_time=start, end_time=end ) self.report = SQLSourceStatus() - self.engine = get_engine(self.config.serviceConnection) + self.engine = get_engine(self.connection) @classmethod def create(cls, config_dict, metadata_config: WorkflowConfig): diff --git a/ingestion/src/metadata/ingestion/source/mssql_usage.py b/ingestion/src/metadata/ingestion/source/mssql_usage.py index 095dc035ea0..2aef450513b 100644 --- a/ingestion/src/metadata/ingestion/source/mssql_usage.py +++ b/ingestion/src/metadata/ingestion/source/mssql_usage.py @@ -40,7 +40,7 @@ class MssqlUsageSource(Source[TableQuery]): self.analysis_date = start self.sql_stmt = MSSQL_SQL_USAGE_STATEMENT.format(start_date=start, end_date=end) self.report = SQLSourceStatus() - self.engine = get_engine(self.config.serviceConnection) + self.engine = get_engine(self.connection) @classmethod def create(cls, config_dict, metadata_config: WorkflowConfig): diff --git a/ingestion/src/metadata/ingestion/source/redshift_usage.py b/ingestion/src/metadata/ingestion/source/redshift_usage.py index 47c5be1b705..2fa1b46baec 100644 --- a/ingestion/src/metadata/ingestion/source/redshift_usage.py +++ b/ingestion/src/metadata/ingestion/source/redshift_usage.py @@ -64,7 +64,7 @@ class RedshiftUsageSource(Source[TableQuery]): self._extract_iter: Union[None, Iterator] = None self._database = "redshift" self.status = SQLSourceStatus() - self.engine = get_engine(self.config.serviceConnection) + self.engine = get_engine(self.service_connection) @classmethod def create(cls, config_dict, metadata_config: WorkflowConfig): diff --git a/ingestion/src/metadata/ingestion/source/snowflake_usage.py b/ingestion/src/metadata/ingestion/source/snowflake_usage.py index d6e8271751d..55d6fc689f8 100644 --- a/ingestion/src/metadata/ingestion/source/snowflake_usage.py +++ b/ingestion/src/metadata/ingestion/source/snowflake_usage.py @@ -70,7 +70,7 @@ class SnowflakeUsageSource(Source[TableQuery]): self._extract_iter: Union[None, Iterator] = None self._database = "Snowflake" self.report = SQLSourceStatus() - self.engine = get_engine(self.config.serviceConnection) + self.engine = get_engine(self.service_connection) @classmethod def create(cls, config_dict, metadata_config: WorkflowConfig):