Skip to content

fix OpenLineage extraction for GCP deferrable operators#40521

Merged
potiuk merged 1 commit intoapache:mainfrom
kacpermuda:fix-ol-gcp-deferrable
Jul 1, 2024
Merged

fix OpenLineage extraction for GCP deferrable operators#40521
potiuk merged 1 commit intoapache:mainfrom
kacpermuda:fix-ol-gcp-deferrable

Conversation

@kacpermuda
Copy link
Contributor

After being deferred in case of some operators, OpenLineage methods do not have access to job_id so we need to reassign it in execute_complete() and also adjust OL methods to create any other attributes that might be missing after deferral. Similar to #40457

Also removing unnecessary private self._job_id from BigQueryToGCSOperator, i created it some time ago but it duplicates public self.job_id and it's used only by OpenLineage, so we safely can remove it.


^ 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 area:providers provider:google Google (including GCP) related issues labels Jul 1, 2024
@kacpermuda kacpermuda force-pushed the fix-ol-gcp-deferrable branch from 7c86a02 to 76acff0 Compare July 1, 2024 15:23
Signed-off-by: Kacper Muda <mudakacper@gmail.com>
@kacpermuda kacpermuda force-pushed the fix-ol-gcp-deferrable branch from 76acff0 to c523cba Compare July 1, 2024 15:46
@potiuk potiuk merged commit 47e7e25 into apache:main Jul 1, 2024
@kacpermuda kacpermuda deleted the fix-ol-gcp-deferrable branch July 1, 2024 19:00
romsharon98 pushed a commit to romsharon98/airflow that referenced this pull request Jul 26, 2024
Signed-off-by: Kacper Muda <mudakacper@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:providers provider:google Google (including GCP) related issues

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants