Fix config (#195)

Long wrapping of primitive causing NPE
GoogleCloudDataproc · Jan 9, 2025 · 299e842 · 299e842
1 parent 68792dd
commit 299e842
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/...igquery/src/main/java/com/google/cloud/flink/bigquery/sink/writer/CreateTableOptions.java b/...igquery/src/main/java/com/google/cloud/flink/bigquery/sink/writer/CreateTableOptions.java
@@ -26,7 +26,7 @@ public class CreateTableOptions {
     private final boolean enableTableCreation;
     private final String partitionField;
     private final TimePartitioning.Type partitionType;
-    private final Long partitionExpirationMillis;
+    private final long partitionExpirationMillis;
     private final List<String> clusteredFields;
     private final String region;
 
@@ -40,7 +40,11 @@ public CreateTableOptions(
         this.enableTableCreation = enableTableCreation;
         this.partitionField = partitionField;
         this.partitionType = partitionType;
-        this.partitionExpirationMillis = partitionExpirationMillis;
+        if (partitionExpirationMillis == null) {
+            this.partitionExpirationMillis = 0;
+        } else {
+            this.partitionExpirationMillis = partitionExpirationMillis;
+        }
         this.clusteredFields = clusteredFields;
         this.region = region;
     }
@@ -57,7 +61,7 @@ public TimePartitioning.Type getPartitionType() {
         return partitionType;
     }
 
-    public Long getPartitionExpirationMillis() {
+    public long getPartitionExpirationMillis() {
         return partitionExpirationMillis;
     }