Avoid NPE warning.

tvalentyn · tvalentyn · commit 7ebc3819c176 · 2025-07-23T20:31:29.000+02:00
diff --git a/sdks/java/io/jdbc/src/main/java/org/apache/beam/sdk/io/jdbc/JdbcIO.java b/sdks/java/io/jdbc/src/main/java/org/apache/beam/sdk/io/jdbc/JdbcIO.java
@@ -2196,9 +2196,10 @@ public PCollection<V> expand(PCollection<T> input) {
           autoSharding == null || (autoSharding && input.isBounded() != IsBounded.UNBOUNDED),
           "Autosharding is only supported for streaming pipelines.");
 
-      long batchSize = getBatchSize() == null ? DEFAULT_BATCH_SIZE : getBatchSize();
-      long maxBufferingDuration =
-          getMaxBatchBufferingDuration() == null
+      Long batchSizeAsLong = getBatchSize();
+      long batchSize = batchSizeAsLong == null ? DEFAULT_BATCH_SIZE : batchSizeAsLong;
+      Long maxBufferingDurationAsLong = getMaxBatchBufferingDuration();
+      long maxBufferingDuration = maxBufferingDurationAsLong == null
               ? DEFAULT_MAX_BATCH_BUFFERING_DURATION
               : getMaxBatchBufferingDuration();
 
@@ -2215,7 +2216,7 @@ public PCollection<V> expand(PCollection<T> input) {
                       .setStatement(getStatement())
                       .setRetryConfiguration(getRetryConfiguration())
                       .setReturnResults(true)
-                      .setBatchSize(1L)  // We are writing iterables 1 at a time.
+                      .setBatchSize(1L) // We are writing iterables 1 at a time.
                       .setMaxBatchBufferingDuration(maxBufferingDuration)
                       .build())));
     }
