From 0d95a5ff7f9020467eda850ab84f881caa6569c3 Mon Sep 17 00:00:00 2001
From: bunny <1319900154@qq.com>
Date: Tue, 8 Jul 2025 20:33:22 +0800
Subject: [PATCH] =?UTF-8?q?:ambulance:=20=E5=90=8E=E7=AB=AF=E6=96=87?=
 =?UTF-8?q?=E4=BB=B6=E7=94=9F=E6=88=90=E6=8A=A5=E9=94=99?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../core/template/VmsTBaseTemplateGenerator.java    | 13 ++++++++-----
 src/main/resources/application.yml                  |  5 +++++
 2 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/src/main/java/cn/bunny/core/template/VmsTBaseTemplateGenerator.java b/src/main/java/cn/bunny/core/template/VmsTBaseTemplateGenerator.java
index 6322f6f..1e69879 100644
--- a/src/main/java/cn/bunny/core/template/VmsTBaseTemplateGenerator.java
+++ b/src/main/java/cn/bunny/core/template/VmsTBaseTemplateGenerator.java
@@ -32,6 +32,9 @@ public class VmsTBaseTemplateGenerator extends AbstractTemplateGenerator {
         // 处理表名称，替换前缀
         String tableName = tableMetaData.getTableName();
         String[] prefixes = dto.getTablePrefixes().split("[,，]");
+
+        tableMetaData.setCleanTableName(tableName);
+
         for (String prefix : prefixes) {
             if (tableName.startsWith(prefix)) {
                 String handlerTableName = tableName.replace(prefix, "");
@@ -42,7 +45,6 @@ public class VmsTBaseTemplateGenerator extends AbstractTemplateGenerator {
         this.dto = dto;
         this.path = path;
         this.tableMetaData = tableMetaData;
-
     }
 
     /**
@@ -53,8 +55,9 @@ public class VmsTBaseTemplateGenerator extends AbstractTemplateGenerator {
     @Override
     public void addContext(VelocityContext context) {
         // 当前的表名
-        String handlerTableName = tableMetaData.getCleanTableName();
-        handlerTableName = StringUtils.hasText(handlerTableName) ? handlerTableName : tableMetaData.getTableName();
+        String cleanTableName = tableMetaData.getCleanTableName();
+        cleanTableName = StringUtils.hasText(cleanTableName) ? cleanTableName : tableMetaData.getTableName();
+
         // 表的注释内容
         String comment = tableMetaData.getComment();
 
@@ -75,11 +78,11 @@ public class VmsTBaseTemplateGenerator extends AbstractTemplateGenerator {
         context.put("package", dto.getPackageName());
 
         // 将类名称转成小驼峰
-        String lowerCamelCase = MysqlTypeConvertUtil.convertToCamelCase(handlerTableName, false);
+        String lowerCamelCase = MysqlTypeConvertUtil.convertToCamelCase(cleanTableName, false);
         context.put("classLowercaseName", lowerCamelCase);
 
         // 将类名称转成大驼峰
-        String upperCameCase = MysqlTypeConvertUtil.convertToCamelCase(handlerTableName, true);
+        String upperCameCase = MysqlTypeConvertUtil.convertToCamelCase(cleanTableName, true);
         context.put("classUppercaseName", upperCameCase);
 
         // 添加中划线
diff --git a/src/main/resources/application.yml b/src/main/resources/application.yml
index 0275386..e8881c6 100644
--- a/src/main/resources/application.yml
+++ b/src/main/resources/application.yml
@@ -6,6 +6,11 @@ spring:
     active: prod
   application:
     name: generator-code
+  devtools:
+    livereload:
+      enabled: true
+      port: 8880
+
   thymeleaf:
     check-template-location: false