admin/mybatis-flex
1
0
Fork 0
mirror of https://gitee.com/mybatis-flex/mybatis-flex.git synced 2025-12-07 00:58:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: close https://github.com/mybatis-flex/mybatis-flex/issues/471

Browse Source
This commit is contained in:
Michael Yang 2025-01-10 09:17:36 +08:00
parent c88155de85
commit 39b0efbd6a
1 changed files with 68 additions and 79 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

147
mybatis-flex-core/src/main/java/com/mybatisflex/core/relation/RelationManager.java
View File

@ -309,98 +309,87 @@ public class RelationManager {
return;
}
String currentDsKey = DataSourceKey.get();
try {
relations.forEach(relation -> {
relations.forEach(relation -> {
//ignore
if (ignoreRelations != null && (ignoreRelations.contains(relation.getSimpleName())
|| ignoreRelations.contains(relation.getName()))) {
return;
}
//ignore
if (ignoreRelations != null && (ignoreRelations.contains(relation.getSimpleName())
|| ignoreRelations.contains(relation.getName()))) {
return;
}
//only query
if (queryRelations != null && !queryRelations.isEmpty()
&& !queryRelations.contains(relation.getSimpleName())
&& !queryRelations.contains(relation.getName())) {
return;
}
//only query
if (queryRelations != null && !queryRelations.isEmpty()
&& !queryRelations.contains(relation.getSimpleName())
&& !queryRelations.contains(relation.getName())) {
return;
}
//注解配置的数据源
String configDsKey = relation.getDataSource();
if (StringUtil.noText(configDsKey) && currentDsKey != null) {
configDsKey = currentDsKey;
}
//注解配置的数据源
String relationDsKey = relation.getDataSource();
if (StringUtil.hasText(relationDsKey)) {
DataSourceKey.use(relationDsKey);
}
try {
if (StringUtil.hasText(configDsKey)) {
DataSourceKey.use(configDsKey);
try {
Set<Object> targetValues;
List<Row> mappingRows = null;
//通过中间表关联查询
if (relation.isRelationByMiddleTable()) {
Set selfFieldValues = relation.getSelfFieldValues(entities);
// 当数据对应的字段没有值的情况下，直接返回
if (selfFieldValues.isEmpty()) {
return;
}
QueryWrapper queryWrapper = QueryWrapper.create().select()
.from(relation.getJoinTable());
if (selfFieldValues.size() > 1) {
queryWrapper.where(column(relation.getJoinSelfColumn()).in(selfFieldValues));
} else {
queryWrapper.where(column(relation.getJoinSelfColumn()).eq(selfFieldValues.iterator().next()));
}
Set<Object> targetValues;
List<Row> mappingRows = null;
//通过中间表关联查询
if (relation.isRelationByMiddleTable()) {
Set selfFieldValues = relation.getSelfFieldValues(entities);
// 当数据对应的字段没有值的情况下，直接返回
if (selfFieldValues.isEmpty()) {
return;
}
QueryWrapper queryWrapper = QueryWrapper.create().select()
.from(relation.getJoinTable());
if (selfFieldValues.size() > 1) {
queryWrapper.where(column(relation.getJoinSelfColumn()).in(selfFieldValues));
} else {
queryWrapper.where(column(relation.getJoinSelfColumn()).eq(selfFieldValues.iterator().next()));
}
mappingRows = mapper.selectRowsByQuery(queryWrapper);
if (CollectionUtil.isEmpty(mappingRows)) {
return;
}
targetValues = new HashSet<>();
for (Row mappingData : mappingRows) {
Object targetValue = mappingData.getIgnoreCase(relation.getJoinTargetColumn());
if (targetValue != null) {
targetValues.add(targetValue);
}
}
}
//通过外键字段关联查询
else {
targetValues = relation.getSelfFieldValues(entities);
}
if (CollectionUtil.isEmpty(targetValues)) {
mappingRows = mapper.selectRowsByQuery(queryWrapper);
if (CollectionUtil.isEmpty(mappingRows)) {
return;
}
//仅绑定字段:As目标实体类 不进行字段绑定:As映射类型
QueryWrapper queryWrapper = relation.buildQueryWrapper(targetValues);
List<?> targetObjectList = mapper.selectListByQueryAs(queryWrapper, relation.isOnlyQueryValueField() ? relation.getTargetEntityClass() : relation.getMappingType());
if (CollectionUtil.isNotEmpty(targetObjectList)) {
targetValues = new HashSet<>();
//递归查询
doQueryRelations(mapper, targetObjectList, currentDepth + 1, maxDepth, ignoreRelations, queryRelations);
//进行内存 join
relation.join(entities, targetObjectList, mappingRows);
}
} finally {
if (StringUtil.hasText(configDsKey)) {
DataSourceKey.clear();
for (Row mappingData : mappingRows) {
Object targetValue = mappingData.getIgnoreCase(relation.getJoinTargetColumn());
if (targetValue != null) {
targetValues.add(targetValue);
}
}
}
});
} finally {
if (currentDsKey != null) {
DataSourceKey.use(currentDsKey);
//通过外键字段关联查询
else {
targetValues = relation.getSelfFieldValues(entities);
}
if (CollectionUtil.isEmpty(targetValues)) {
return;
}
//仅绑定字段:As目标实体类 不进行字段绑定:As映射类型
QueryWrapper queryWrapper = relation.buildQueryWrapper(targetValues);
List<?> targetObjectList = mapper.selectListByQueryAs(queryWrapper, relation.isOnlyQueryValueField() ? relation.getTargetEntityClass() : relation.getMappingType());
if (CollectionUtil.isNotEmpty(targetObjectList)) {
//递归查询
doQueryRelations(mapper, targetObjectList, currentDepth + 1, maxDepth, ignoreRelations, queryRelations);
//进行内存 join
relation.join(entities, targetObjectList, mappingRows);
}
} finally {
if (StringUtil.hasText(relationDsKey)) {
DataSourceKey.clear();
}
}
}
});
}
}