From 5f543d001fa9671f8dbfd5d2a9b812c14c828d9e Mon Sep 17 00:00:00 2001 From: Jacky Lee Date: Wed, 21 Sep 2022 09:12:19 +0800 Subject: [PATCH] delete tmp dir when application exit (#1115) --- .../core/src/main/java/com/intel/oap/vectorized/JniUtils.java | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/native-sql-engine/core/src/main/java/com/intel/oap/vectorized/JniUtils.java b/native-sql-engine/core/src/main/java/com/intel/oap/vectorized/JniUtils.java index 64e485195..565c1dff9 100644 --- a/native-sql-engine/core/src/main/java/com/intel/oap/vectorized/JniUtils.java +++ b/native-sql-engine/core/src/main/java/com/intel/oap/vectorized/JniUtils.java @@ -37,6 +37,8 @@ import java.util.jar.JarEntry; import java.util.jar.JarFile; +import org.apache.commons.io.FileUtils; + /** Helper class for JNI related operations. */ public class JniUtils { private static final String LIBRARY_NAME = "spark_columnar_jni"; @@ -80,6 +82,7 @@ private JniUtils(String _tmp_dir) throws IOException, IllegalAccessException, Il } else { Path folder = Paths.get(_tmp_dir); Path path = Files.createTempDirectory(folder, "spark_columnar_plugin_"); + FileUtils.forceDeleteOnExit(new File(path.toUri())); tmp_dir = path.toAbsolutePath().toString(); } } @@ -146,6 +149,7 @@ private static void loadLibraryFromJar(String source_jar, String tmp_dir) throws File tmp_dir_handler = new File(tmp_dir + "/tmp"); if (!tmp_dir_handler.exists()) { tmp_dir_handler.mkdirs(); + FileUtils.forceDeleteOnExit(tmp_dir_handler); } if (urlConnection instanceof JarURLConnection) {