Run migrations on start

Heapy · Feb 6, 2024 · 2dad462 · 2dad462
1 parent 6af00a5
commit 2dad462
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/tgkotbot-dataops/src/main/kotlin/RunMigrations.kt b/tgkotbot-dataops/src/main/kotlin/RunMigrations.kt
@@ -1,6 +1,7 @@
 @file:JvmName("RunMigrations")
 
 import org.flywaydb.core.Flyway
+import javax.sql.DataSource
 
 fun main() {
     val env = System.getenv()
@@ -23,3 +24,15 @@ fun main() {
         .load()
         .migrate()
 }
+
+fun runMigrations(
+    dataSource: DataSource,
+) {
+    Flyway
+        .configure()
+        .locations("classpath:migrations")
+        .dataSource(dataSource)
+        .loggers("slf4j")
+        .load()
+        .migrate()
+}
diff --git a/tgkotbot/build.gradle.kts b/tgkotbot/build.gradle.kts
@@ -18,6 +18,7 @@ tasks.distZip {
 dependencies {
     implementation(projects.core)
     implementation(projects.tgkotbotDatabase)
+    implementation(projects.tgkotbotDataops)
 
     implementation(libs.kotlin.stdlib)
     implementation(libs.kotlin.coroutines)

diff --git a/tgkotbot/src/main/kotlin/io/heapy/kotbot/ApplicationFactory.kt b/tgkotbot/src/main/kotlin/io/heapy/kotbot/ApplicationFactory.kt
@@ -30,6 +30,7 @@ import org.jooq.DSLContext
 import org.jooq.SQLDialect
 import org.jooq.impl.DSL
 import org.postgresql.ds.PGSimpleDataSource
+import runMigrations
 import java.lang.management.ManagementFactory
 import kotlin.concurrent.thread
 
@@ -212,6 +213,7 @@ open class ApplicationFactory {
     }
 
     open suspend fun start() {
+        runMigrations(hikariDataSource)
         Runtime.getRuntime().addShutdownHook(thread(start = false) {
             log.info("Shutdown hook called.")
             applicationScope.cancel("Shutdown hook called.")