From 4ac269acd1c9abce4f07ad541bdddbd133c04a93 Mon Sep 17 00:00:00 2001 From: Joel K Biju Date: Fri, 1 Sep 2023 19:37:19 +0530 Subject: [PATCH 1/2] Adding logs and removing exception --- src/operators/deferred_job_result.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/operators/deferred_job_result.py b/src/operators/deferred_job_result.py index c2ee8f6..239bf2f 100644 --- a/src/operators/deferred_job_result.py +++ b/src/operators/deferred_job_result.py @@ -62,7 +62,10 @@ def execute(self, context, session=None, event=None): self.log.info('Waiting for task "%s" to reach state %s...', str(task), self.wait_for_state) if task.state == State.FAILED: - raise ErgoFailedResultException(400, "Cron execution failed") + error_message = "Cron execution failed" + self.log.error(error_message) + ti = context['ti'] + ti.set_state(State.FAILED) self.log.info('Task - %s reached state %s', str(task), task.state) return From 4fddc823f231ec42641862237ac219aaf8dade56 Mon Sep 17 00:00:00 2001 From: Joel K Biju Date: Sat, 2 Sep 2023 19:02:32 +0530 Subject: [PATCH 2/2] Adding logging --- src/operators/deferred_job_result.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/src/operators/deferred_job_result.py b/src/operators/deferred_job_result.py index 239bf2f..e13177d 100644 --- a/src/operators/deferred_job_result.py +++ b/src/operators/deferred_job_result.py @@ -62,10 +62,11 @@ def execute(self, context, session=None, event=None): self.log.info('Waiting for task "%s" to reach state %s...', str(task), self.wait_for_state) if task.state == State.FAILED: - error_message = "Cron execution failed" - self.log.error(error_message) - ti = context['ti'] - ti.set_state(State.FAILED) + if job is not None: + self.log.info('Job - (%s)' + (f'responded back at {job.response_at}' if job.response_at else ''), str(job)) + raise ErgoFailedResultException(job.result_code, job.error_msg) + else: + raise ErgoFailedResultException(400, "Cron execution failed due to unknown reason") self.log.info('Task - %s reached state %s', str(task), task.state) return