Proper shutdown

app/src/main/kotlin/SimpleNotes.kt

@@ -1,25 +1,24 @@
 package be.simplenotes.app
 
+import be.simplenotes.app.extensions.addShutdownHook
 import be.simplenotes.app.modules.*
 import be.simplenotes.domain.domainModule
 import be.simplenotes.persistance.persistanceModule
 import be.simplenotes.search.searchModule
 import org.koin.core.context.startKoin
 
-fun main() {
-    startKoin {
-        modules(
-            serverModule,
-            persistanceModule,
-            configModule,
-            baseModule,
-            userModule,
-            noteModule,
-            settingsModule,
-            domainModule,
-            searchModule,
-            apiModule,
-            jsonModule
-        )
-    }
-}
+fun main() = startKoin {
+    modules(
+        serverModule,
+        persistanceModule,
+        configModule,
+        baseModule,
+        userModule,
+        noteModule,
+        settingsModule,
+        domainModule,
+        searchModule,
+        apiModule,
+        jsonModule
+    )
+}.addShutdownHook()

app/src/main/kotlin/extensions/KoinExtensions.kt

@@ -0,0 +1,12 @@
+package be.simplenotes.app.extensions
+
+import org.koin.core.KoinApplication
+import kotlin.concurrent.thread
+
+fun KoinApplication.addShutdownHook() {
+    Runtime.getRuntime().addShutdownHook(
+        thread(start = false) {
+            close()
+        }
+    )
+}

app/src/main/kotlin/modules/ServerModule.kt

@@ -18,10 +18,11 @@ import org.http4k.server.asServer
 import org.koin.core.qualifier.named
 import org.koin.core.qualifier.qualifier
 import org.koin.dsl.module
+import org.koin.dsl.onClose
 import org.http4k.server.ServerConfig as Http4kServerConfig
 
 val serverModule = module {
-    single(createdAtStart = true) { Server(get(), get()).start() }
+    single(createdAtStart = true) { Server(get(), get()).start() } onClose { it?.stop() }
     single { get<RoutingHttpHandler>().asServer(get()) }
     single<Http4kServerConfig> {
         val config = get<ServerConfig>()