Usage Engine Connection Fix (#4231)

This commit is contained in:
Mayur Singal 2022-04-19 20:21:59 +05:30 committed by GitHub
parent 8cecd61796
commit 7283b858e4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 4 additions and 4 deletions

View File

@ -43,7 +43,7 @@ class ClickhouseUsageSource(Source[TableQuery]):
start_time=start, end_time=end start_time=start, end_time=end
) )
self.report = SQLSourceStatus() self.report = SQLSourceStatus()
self.engine = get_engine(self.config.serviceConnection) self.engine = get_engine(self.connection)
@classmethod @classmethod
def create(cls, config_dict, metadata_config: WorkflowConfig): def create(cls, config_dict, metadata_config: WorkflowConfig):

View File

@ -40,7 +40,7 @@ class MssqlUsageSource(Source[TableQuery]):
self.analysis_date = start self.analysis_date = start
self.sql_stmt = MSSQL_SQL_USAGE_STATEMENT.format(start_date=start, end_date=end) self.sql_stmt = MSSQL_SQL_USAGE_STATEMENT.format(start_date=start, end_date=end)
self.report = SQLSourceStatus() self.report = SQLSourceStatus()
self.engine = get_engine(self.config.serviceConnection) self.engine = get_engine(self.connection)
@classmethod @classmethod
def create(cls, config_dict, metadata_config: WorkflowConfig): def create(cls, config_dict, metadata_config: WorkflowConfig):

View File

@ -64,7 +64,7 @@ class RedshiftUsageSource(Source[TableQuery]):
self._extract_iter: Union[None, Iterator] = None self._extract_iter: Union[None, Iterator] = None
self._database = "redshift" self._database = "redshift"
self.status = SQLSourceStatus() self.status = SQLSourceStatus()
self.engine = get_engine(self.config.serviceConnection) self.engine = get_engine(self.service_connection)
@classmethod @classmethod
def create(cls, config_dict, metadata_config: WorkflowConfig): def create(cls, config_dict, metadata_config: WorkflowConfig):

View File

@ -70,7 +70,7 @@ class SnowflakeUsageSource(Source[TableQuery]):
self._extract_iter: Union[None, Iterator] = None self._extract_iter: Union[None, Iterator] = None
self._database = "Snowflake" self._database = "Snowflake"
self.report = SQLSourceStatus() self.report = SQLSourceStatus()
self.engine = get_engine(self.config.serviceConnection) self.engine = get_engine(self.service_connection)
@classmethod @classmethod
def create(cls, config_dict, metadata_config: WorkflowConfig): def create(cls, config_dict, metadata_config: WorkflowConfig):