feat: ApplicationEngine shutdown hooks

illumi-engineering · Nov 11, 2024 · 28d2513 · 28d2513
1 parent f6df411
commit 28d2513
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/core/src/main/kotlin/sh/illumi/kraft/engine/ApplicationEngine.kt b/core/src/main/kotlin/sh/illumi/kraft/engine/ApplicationEngine.kt
@@ -18,6 +18,12 @@ import sh.illumi.kraft.util.argsMatchParams
 abstract class ApplicationEngine {
     lateinit var rootLayer: ApplicationLayer private set
 
+    fun registerShutdownHook() {
+        Runtime.getRuntime().addShutdownHook(Thread {
+            rootLayer.shutdown()
+        })
+    }
+
     fun <TLayer : ApplicationLayer> startRoot(createRoot: suspend CoroutineScope.() -> ApplicationLayer) = runBlocking {
         rootLayer = createRoot()
         rootLayer.start()