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

feat: add RelationManager.addQueryRelations() methods close #I7VLR3

Browse Source
This commit is contained in:
开源海哥 2023-08-25 18:06:26 +08:00
parent 65fad93e50
commit fc4de0ad1a
1 changed files with 61 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -67,11 +67,18 @@ public class RelationManager {
private static ThreadLocal<Set<String>> ignoreRelations = new ThreadLocal<>();
/**
* 查询时，仅查询这个配置的 Relations
*/
private static ThreadLocal<Set<String>> onlyQueryRelations = new ThreadLocal<>();
/**
* 每次查询是否自动清除 depth extraConditionParams ignoreRelations 的配置
*/
private static ThreadLocal<Boolean> autoClearConfig = ThreadLocal.withInitial(() -> true);
public static int getDefaultQueryDepth() {
return defaultQueryDepth;
}
@ -115,6 +122,11 @@ public class RelationManager {
}
//////ignore relations //////
public static Set<String> getIgnoreRelations() {
return ignoreRelations.get();
}
public static void setIgnoreRelations(Set<String> ignoreRelations) {
RelationManager.ignoreRelations.set(ignoreRelations);
}
@ -141,15 +153,49 @@ public class RelationManager {
relations.addAll(Arrays.asList(ignoreRelations));
}
public static Set<String> getIgnoreRelations() {
return ignoreRelations.get();
}
public static void clearIgnoreRelations() {
ignoreRelations.remove();
}
//////query relations //////
public static Set<String> getQueryRelations() {
return onlyQueryRelations.get();
}
public static void setQueryRelations(Set<String> queryRelations) {
RelationManager.onlyQueryRelations.set(queryRelations);
}
public static <T> void addQueryRelations(LambdaGetter<T>... queryRelations) {
Set<String> relations = RelationManager.onlyQueryRelations.get();
if (relations == null) {
relations = new HashSet<>();
setQueryRelations(relations);
}
for (LambdaGetter<T> lambdaGetter : queryRelations) {
String fieldName = LambdaUtil.getFieldName(lambdaGetter);
relations.add(fieldName);
}
}
public static void addQueryRelations(String... queryRelations) {
Set<String> relations = RelationManager.onlyQueryRelations.get();
if (relations == null) {
relations = new HashSet<>();
setQueryRelations(relations);
}
relations.addAll(Arrays.asList(queryRelations));
}
public static void clearQueryRelations() {
onlyQueryRelations.remove();
}
public static void setAutoClearConfig(boolean enable) {
autoClearConfig.set(enable);
}
@ -214,7 +260,7 @@ public class RelationManager {
public static <Entity> void queryRelations(BaseMapper<?> mapper, List<Entity> entities) {
doQueryRelations(mapper, entities, 0, depthThreadLocal.get(), ignoreRelations.get());
doQueryRelations(mapper, entities, 0, depthThreadLocal.get(), ignoreRelations.get(), onlyQueryRelations.get());
clearConfigIfNecessary();
}
@ -226,13 +272,14 @@ public class RelationManager {
if (autoClearEnable != null && autoClearEnable) {
depthThreadLocal.remove();
extraConditionParams.remove();
onlyQueryRelations.remove();
ignoreRelations.remove();
}
}
@SuppressWarnings({"rawtypes", "unchecked"})
static <Entity> void doQueryRelations(BaseMapper<?> mapper, List<Entity> entities, int currentDepth, int maxDepth, Set<String> ignoreRelations) {
static <Entity> void doQueryRelations(BaseMapper<?> mapper, List<Entity> entities, int currentDepth, int maxDepth, Set<String> ignoreRelations, Set<String> queryRelations) {
if (CollectionUtil.isEmpty(entities)) {
return;
}
@ -257,6 +304,14 @@ public class RelationManager {
return;
}
//only query
if (queryRelations != null && !queryRelations.isEmpty()
&& !queryRelations.contains(relation.getSimpleName())
&& !queryRelations.contains(relation.getName())) {
return;
}
Set<Object> targetValues;
List<Row> mappingRows = null;
@ -309,7 +364,7 @@ public class RelationManager {
if (CollectionUtil.isNotEmpty(targetObjectList)) {
//递归查询
doQueryRelations(mapper, targetObjectList, currentDepth + 1, maxDepth, ignoreRelations);
doQueryRelations(mapper, targetObjectList, currentDepth + 1, maxDepth, ignoreRelations, queryRelations);
//进行内存 join
relation.join(entities, targetObjectList, mappingRows);