#129 remove temporary file when the job ends

TheHive-Project · Sep 12, 2018 · 60956ae · 60956ae
1 parent 334feb1
commit 60956ae
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/app/org/thp/cortex/services/JobSrv.scala b/app/org/thp/cortex/services/JobSrv.scala
@@ -1,10 +1,10 @@
 package org.thp.cortex.services
 
 import java.io.{ ByteArrayOutputStream, InputStream }
-import java.nio.file.Files
+import java.nio.file.{ Files, Paths }
 import java.util.Date
-import javax.inject.{ Inject, Singleton }
 
+import javax.inject.{ Inject, Singleton }
 import akka.NotUsed
 import akka.actor.ActorSystem
 import akka.stream.Materializer
@@ -403,6 +403,11 @@ class JobSrv(
             val errorMessage = (error + output).take(8192)
             endJob(job, JobStatus.Failure, Some(s"Invalid output\n$errorMessage"))
         }
+        finally {
+          (input \ "file").asOpt[String].foreach { filename ⇒
+            Files.deleteIfExists(Paths.get(filename))
+          }
+        }
       }(executionContext)
   }