From 7283b858e48c272bdee12c32a800b238b7efb596 Mon Sep 17 00:00:00 2001
From: Mayur Singal <39544459+ulixius9@users.noreply.github.com>
Date: Tue, 19 Apr 2022 20:21:59 +0530
Subject: [PATCH] Usage Engine Connection Fix (#4231)

---
 ingestion/src/metadata/ingestion/source/clickhouse_usage.py | 2 +-
 ingestion/src/metadata/ingestion/source/mssql_usage.py      | 2 +-
 ingestion/src/metadata/ingestion/source/redshift_usage.py   | 2 +-
 ingestion/src/metadata/ingestion/source/snowflake_usage.py  | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

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):