From 56ab702568b2ce72ecfaae166a1e790b1ed08cc8 Mon Sep 17 00:00:00 2001
From: ray <1490493387@qq.com>
Date: Wed, 31 Jul 2024 07:46:57 +0800
Subject: [PATCH] =?UTF-8?q?refactor:=20=E5=AE=8C=E5=96=84=20mapper.xml=20?=
 =?UTF-8?q?=E6=97=A5=E6=9C=9F=E6=9D=A1=E4=BB=B6=E6=9F=A5=E8=AF=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../templates/generator/mapper.xml.vm         | 34 +++++++++++--------
 1 file changed, 20 insertions(+), 14 deletions(-)

diff --git a/src/main/resources/templates/generator/mapper.xml.vm b/src/main/resources/templates/generator/mapper.xml.vm
index 55a33559..ed2fdf96 100644
--- a/src/main/resources/templates/generator/mapper.xml.vm
+++ b/src/main/resources/templates/generator/mapper.xml.vm
@@ -9,7 +9,7 @@
             #set ($fields = [])
             #foreach($fieldConfig in ${fieldConfigs})
                 #if($fieldConfig.isShowInList)
-                    #set ($discard = $fields.add(${fieldConfig.columnName}))
+                #set ($discard = $fields.add(${fieldConfig.columnName}))
                 #end
             #end
             #foreach($field in $fields)
@@ -22,32 +22,38 @@
             #if($fieldConfigs)
                 #foreach($fieldConfig in ${fieldConfigs})
                     #if($fieldConfig.isShowInQuery)
-                        <if test="queryParams.${fieldConfig.fieldName} != null">
+            <if test="queryParams.${fieldConfig.fieldName} != null">
                             #set ($queryType = ${fieldConfig.queryType})
                             #if($queryType == "EQ")
-                                AND ${fieldConfig.columnName} = #{queryParams.${fieldConfig.fieldName}}
+                AND ${fieldConfig.columnName} = #{queryParams.${fieldConfig.fieldName}}
                             #elseif($queryType == "LIKE")
-                                AND ${fieldConfig.columnName} LIKE CONCAT('%', #{queryParams.${fieldConfig.fieldName}}, '%')
+                AND ${fieldConfig.columnName} LIKE CONCAT('%', #{queryParams.${fieldConfig.fieldName}}, '%')
                             #elseif($queryType == "IN")
-                                AND ${fieldConfig.columnName} IN
+                <if test="queryParams.${fieldConfig.fieldName}.size() > 0">
+                    AND ${fieldConfig.columnName} IN  <foreach collection="queryParams.${fieldConfig.fieldName}" item="item" open="(" close=")" separator=",">#{item}</foreach>
+                </if>
                             #elseif($queryType == "BETWEEN")
-                                AND ${fieldConfig.columnName} BETWEEN #{queryParams.${fieldConfig.fieldName}Start} AND #{queryParams.${fieldConfig.fieldName}End}
+                                #if(${fieldConfig.fieldType} == "LocalDateTime")
+                AND ${fieldConfig.columnName} BETWEEN STR_TO_DATE(#{queryParams.${fieldConfig.fieldName}Start}, '%Y-%m-%d %H:%i:%s') AND STR_TO_DATE(#{queryParams.${fieldConfig.fieldName}End}, '%Y-%m-%d %H:%i:%s')
+                                #else
+                AND ${fieldConfig.columnName} BETWEEN #{queryParams.${fieldConfig.fieldName}Start} AND #{queryParams.${fieldConfig.fieldName}End}
+                                #end
                             #elseif($queryType == "GT")
-                                AND ${fieldConfig.columnName} > #{queryParams.${fieldConfig.fieldName}}
+                AND ${fieldConfig.columnName} > #{queryParams.${fieldConfig.fieldName}}
                             #elseif($queryType == "GE")
-                                AND ${fieldConfig.columnName} >= #{queryParams.${fieldConfig.fieldName}}
+                AND ${fieldConfig.columnName} >= #{queryParams.${fieldConfig.fieldName}}
                             #elseif($queryType == "LT")
-                                AND ${fieldConfig.columnName} &lt; #{queryParams.${fieldConfig.fieldName}}
+                AND ${fieldConfig.columnName} &lt; #{queryParams.${fieldConfig.fieldName}}
                             #elseif($queryType == "LE")
-                                AND ${fieldConfig.columnName} &lt;= #{queryParams.${fieldConfig.fieldName}}
+                AND ${fieldConfig.columnName} &lt;= #{queryParams.${fieldConfig.fieldName}}
                             #elseif($queryType == "NE")
-                                AND ${fieldConfig.columnName} != #{queryParams.${fieldConfig.fieldName}}
+                AND ${fieldConfig.columnName} != #{queryParams.${fieldConfig.fieldName}}
                             #elseif($queryType == "LIKE_LEFT")
-                                AND ${fieldConfig.columnName} LIKE CONCAT('%', #{queryParams.${fieldConfig.fieldName}})
+                AND ${fieldConfig.columnName} LIKE CONCAT('%', #{queryParams.${fieldConfig.fieldName}})
                             #elseif($queryType == "LIKE_RIGHT")
-                                AND ${fieldConfig.columnName} LIKE CONCAT(#{queryParams.${fieldConfig.fieldName}}, '%')
+                AND ${fieldConfig.columnName} LIKE CONCAT(#{queryParams.${fieldConfig.fieldName}}, '%')
                             #end
-                        </if>
+            </if>
                     #end
                 #end
             #end