Implemented handler for SIGTERM

taxfix · Jun 19, 2019 · b97c27d · b97c27d
1 parent a31afbf
commit b97c27d
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/main/scala/taxfix/graknsupervisor/GraknSupervisor.scala b/src/main/scala/taxfix/graknsupervisor/GraknSupervisor.scala
@@ -29,11 +29,14 @@ object GraknSupervisor extends App {
   var killed = false
 
   // Setup the callback for exit signal to shutdown Grakn gracefully
-  Signal.handle(new Signal("INT"), (_: Signal) => {
+  Signal.handle(new Signal("INT"), (_: Signal) => kill())
+  Signal.handle(new Signal("TERM"), (_: Signal) => kill())
+
+  def kill() = {
     killed = true
     log.log("supervisor", "message", s"Received kill signal! Stopping all Grakn processes")
     handleSupervisorExit()
-  })
+  }
 
 
   val storageExitFuture = Future {