From 74b1016d3b49987fda158ce46878ec94b730c7a4 Mon Sep 17 00:00:00 2001
From: Ryan Hill <ryanhill@microsoft.com>
Date: Fri, 7 May 2021 13:16:17 -0700
Subject: [PATCH] Fix cleanup on exit

---
 java/src/main/java/ai/onnxruntime/OnnxRuntime.java | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/java/src/main/java/ai/onnxruntime/OnnxRuntime.java b/java/src/main/java/ai/onnxruntime/OnnxRuntime.java
index 2771794260..8eee1fedc6 100644
--- a/java/src/main/java/ai/onnxruntime/OnnxRuntime.java
+++ b/java/src/main/java/ai/onnxruntime/OnnxRuntime.java
@@ -118,12 +118,12 @@ final class OnnxRuntime {
    *
    * @param file The file to remove.
    */
-  private static void cleanUp(File file) {
+  private static void cleanUp(File file, boolean on_exit_only=false) {
     if (!file.exists()) {
       return;
     }
     logger.log(Level.FINE, "Deleting " + file);
-    if (!file.delete()) {
+    if (on_exit_only || !file.delete()) {
       logger.log(Level.FINE, "Deleting " + file + " on exit");
       file.deleteOnExit();
     }
@@ -213,7 +213,7 @@ final class OnnxRuntime {
         logger.log(Level.FINE, "Loaded native library '" + library + "' from resource path");
       }
     } finally {
-      if (system_load) cleanUp(tempFile);
+      cleanUp(tempFile, !system_load);
     }
   }