Add mybatis flex mapper.xml

tangllty · May 29, 2024 · be6624a · be6624a
1 parent b91f79c
commit be6624a
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 2 deletions.
diff --git a/tang-tool/tang-generator/src/main/java/com/tang/generator/utils/VelocityUtils.java b/tang-tool/tang-generator/src/main/java/com/tang/generator/utils/VelocityUtils.java
@@ -45,6 +45,7 @@
 import static com.tang.generator.constants.VmTemplate.VM_VUE_INDEX_TS;
 import static com.tang.generator.constants.VmTemplate.VM_VUE_TYPES;
 import static com.tang.generator.constants.VmTemplate.VM_XML_MYBATIS_MAPPER;
+import static com.tang.generator.constants.VmTemplate.VM_XML_MYBATIS_FLEX_MAPPER;
 import static com.tang.generator.constants.VmTemplate.VM_XML_MYBATIS_PLUS_MAPPER;
 
 /**
@@ -236,7 +237,7 @@ private static List<String> getJavaTemplateList(LanguageType languageType, OrmTy
         return switch (languageType) {
             case JAVA -> switch (ormType) {
                 case MYBATIS -> List.of(VM_JAVA_MYBATIS_ENTITY, VM_JAVA_MYBATIS_MAPPER, VM_JAVA_MYBATIS_SERVICE, VM_JAVA_MYBATIS_SERVICE_IMPL, VM_JAVA_MYBATIS_CONTROLLER, VM_XML_MYBATIS_MAPPER);
-                case MYBATIS_FLEX -> List.of(VM_JAVA_MYBATISFLEX_ENTITY, VM_JAVA_MYBATISFLEX_MAPPER, VM_JAVA_MYBATISFLEX_SERVICE, VM_JAVA_MYBATISFLEX_SERVICE_IMPL, VM_JAVA_MYBATISFLEX_CONTROLLER);
+                case MYBATIS_FLEX -> List.of(VM_JAVA_MYBATISFLEX_ENTITY, VM_JAVA_MYBATISFLEX_MAPPER, VM_JAVA_MYBATISFLEX_SERVICE, VM_JAVA_MYBATISFLEX_SERVICE_IMPL, VM_JAVA_MYBATISFLEX_CONTROLLER, VM_XML_MYBATIS_FLEX_MAPPER);
                 case MYBATIS_PLUS -> List.of(VM_JAVA_MYBATIS_PLUS_ENTITY, VM_JAVA_MYBATIS_PLUS_MAPPER, VM_JAVA_MYBATIS_PLUS_SERVICE, VM_JAVA_MYBATIS_PLUS_SERVICE_IMPL, VM_JAVA_MYBATIS_PLUS_CONTROLLER, VM_XML_MYBATIS_PLUS_MAPPER);
                 default -> Collections.emptyList();
             };
@@ -285,7 +286,7 @@ public static String getFileName(String template, GenTable table) {
             case VM_KOTLIN_MYBATIS_SERVICE -> format("{}/service/{}Service.kt", kotlinPath, className);
             case VM_KOTLIN_MYBATIS_SERVICE_IMPL -> format("{}/service/impl/{}ServiceImpl.kt", kotlinPath, className);
             case VM_KOTLIN_MYBATIS_CONTROLLER -> format("{}/controller/{}Controller.kt", kotlinPath, className);
-            case VM_XML_MYBATIS_MAPPER, VM_XML_MYBATIS_PLUS_MAPPER -> format("{}/{}Mapper.xml", mybatisPath, className);
+            case VM_XML_MYBATIS_MAPPER, VM_XML_MYBATIS_FLEX_MAPPER, VM_XML_MYBATIS_PLUS_MAPPER -> format("{}/{}Mapper.xml", mybatisPath, className);
             case VM_VUE_INDEX -> format("{}/{}/index.vue", vuePath, businessName);
             case VM_VUE_INDEX_TS -> format("{}/{}/index.ts", apiPath, businessName);
             case VM_VUE_TYPES -> format("{}/{}/types.ts", apiPath, businessName);

diff --git a/tang-tool/tang-generator/src/main/kotlin/com/tang/generator/constants/VmTemplate.kt b/tang-tool/tang-generator/src/main/kotlin/com/tang/generator/constants/VmTemplate.kt
@@ -49,6 +49,8 @@ object VmTemplate {
 
     const val VM_XML_MYBATIS_MAPPER = "vm/xml/mybatis/mapper.xml.vm"
 
+    const val VM_XML_MYBATIS_FLEX_MAPPER = "vm/xml/mybatisflex/mapper.xml.vm"
+
     const val VM_XML_MYBATIS_PLUS_MAPPER = "vm/xml/mybatisplus/mapper.xml.vm"
 
     const val VM_VUE_INDEX = "vm/vue/index.vue.vm"

diff --git a/tang-tool/tang-generator/src/main/resources/vm/xml/mybatisflex/mapper.xml.vm b/tang-tool/tang-generator/src/main/resources/vm/xml/mybatisflex/mapper.xml.vm
@@ -0,0 +1,16 @@
+<?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.${ClassName}Mapper">
+
+    <resultMap id="${ClassName}Map" type="${packageName}.entity.$ClassName">
+#foreach ($column in $columnList)
+        <result property="$column.javaField" column="$column.columnName" />
+#end
+    </resultMap>
+
+    <sql id="${ClassName}Vo">
+        select#foreach ($column in $columnList) ${tableAlias}.$column.columnName#if ($foreach.count != $columnList.size()),#end#end
+        from $tableName $tableAlias
+    </sql>
+
+</mapper>