Skip to content

Create a new method used to resume the task in order to implement specific logic for operators#33424

Merged
hussein-awala merged 6 commits into
apache:mainfrom
hussein-awala:sensor_soft_fail
Aug 21, 2023
Merged

Create a new method used to resume the task in order to implement specific logic for operators#33424
hussein-awala merged 6 commits into
apache:mainfrom
hussein-awala:sensor_soft_fail

Conversation

@hussein-awala

Copy link
Copy Markdown
Member

related: #33405
related: #33196

This PR creates a new method used by the TI to resume the execution, and moves handling the triggers errors to this method. In this way we can simplify handling the trigger result which should not be in TaskInstance, and implement specific processing for some operators.

With this solution we can easily handle the soft fail for all the sensors (the core operators, where we still need a solution for providers sensors when using an older version of Airflow).


^ Add meaningful description above
Read the Pull Request Guidelines for more information.
In case of fundamental code changes, an Airflow Improvement Proposal (AIP) is needed.
In case of a new dependency, check compliance with the ASF 3rd Party License Policy.
In case of backwards incompatible changes please leave a note in a newsfragment file, named {pr_number}.significant.rst or {issue_number}.significant.rst, in newsfragments.

@hussein-awala

Copy link
Copy Markdown
Member Author

I will try to add a test tomorrow.

Comment thread airflow/sensors/base.py Outdated
Comment thread airflow/sensors/base.py Outdated
@hussein-awala hussein-awala added this to the Airflow 2.7.1 milestone Aug 17, 2023
@hussein-awala hussein-awala added the type:bug-fix Changelog: Bug Fixes label Aug 17, 2023

@uranusjr uranusjr left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Mosly look good to me, just a couple of minor questions regarding exception handling

Comment thread airflow/sensors/base.py Outdated
Comment thread airflow/sensors/base.py Outdated
Co-authored-by: Tzu-ping Chung <uranusjr@gmail.com>
@ephraimbuddy

Copy link
Copy Markdown
Contributor

This doesn't look like it's a bug fix?

ephraimbuddy pushed a commit that referenced this pull request Aug 28, 2023
…cific logic for operators (#33424)

* Create a generic method used to resume the task in order to implement specific logic for each operator

* Provide resume_execution args

* add a unit test

* Fix soft fail after a trigger timeout

* wrap the exception with a str to get the message

* Apply suggestions from code review

Co-authored-by: Tzu-ping Chung <uranusjr@gmail.com>

---------

Co-authored-by: Tzu-ping Chung <uranusjr@gmail.com>
(cherry picked from commit 852f85b)
Lee-W added a commit to astronomer/airflow that referenced this pull request Aug 31, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

type:bug-fix Changelog: Bug Fixes

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants