feat(generator): 代码生成新增 Mapper.xml 模板

continew-org · Dec 12, 2024 · b519364 · b519364
1 parent a7296a3
commit b519364
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 5 deletions.
diff --git a/...tor/src/main/java/top/continew/admin/generator/config/properties/GeneratorProperties.java b/...tor/src/main/java/top/continew/admin/generator/config/properties/GeneratorProperties.java
@@ -78,6 +78,11 @@ public static class TemplateConfig {
          */
         private String extension = FileNameUtil.EXT_JAVA;
 
+        /**
+         * 后缀
+         */
+        private String suffix;
+
         /**
          * 是否为后端模板
          */

diff --git a/...nerator/src/main/java/top/continew/admin/generator/service/impl/GeneratorServiceImpl.java b/...nerator/src/main/java/top/continew/admin/generator/service/impl/GeneratorServiceImpl.java
@@ -289,14 +289,12 @@ private List<GeneratePreviewResp> preview(String tableName) {
         // 渲染代码
         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 @@ private List<GeneratePreviewResp> preview(String tableName) {
             // 处理其他配置
             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
@@ -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>
diff --git 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