Make the GpuTimeZoneDB class idempotent, such that when it is shutdow…

…n, it can be recovered and useable again (#1670)
NVIDIA · Jan 2, 2024 · 4358580 · 4358580
1 parent 38e503c
commit 4358580
Showing 1 changed file with 2 additions and 0 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
@@ -90,6 +90,8 @@ public Thread newThread(Runnable r) {
   public static void shutdown() {
     if (instance.isLoaded()) {
       instance.close();
+      // Recreate a new instance to reload the database if necessary
+      instance = new GpuTimeZoneDB();
     }
   }