From adbd10d8a2f8447688d2a907d7c8e7ef88f17091 Mon Sep 17 00:00:00 2001
From: zhouwentao <1577701412@qq.com>
Date: Mon, 14 Aug 2023 15:29:41 +0800
Subject: [PATCH] updates

---
 .../codegenerate/entity/TableEntityVo.java    | 20 +++++++++++++++----
 src/main/resources/yangliu/XXXMapper.xmli     |  9 +++++++++
 2 files changed, 25 insertions(+), 4 deletions(-)

diff --git a/src/main/java/org/yangliu/codegenerate/entity/TableEntityVo.java b/src/main/java/org/yangliu/codegenerate/entity/TableEntityVo.java
index 7cb0242..ae6b16b 100644
--- a/src/main/java/org/yangliu/codegenerate/entity/TableEntityVo.java
+++ b/src/main/java/org/yangliu/codegenerate/entity/TableEntityVo.java
@@ -6,10 +6,13 @@ package org.yangliu.codegenerate.entity;
  * @Date 2023/8/13 20:44
  */
 public class TableEntityVo {
-    //字段
+    //数据库中字段
+    private String fieldColumn;
+    //实体类字段
     private String fieldName;
+    //java中字段类型
     private String fieldType;
-    //字段类型
+    //数据库中字段类型
     private String fieldDbType;
     //字段描述
     private String filedComment;
@@ -18,11 +21,20 @@ public class TableEntityVo {
     public TableEntityVo() {
     }
 
-    public TableEntityVo(String fieldName,String fieldType,String fieldDbType,String filedComment) {
-        this.filedComment = filedComment;
+    public TableEntityVo(String fieldColumn,String fieldName,String fieldType,String fieldDbType,String filedComment) {
+        this.fieldColumn = fieldColumn;
         this.fieldName = fieldName;
         this.fieldType = fieldType;
         this.fieldDbType = fieldDbType;
+        this.filedComment = filedComment;
+    }
+
+    public String getFieldColumn() {
+        return fieldColumn;
+    }
+
+    public void setFieldColumn(String fieldColumn) {
+        this.fieldColumn = fieldColumn;
     }
 
     public String getFiledComment() {
diff --git a/src/main/resources/yangliu/XXXMapper.xmli b/src/main/resources/yangliu/XXXMapper.xmli
index 9594782..6daabec 100644
--- a/src/main/resources/yangliu/XXXMapper.xmli
+++ b/src/main/resources/yangliu/XXXMapper.xmli
@@ -1,5 +1,14 @@
 <?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="${bussiPackage}.mapper.${entityName}Mapper">
+<resultMap type="${bussiPackage}.${entityPackage}.${entityName}" id="${entityName?uncap_first}Result">
+<#list originalColumns as po>
+	<#if po.fieldName == primaryKeyField>
+    	<id property="${po.fieldName}" column="${po.fieldColumn}"/>
+	<#else>
+	    <result property="${po.fieldName}" column="${po.fieldColumn}"/>
+	</#if>
+</#list>
+</resultMap>
 
 </mapper>
\ No newline at end of file