Fixes a regression related to BQ read transform upgrad via the Transf…

…ormService (#31685)
apache · Jun 26, 2024 · 90d3f8a · 90d3f8a
1 parent c1c853f
commit 90d3f8a
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/...oud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigquery/BigQueryIOTranslation.java b/...oud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigquery/BigQueryIOTranslation.java
@@ -274,10 +274,16 @@ public TypedRead<?> fromConfigRow(Row configRow, PipelineOptions options) {
         if (queryTempDataset != null) {
           builder = builder.setQueryTempDataset(queryTempDataset);
         }
-        String queryTempProject = configRow.getString("query_temp_project");
-        if (queryTempProject != null) {
-          builder = builder.setQueryTempProject(queryTempProject);
+
+        if (TransformUpgrader.compareVersions(updateCompatibilityBeamVersion, "2.57.0") >= 0) {
+          // This property was added for Beam 2.57.0 hence not available when
+          // upgrading the transform from previous Beam versions.
+          String queryTempProject = configRow.getString("query_temp_project");
+          if (queryTempProject != null) {
+            builder = builder.setQueryTempProject(queryTempProject);
+          }
         }
+
         byte[] methodBytes = configRow.getBytes("method");
         if (methodBytes != null) {
           builder = builder.setMethod((TypedRead.Method) fromByteArray(methodBytes));