From b51936445d36a76650d94225b3ecfa81b86e066c Mon Sep 17 00:00:00 2001
From: Charles7c <charles7c@126.com>
Date: Thu, 12 Dec 2024 21:16:58 +0800
Subject: [PATCH] =?UTF-8?q?feat(generator):=20=E4=BB=A3=E7=A0=81=E7=94=9F?=
 =?UTF-8?q?=E6=88=90=E6=96=B0=E5=A2=9E=20Mapper.xml=20=E6=A8=A1=E6=9D=BF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../generator/config/properties/GeneratorProperties.java  | 5 +++++
 .../generator/service/impl/GeneratorServiceImpl.java      | 8 +++-----
 .../src/main/resources/templates/backend/MapperXml.ftl    | 4 ++++
 .../src/main/resources/config/application-generator.yml   | 5 +++++
 4 files changed, 17 insertions(+), 5 deletions(-)
 create mode 100644 continew-plugin/continew-plugin-generator/src/main/resources/templates/backend/MapperXml.ftl

diff --git a/continew-plugin/continew-plugin-generator/src/main/java/top/continew/admin/generator/config/properties/GeneratorProperties.java b/continew-plugin/continew-plugin-generator/src/main/java/top/continew/admin/generator/config/properties/GeneratorProperties.java
index dbd489f8..a798f8e1 100644
--- a/continew-plugin/continew-plugin-generator/src/main/java/top/continew/admin/generator/config/properties/GeneratorProperties.java
+++ b/continew-plugin/continew-plugin-generator/src/main/java/top/continew/admin/generator/config/properties/GeneratorProperties.java
@@ -78,6 +78,11 @@ public class GeneratorProperties {
          */
         private String extension = FileNameUtil.EXT_JAVA;
 
+        /**
+         * 后缀
+         */
+        private String suffix;
+
         /**
          * 是否为后端模板
          */
diff --git a/continew-plugin/continew-plugin-generator/src/main/java/top/continew/admin/generator/service/impl/GeneratorServiceImpl.java b/continew-plugin/continew-plugin-generator/src/main/java/top/continew/admin/generator/service/impl/GeneratorServiceImpl.java
index db842ead..8b72e6d2 100644
--- a/continew-plugin/continew-plugin-generator/src/main/java/top/continew/admin/generator/service/impl/GeneratorServiceImpl.java
+++ b/continew-plugin/continew-plugin-generator/src/main/java/top/continew/admin/generator/service/impl/GeneratorServiceImpl.java
@@ -289,14 +289,12 @@ public class GeneratorServiceImpl implements GeneratorService {
         // 渲染代码
         String classNamePrefix = innerGenConfig.getClassNamePrefix();
         Map<String, GeneratorProperties.TemplateConfig> templateConfigMap = generatorProperties.getTemplateConfigs();
-
         TemplateEngine engine = TemplateUtil
             .createEngine(new TemplateConfig("templates", TemplateConfig.ResourceMode.CLASSPATH));
-        if (engine instanceof FreemarkerEngine) {
-            ((FreemarkerEngine)engine).getConfiguration()
+        if (engine instanceof FreemarkerEngine freemarkerEngine) {
+            freemarkerEngine.getConfiguration()
                 .setSharedVariable("statics", BeansWrapper.getDefaultInstance().getStaticModels());
         }
-
         for (Map.Entry<String, GeneratorProperties.TemplateConfig> templateConfigEntry : templateConfigMap.entrySet()) {
             GeneratorProperties.TemplateConfig templateConfig = templateConfigEntry.getValue();
             // 移除需要忽略的字段
@@ -309,7 +307,7 @@ public class GeneratorServiceImpl implements GeneratorService {
             // 处理其他配置
             innerGenConfig.setSubPackageName(templateConfig.getPackageName());
             String classNameSuffix = templateConfigEntry.getKey();
-            String className = classNamePrefix + classNameSuffix;
+            String className = classNamePrefix + StrUtil.blankToDefault(templateConfig.getSuffix(), classNameSuffix);
             innerGenConfig.setClassName(className);
             boolean isBackend = templateConfig.isBackend();
             String extension = templateConfig.getExtension();
diff --git a/continew-plugin/continew-plugin-generator/src/main/resources/templates/backend/MapperXml.ftl b/continew-plugin/continew-plugin-generator/src/main/resources/templates/backend/MapperXml.ftl
new file mode 100644
index 00000000..a4d66051
--- /dev/null
+++ b/continew-plugin/continew-plugin-generator/src/main/resources/templates/backend/MapperXml.ftl
@@ -0,0 +1,4 @@
+<?xml version="1.0" encoding="UTF-8" ?>
+<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd" >
+<mapper namespace="${packageName}.mapper.${classNamePrefix}Mapper">
+</mapper>
\ No newline at end of file
diff --git a/continew-webapi/src/main/resources/config/application-generator.yml b/continew-webapi/src/main/resources/config/application-generator.yml
index 7a808fad..e8e95307 100644
--- a/continew-webapi/src/main/resources/config/application-generator.yml
+++ b/continew-webapi/src/main/resources/config/application-generator.yml
@@ -100,6 +100,11 @@ generator:
     Mapper:
       templatePath: backend/Mapper.ftl
       packageName: mapper
+    MapperXml:
+      templatePath: backend/MapperXml.ftl
+      packageName: mapper
+      extension: .xml
+      suffix: Mapper
     Service:
       templatePath: backend/Service.ftl
       packageName: service