third_codes/mybatis-plus-join
1
0
Fork 0
mirror of https://gitee.com/best_handsome/mybatis-plus-join synced 2025-07-11 00:02:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化resultMap

Browse Source
This commit is contained in:
admin 2021-06-11 10:44:52 +08:00
parent 47ef74451e
commit 4bbe963579
1 changed files with 31 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/main/java/com/baomidou/mybatisplus/core/metadata/MPJTableInfoHelper.java
View File

@ -5,6 +5,9 @@ import com.baomidou.mybatisplus.core.config.GlobalConfig;
import com.baomidou.mybatisplus.core.toolkit.*; import com.baomidou.mybatisplus.core.toolkit.*;
import org.apache.ibatis.logging.Log; import org.apache.ibatis.logging.Log;
import org.apache.ibatis.logging.LogFactory; import org.apache.ibatis.logging.LogFactory;
import org.apache.ibatis.mapping.ResultFlag;
import org.apache.ibatis.mapping.ResultMap;
import org.apache.ibatis.mapping.ResultMapping;
import org.apache.ibatis.reflection.Reflector; import org.apache.ibatis.reflection.Reflector;
import org.apache.ibatis.reflection.ReflectorFactory; import org.apache.ibatis.reflection.ReflectorFactory;
import org.apache.ibatis.session.Configuration; import org.apache.ibatis.session.Configuration;
@ -86,7 +89,7 @@ public class MPJTableInfoHelper {
initTableFields(clazz, globalConfig, tableInfo, excludePropertyList); initTableFields(clazz, globalConfig, tableInfo, excludePropertyList);
/* 自动构建 resultMap */ /* 自动构建 resultMap */
tableInfo.initResultMapIfNeed(); initResultMapIfNeed(tableInfo);
/* 添加缓存 */ /* 添加缓存 */
TABLE_INFO_CACHE.put(clazz, tableInfo); TABLE_INFO_CACHE.put(clazz, tableInfo);
@ -96,6 +99,33 @@ public class MPJTableInfoHelper {
return tableInfo; return tableInfo;
} }
/**
* 自动构建 resultMap 并注入(如果条件符合的话)
*/
private static void initResultMapIfNeed(TableInfo tableInfo) {
if (tableInfo.isAutoInitResultMap() && null == tableInfo.getResultMap()) {
String id = tableInfo.getCurrentNamespace() + ".mybatis-plus-join_" + tableInfo.getEntityType().getSimpleName();
tableInfo.setResultMap(id);
if (tableInfo.getConfiguration().getResultMapNames().contains(id)) {
tableInfo.getConfiguration().getResultMap(id);
}
List<ResultMapping> resultMappings = new ArrayList<>();
if (tableInfo.havePK()) {
ResultMapping idMapping = new ResultMapping.Builder(tableInfo.getConfiguration(), tableInfo.getKeyProperty(),
tableInfo.getKeyColumn(), tableInfo.getKeyType())
.flags(Collections.singletonList(ResultFlag.ID)).build();
resultMappings.add(idMapping);
}
if (CollectionUtils.isNotEmpty(tableInfo.getFieldList())) {
tableInfo.getFieldList().forEach(i -> resultMappings.add(i.getResultMapping(tableInfo.getConfiguration())));
}
ResultMap resultMap = new ResultMap.Builder(tableInfo.getConfiguration(), id, tableInfo.getEntityType(),
resultMappings).build();
tableInfo.getConfiguration().addResultMap(resultMap);
}
}
/** /**
* <p> * <p>
* 初始化 表数据库类型,表名,resultMap * 初始化 表数据库类型,表名,resultMap