Proper shutdown

app/src/main/kotlin/SimpleNotes.kt

@@ -1,13 +1,13 @@
 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() {
+fun main() = startKoin {
-    startKoin {
     modules(
         serverModule,
         persistanceModule,
@@ -21,5 +21,4 @@ fun main() {
         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>())