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

feat: Add OpenLineage support for CloudSQLExecuteQueryOperator and some SQLtoGCS transfer operators #45182

Draft
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

kacpermuda
Copy link
Contributor

@kacpermuda kacpermuda commented Dec 23, 2024

This PR adds OpenLineage support for CloudSQLExecuteQueryOperator and some SQLtoGCS transfer operators. The logic is very similar (almost the same) as in SQLExecuteQueryOperator where we take advantage of DB hooks being instrumented. I'll add some unit tests shortly

I do however have some troubles with testing this change in real life. I wanted to run system tests for CloudSQLExecuteQueryOperator, but they seem to fail even on current main (breeze testing system-tests providers/tests/system/google/cloud/cloud_sql/example_cloud_sql_query.py). Also, the system test dashboard provided by Google is not available, so I can't verify if it fails only for me. Would appreciate any help in running those tests


^ 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 Dec 23, 2024
@kacpermuda kacpermuda force-pushed the feat-ol-google-cloud-sql-operator branch 2 times, most recently from c004232 to 74aa67e Compare December 24, 2024 08:54
…me SQLtoGCS transfer operators

Signed-off-by: Kacper Muda <[email protected]>
@kacpermuda kacpermuda force-pushed the feat-ol-google-cloud-sql-operator branch from 74aa67e to 2782f2f Compare December 24, 2024 10:05
@kacpermuda kacpermuda changed the title feat: Add OpenLineage support for CloudSQLExecuteQueryOperator feat: Add OpenLineage support for CloudSQLExecuteQueryOperator and some SQLtoGCS transfer operators Dec 24, 2024
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.

1 participant