Fix Alluxio CacheManager leak on coordinator

trinodb · May 15, 2024 · 870c0bb · 870c0bb
1 parent 212455d
commit 870c0bb
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/...-alluxio/src/main/java/io/trino/filesystem/alluxio/AlluxioCoordinatorFileSystemCache.java b/...-alluxio/src/main/java/io/trino/filesystem/alluxio/AlluxioCoordinatorFileSystemCache.java
@@ -25,6 +25,7 @@
 import io.trino.filesystem.TrinoInputStream;
 import io.trino.filesystem.cache.AllowFilesystemCacheOnCoordinator;
 import io.trino.filesystem.cache.TrinoFileSystemCache;
+import jakarta.annotation.PreDestroy;
 
 import java.io.IOException;
 import java.util.Optional;
@@ -64,6 +65,15 @@ public AlluxioCoordinatorFileSystemCache(
         }
     }
 
+    @PreDestroy
+    public void shutdown()
+            throws Exception
+    {
+        if (alluxioFileSystemCache.isPresent()) {
+            alluxioFileSystemCache.get().shutdown();
+        }
+    }
+
     @Override
     public TrinoInput cacheInput(TrinoInputFile delegate, String key)
             throws IOException