Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Merged
merged 6 commits into from
Aug 21, 2023

Conversation

hussein-awala
Copy link
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.

@boring-cyborg boring-cyborg bot added the area:core-operators Operators, Sensors and hooks within Core Airflow label Aug 15, 2023
@hussein-awala
Copy link
Member Author

I will try to add a test tomorrow.

airflow/sensors/base.py Outdated Show resolved Hide resolved
airflow/sensors/base.py Outdated Show resolved Hide resolved
@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
Copy link
Member

@uranusjr uranusjr left a comment

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

airflow/sensors/base.py Outdated Show resolved Hide resolved
airflow/sensors/base.py Outdated Show resolved Hide resolved
Co-authored-by: Tzu-ping Chung <uranusjr@gmail.com>
@hussein-awala hussein-awala merged commit 852f85b into apache:main Aug 21, 2023
42 checks passed
@ephraimbuddy
Copy link
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
area:core-operators Operators, Sensors and hooks within Core Airflow type:bug-fix Changelog: Bug Fixes
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants