@@ -2,11 +2,9 @@ package co.statu.rule.database
22
33import co.statu.parsek.api.ParsekPlugin
44import co.statu.parsek.api.config.PluginConfigManager
5- import co.statu.rule.database.api.DatabaseHelper
65import co.statu.rule.database.impl.SchemeVersionDaoImpl
76import co.statu.rule.database.model.SchemeVersion
87import io.vertx.core.Vertx
9- import io.vertx.core.json.JsonObject
108import io.vertx.jdbcclient.JDBCConnectOptions
119import io.vertx.jdbcclient.JDBCPool
1210import io.vertx.sqlclient.Pool
@@ -127,7 +125,6 @@ class DatabaseManager(
127125
128126 suspend fun initialize (
129127 plugin : ParsekPlugin ,
130- databaseHelper : DatabaseHelper ? = null
131128 ) {
132129 if (this .tables[plugin] == null ) {
133130 this .tables[plugin] = mutableListOf ()
@@ -137,8 +134,8 @@ class DatabaseManager(
137134 this .migrations[plugin] = mutableListOf ()
138135 }
139136
140- databaseHelper?.tables?. let { this .tables[plugin]!! .addAll(it) }
141- databaseHelper?.migrations?. let { this .migrations[plugin]!! .addAll(it) }
137+ this .tables[plugin] = plugin.pluginBeanContext.getBeansOfType( Dao :: class .java).values.map { it as Dao }.toMutableList()
138+ this .migrations[plugin] = plugin.pluginBeanContext.getBeansOfType( DatabaseMigration :: class .java).values.map { it as DatabaseMigration }.toMutableList()
142139
143140 val jdbcPool: Pool
144141
@@ -155,7 +152,7 @@ class DatabaseManager(
155152
156153 try {
157154 lastSchemeVersion = schemeVersionDaoImpl.getLastSchemeVersion(plugin.pluginId, jdbcPool)
158- } catch (e : BatchUpdateException ) {
155+ } catch (_ : BatchUpdateException ) {
159156 try {
160157 if (plugin is DatabasePlugin ) {
161158 logger.warn(" First time installing scheme version (first time app installation)" )
@@ -183,9 +180,9 @@ class DatabaseManager(
183180 checkMigration(plugin, jdbcPool, lastSchemeVersion)
184181 }
185182
186- suspend fun migrateNewPluginId (exPluginId : String , newPluginId : String , plugin : ParsekPlugin ) {
183+ suspend fun migrateNewPluginId (exPluginId : String , plugin : ParsekPlugin ) {
187184 try {
188- schemeVersionDaoImpl.renamePluginId(exPluginId, newPluginId , getConnectionPool())
185+ schemeVersionDaoImpl.renamePluginId(exPluginId, plugin.pluginId , getConnectionPool())
189186 } catch (e: Exception ) {
190187 if (plugin !is DatabasePlugin ) {
191188 logger.error(e.message)
0 commit comments