From bfa8632c1b847f9c753fc17de52895ceb4518e0b Mon Sep 17 00:00:00 2001
From: bunny <1319900154@qq.com>
Date: Sat, 5 Jul 2025 19:14:42 +0800
Subject: [PATCH] =?UTF-8?q?:bug:=20=E4=BF=AE=E6=94=B9=E4=BB=A3=E7=A0=81?=
 =?UTF-8?q?=E7=94=9F=E6=88=90=E5=99=A8=E5=8E=BB=E9=99=A4=E8=A1=A8=E5=89=8D?=
 =?UTF-8?q?=E7=BC=80=E4=B8=BA=E7=A9=BA=E6=83=85=E5=86=B5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../core/template/VmsTBaseTemplateGenerator.java      | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/generator-code-server/src/main/java/cn/bunny/core/template/VmsTBaseTemplateGenerator.java b/generator-code-server/src/main/java/cn/bunny/core/template/VmsTBaseTemplateGenerator.java
index f956944..e1563c9 100644
--- a/generator-code-server/src/main/java/cn/bunny/core/template/VmsTBaseTemplateGenerator.java
+++ b/generator-code-server/src/main/java/cn/bunny/core/template/VmsTBaseTemplateGenerator.java
@@ -7,6 +7,7 @@ import com.google.common.base.CaseFormat;
 import org.apache.velocity.Template;
 import org.apache.velocity.VelocityContext;
 import org.apache.velocity.app.Velocity;
+import org.springframework.util.StringUtils;
 
 import java.io.StringWriter;
 import java.text.SimpleDateFormat;
@@ -28,10 +29,6 @@ public class VmsTBaseTemplateGenerator extends AbstractTemplateGenerator {
      * @param tableMetaData 表名称
      */
     public VmsTBaseTemplateGenerator(VmsArgumentDto dto, String path, TableMetaData tableMetaData) {
-        this.dto = dto;
-        this.path = path;
-        this.tableMetaData = tableMetaData;
-
         // 处理表名称，替换前缀
         String tableName = tableMetaData.getTableName();
         String[] prefixes = dto.getTablePrefixes().split("[,，]");
@@ -41,6 +38,10 @@ public class VmsTBaseTemplateGenerator extends AbstractTemplateGenerator {
                 tableMetaData.setCleanTableName(handlerTableName);
             }
         }
+
+        this.dto = dto;
+        this.path = path;
+        this.tableMetaData = tableMetaData;
     }
 
     /**
@@ -52,6 +53,8 @@ public class VmsTBaseTemplateGenerator extends AbstractTemplateGenerator {
     public void addContext(VelocityContext context) {
         // 当前的表名
         String handlerTableName = tableMetaData.getCleanTableName();
+        handlerTableName = StringUtils.hasText(handlerTableName) ? handlerTableName : tableMetaData.getTableName();
+        
         // 表的注释内容
         String comment = tableMetaData.getComment();