feat: add migrating new plugin id

duruer · duruer · commit 0cbb4749e94a · 2024-02-22T23:09:22.000+03:00
diff --git a/src/main/kotlin/co/statu/rule/database/DatabaseManager.kt b/src/main/kotlin/co/statu/rule/database/DatabaseManager.kt
@@ -166,6 +166,17 @@ class DatabaseManager(
         checkMigration(plugin, jdbcPool, lastSchemeVersion)
     }
 
+    suspend fun migrateNewPluginId(exPluginId: String, newPluginId: String, plugin: ParsekPlugin) {
+        try {
+            schemeVersionDaoImpl.renamePluginId(exPluginId, newPluginId, getConnectionPool())
+        } catch (e: Exception) {
+            if (plugin !is DatabasePlugin) {
+                logger.error(e.message)
+                exitProcess(1)
+            }
+        }
+    }
+
     internal fun getLatestMigration(plugin: ParsekPlugin) = migrations[plugin]?.maxByOrNull { it.SCHEME_VERSION }
 
     suspend fun checkMigration(plugin: ParsekPlugin, jdbcPool: JDBCPool, lastSchemeVersion: SchemeVersion?) {
diff --git a/src/main/kotlin/co/statu/rule/database/ParsekEventHandler.kt b/src/main/kotlin/co/statu/rule/database/ParsekEventHandler.kt
@@ -26,6 +26,12 @@ class ParsekEventHandler : ParsekEventListener {
             logger
         )
 
+        databaseManager.migrateNewPluginId(
+            "database",
+            DatabasePlugin.INSTANCE.context.pluginId,
+            DatabasePlugin.INSTANCE
+        )
+
         databaseManager.initialize(DatabasePlugin.INSTANCE, tables, migrations)
 
         val handlers = DatabasePlugin.INSTANCE.context.pluginEventManager.getEventHandlers<DatabaseEventListener>()
diff --git a/src/main/kotlin/co/statu/rule/database/dao/SchemeVersionDao.kt b/src/main/kotlin/co/statu/rule/database/dao/SchemeVersionDao.kt
@@ -14,4 +14,10 @@ abstract class SchemeVersionDao : Dao<SchemeVersion>(SchemeVersion::class) {
         pluginId: String,
         jdbcPool: JDBCPool
     ): SchemeVersion?
+
+    abstract suspend fun renamePluginId(
+        exPluginId: String,
+        newPluginId: String,
+        jdbcPool: JDBCPool
+    )
 }
diff --git a/src/main/kotlin/co/statu/rule/database/impl/SchemeVersionDaoImpl.kt b/src/main/kotlin/co/statu/rule/database/impl/SchemeVersionDaoImpl.kt
@@ -63,4 +63,18 @@ class SchemeVersionDaoImpl : SchemeVersionDao() {
 
         return row.toEntity()
     }
+
+    override suspend fun renamePluginId(
+        exPluginId: String,
+        newPluginId: String,
+        jdbcPool: JDBCPool
+    ) {
+        val query =
+            "ALTER TABLE `${getTablePrefix() + tableName}` UPDATE `pluginId` = ? WHERE `pluginId` = ?;"
+
+        jdbcPool
+            .preparedQuery(query)
+            .execute(Tuple.of(newPluginId, exPluginId))
+            .await()
+    }
 }
