Prevent a double close of fixed transitions HCV

Signed-off-by: Navin Kumar <[email protected]>
NVIDIA · Nov 21, 2023 · 255cb23 · 255cb23
1 parent 52ae251
commit 255cb23
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/main/java/com/nvidia/spark/rapids/jni/GpuTimeZoneDB.java b/src/main/java/com/nvidia/spark/rapids/jni/GpuTimeZoneDB.java
@@ -44,6 +44,8 @@ public class GpuTimeZoneDB {
   private CompletableFuture<Map<String, Integer>> zoneIdToTableFuture;
   private CompletableFuture<HostColumnVector> fixedTransitionsFuture;
 
+  private boolean closed = false;
+
   GpuTimeZoneDB() {
     zoneIdToTableFuture = new CompletableFuture<>();
     fixedTransitionsFuture = new CompletableFuture<>();
@@ -246,8 +248,14 @@ private void doLoadData() {
   }
 
   private void close() {
-    try (HostColumnVector hcv = getHostFixedTransitions()) {
-      // automatically closed
+    synchronized (this) {
+      if (closed) {
+        return;
+      }
+      try (HostColumnVector hcv = getHostFixedTransitions()) {
+        // automatically closed
+        closed = true;
+      }
     }
   }