diff --git a/airflow/providers/google/cloud/example_dags/example_cloud_sql.py b/airflow/providers/google/cloud/example_dags/example_cloud_sql.py index 304201d8ca43d..48c2b64d26a53 100644 --- a/airflow/providers/google/cloud/example_dags/example_cloud_sql.py +++ b/airflow/providers/google/cloud/example_dags/example_cloud_sql.py @@ -123,7 +123,12 @@ # [END howto_operator_cloudsql_patch_body] # [START howto_operator_cloudsql_export_body] export_body = { - "exportContext": {"fileType": "sql", "uri": EXPORT_URI, "sqlExportOptions": {"schemaOnly": False}} + "exportContext": { + "fileType": "sql", + "uri": EXPORT_URI, + "sqlExportOptions": {"schemaOnly": False}, + "offload": True, + } } # [END howto_operator_cloudsql_export_body] # [START howto_operator_cloudsql_import_body] diff --git a/airflow/providers/google/cloud/operators/cloud_sql.py b/airflow/providers/google/cloud/operators/cloud_sql.py index 1441f518b46b4..c2ead7dc88c3d 100644 --- a/airflow/providers/google/cloud/operators/cloud_sql.py +++ b/airflow/providers/google/cloud/operators/cloud_sql.py @@ -154,6 +154,7 @@ ], ), dict(name="csvExportOptions", type="dict", optional=True, fields=[dict(name="selectQuery")]), + dict(name="offload", optional=True), ], ) ] diff --git a/tests/providers/google/cloud/operators/test_cloud_sql.py b/tests/providers/google/cloud/operators/test_cloud_sql.py index 47352d7b4ae3f..fe6a6d92a5028 100644 --- a/tests/providers/google/cloud/operators/test_cloud_sql.py +++ b/tests/providers/google/cloud/operators/test_cloud_sql.py @@ -123,6 +123,7 @@ "databases": [], "sqlExportOptions": {"tables": ["table1", "table2"], "schemaOnly": False}, "csvExportOptions": {"selectQuery": "SELECT * FROM TABLE"}, + "offload": True, } } IMPORT_BODY = {