:zap: 优化代码

blade-core-mybatis/src/main/java/org/springblade/core/mp/support/Condition.java

@@ -18,8 +18,8 @@ package org.springblade.core.mp.support;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
-import org.springblade.core.tool.utils.Func;
 import org.springblade.core.tool.utils.BeanUtil;
+import org.springblade.core.tool.utils.Func;
 
 import java.util.Map;
 
@@ -63,6 +63,8 @@ public class Condition {
 	 * @return
 	 */
 	public static <T> QueryWrapper<T> getQueryWrapper(Map<String, Object> query, Class<T> clazz) {
+		query.remove("current");
+		query.remove("size");
 		QueryWrapper<T> qw = new QueryWrapper<>();
 		qw.setEntity(BeanUtil.newInstance(clazz));
 		if (Func.isNotEmpty(query)) {

blade-core-tool/src/main/java/org/springblade/core/tool/node/ForestNodeManager.java

@@ -30,6 +30,11 @@ public class ForestNodeManager<T extends INode> {
 	 */
 	private List<T> list;
 
+	/**
+	 * 森林的父节点ID
+	 */
+	private List<Integer> parentIds = new ArrayList<>();
+
 	public ForestNodeManager(List<T> items) {
 		list = items;
 	}
@@ -49,6 +54,15 @@ public class ForestNodeManager<T extends INode> {
 		return null;
 	}
 
+	/**
+	 * 增加父节点ID
+	 *
+	 * @param parentId
+	 */
+	public void addParentId(Integer parentId) {
+		parentIds.add(parentId);
+	}
+
 	/**
 	 * 获取树的根节点(一个森林对应多颗树)
 	 *
@@ -57,7 +71,7 @@ public class ForestNodeManager<T extends INode> {
 	public List<T> getRoot() {
 		List<T> roots = new ArrayList<>();
 		for (T forestNode : list) {
-			if (forestNode.getParentId() == 0) {
+			if (forestNode.getParentId() == 0 || parentIds.contains(forestNode.getId())) {
 				roots.add(forestNode);
 			}
 		}

blade-core-tool/src/main/java/org/springblade/core/tool/node/ForestNodeMerger.java

@@ -15,6 +15,7 @@
  */
 package org.springblade.core.tool.node;
 
+import java.util.ArrayList;
 import java.util.List;
 
 /**
@@ -32,13 +33,18 @@ public class ForestNodeMerger {
 	 * @return 多棵树的根节点集合
 	 */
 	public static <T extends INode> List<T> merge(List<T> items) {
+		List<Integer> parentIds = new ArrayList<>();
 		ForestNodeManager<T> forestNodeManager = new ForestNodeManager<>(items);
-		for (T forestNode : items) {
+		items.forEach(forestNode -> {
 			if (forestNode.getParentId() != 0) {
 				INode node = forestNodeManager.getTreeNodeAT(forestNode.getParentId());
-				node.getChildren().add(forestNode);
+				if (node != null) {
+					node.getChildren().add(forestNode);
+				} else {
+					forestNodeManager.addParentId(forestNode.getId());
+				}
 			}
-		}
+		});
 		return forestNodeManager.getRoot();
 	}