From eaeca63f3153a141027ec1d75d1e92b53b38817a Mon Sep 17 00:00:00 2001
From: AiKrai <aikrai001@gmail.com>
Date: Thu, 20 Mar 2025 10:57:14 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8F=82=E8=80=83ebean=E5=89=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 vertx-demo/src/test/kotlin/app/GenerateMigration.kt          | 5 +----
 .../org/aikrai/vertx/db/migration/SqlMigrationGenerator.kt   | 4 +++-
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/vertx-demo/src/test/kotlin/app/GenerateMigration.kt b/vertx-demo/src/test/kotlin/app/GenerateMigration.kt
index 93d7cb8..cacb6a1 100644
--- a/vertx-demo/src/test/kotlin/app/GenerateMigration.kt
+++ b/vertx-demo/src/test/kotlin/app/GenerateMigration.kt
@@ -66,11 +66,8 @@ object GenerateMigration {
     // 创建注解映射器
     val mapper = createMapper()
 
-    // 扫描实体类
-    val entityClasses = scanEntityClasses(entityPackage)
-
     // 生成迁移文件
-    SqlMigrationGenerator.generateMigrations(entityClasses, mapper)
+    SqlMigrationGenerator.generateMigrations(entityPackage, mapper)
 
     println("数据库迁移文件生成完成")
   }
diff --git a/vertx-fw/src/main/kotlin/org/aikrai/vertx/db/migration/SqlMigrationGenerator.kt b/vertx-fw/src/main/kotlin/org/aikrai/vertx/db/migration/SqlMigrationGenerator.kt
index a1d276f..775fe90 100644
--- a/vertx-fw/src/main/kotlin/org/aikrai/vertx/db/migration/SqlMigrationGenerator.kt
+++ b/vertx-fw/src/main/kotlin/org/aikrai/vertx/db/migration/SqlMigrationGenerator.kt
@@ -34,7 +34,9 @@ class SqlMigrationGenerator {
          * @param entityClasses 实体类列表
          * @param mapper 注解映射中间类
          */
-        fun generateMigrations(entityClasses: List<KClass<*>>, mapper: SqlAnnotationMapper) {
+        fun generateMigrations(entityPackage: String, mapper: SqlAnnotationMapper) {
+            val entityClasses = scanEntityClasses(entityPackage)
+
             // 创建必要的目录
             createDirectories()