From 4450c224c74ea6aa2e8ee8c44c2a7138a068d792 Mon Sep 17 00:00:00 2001
From: ToyCat <494355437@qq.com>
Date: Tue, 25 Apr 2023 17:37:10 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=20processor.tablesClassName?=
 =?UTF-8?q?=20=E9=85=8D=E7=BD=AE=E4=B8=8D=E8=B5=B7=E4=BD=9C=E7=94=A8?=
 =?UTF-8?q?=E7=9A=84=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../com/mybatisflex/processor/QueryEntityProcessor.java    | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/mybatis-flex-annotation/src/main/java/com/mybatisflex/processor/QueryEntityProcessor.java b/mybatis-flex-annotation/src/main/java/com/mybatisflex/processor/QueryEntityProcessor.java
index 1ff03001..48904c38 100644
--- a/mybatis-flex-annotation/src/main/java/com/mybatisflex/processor/QueryEntityProcessor.java
+++ b/mybatis-flex-annotation/src/main/java/com/mybatisflex/processor/QueryEntityProcessor.java
@@ -75,7 +75,7 @@ public class QueryEntityProcessor extends AbstractProcessor {
             "import com.mybatisflex.core.table.TableDef;\n" +
             "\n" +
             "// Auto generate by mybatis-flex, do not modify it.\n" +
-            "public class Tables {\n" +
+            "public class @tablesClassName {\n" +
             "@classesInfo" +
             "}\n";
 
@@ -123,7 +123,7 @@ public class QueryEntityProcessor extends AbstractProcessor {
             if ("false".equalsIgnoreCase(enable)) {
                 return true;
             }
-            
+
             String genPath = props.getProperties().getProperty("processor.genPath", "");
             String genTablesPackage = props.getProperties().getProperty("processor.tablesPackage");
             String baseMapperClass = props.getProperties().getProperty("processor.baseMapperClass", "com.mybatisflex.core.BaseMapper");
@@ -358,7 +358,8 @@ public class QueryEntityProcessor extends AbstractProcessor {
 
     private void genTablesClass(String genBasePath, String genPackageName, String className, String classContent) {
         String genContent = classTableTemplate.replace("@package", genPackageName)
-                .replace("@classesInfo", classContent);
+                .replace("@classesInfo", classContent)
+                .replace("@tablesClassName", className);
 
         Writer writer = null;
         try {