fix: drop temp files after request

JetBrains · Dec 15, 2024 · 49f1e22 · 49f1e22
1 parent d410778
commit 49f1e22
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/build.gradle.kts b/build.gradle.kts
@@ -142,7 +142,13 @@ fun buildRunFile() {
             /* language=sh */
             """
                 #!/usr/bin/env sh
-                exec java -noverify -XX:+UseParallelGC -XX:-UseCompressedOops -cp "./:lib/*" "com.compiler.server.CompilerApplicationKt"
+                
+                export COMPILER_SERVER_TMP_PATH="${'$'}(mktemp -d)"
+                
+                exec java -noverify \
+                    -XX:+UseParallelGC -XX:-UseCompressedOops \
+                    -Djava.io.tmpdir="${'$'}COMPILER_SERVER_TMP_PATH" \
+                    -cp "./:lib/*" "com.compiler.server.CompilerApplicationKt"
             """.trimIndent()
         )
     }

diff --git a/src/main/kotlin/com/compiler/server/compiler/components/CliUtils.kt b/src/main/kotlin/com/compiler/server/compiler/components/CliUtils.kt
@@ -119,6 +119,15 @@ fun <T> usingTempDirectory(action: (path: Path) -> T): T {
     action(path)
   } finally {
     path.deleteRecursively()
+
+    System.getenv("COMPILER_SERVER_TMP_PATH")?.let {
+      val directory = File(it)
+      if (directory.exists() && directory.isDirectory) {
+        directory.listFiles()?.forEach { file ->
+          file.deleteRecursively()
+        }
+      }
+    }
   }
 }