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>()

deploy-docker-hub.sh

@@ -5,4 +5,4 @@ rm app/src/main/resources/static/styles*
 
 yarn --cwd css run css-purge \
     && docker build -t hubv/simplenotes:latest . \
-    # && docker push hubv/simplenotes:latest
+    && docker push hubv/simplenotes:latest

persistance/src/main/kotlin/PersistanceModule.kt

@@ -7,8 +7,10 @@ import be.simplenotes.persistance.users.UserRepositoryImpl
 import be.simplenotes.shared.config.DataSourceConfig
 import com.zaxxer.hikari.HikariConfig
 import com.zaxxer.hikari.HikariDataSource
-import me.liuwj.ktorm.database.*
+import me.liuwj.ktorm.database.Database
+import org.koin.dsl.bind
 import org.koin.dsl.module
+import org.koin.dsl.onClose
 import javax.sql.DataSource
 
 interface DbMigrations {
@@ -31,7 +33,7 @@ val persistanceModule = module {
     single<UserRepository> { UserRepositoryImpl(get()) }
     single<NoteRepository> { NoteRepositoryImpl(get()) }
     single<DbMigrations> { DbMigrationsImpl(get(), get()) }
-    single<DataSource> { hikariDataSource(get()) }
+    single { hikariDataSource(get()) } bind DataSource::class onClose { it?.close() }
     single {
         get<DbMigrations>().migrate()
         Database.connect(get<DataSource>())