|
|
|
@ -1,9 +1,11 @@
|
|
|
|
|
package com.ruoyi.common.core.utils;
|
|
|
|
|
|
|
|
|
|
import cn.hutool.core.collection.CollUtil;
|
|
|
|
|
import cn.hutool.core.lang.tree.Tree;
|
|
|
|
|
import cn.hutool.core.lang.tree.TreeNodeConfig;
|
|
|
|
|
import cn.hutool.core.lang.tree.TreeUtil;
|
|
|
|
|
import cn.hutool.core.lang.tree.parser.NodeParser;
|
|
|
|
|
import com.ruoyi.common.core.utils.reflect.ReflectUtils;
|
|
|
|
|
import lombok.AccessLevel;
|
|
|
|
|
import lombok.NoArgsConstructor;
|
|
|
|
|
|
|
|
|
@ -22,8 +24,12 @@ public class TreeBuildUtils extends TreeUtil {
|
|
|
|
|
*/
|
|
|
|
|
public static final TreeNodeConfig DEFAULT_CONFIG = TreeNodeConfig.DEFAULT_CONFIG.setNameKey("label");
|
|
|
|
|
|
|
|
|
|
public static <T> List<Tree<Long>> build(List<T> list, Long parentId, NodeParser<T, Long> nodeParser) {
|
|
|
|
|
return TreeUtil.build(list, parentId, DEFAULT_CONFIG, nodeParser);
|
|
|
|
|
public static <T, K> List<Tree<K>> build(List<T> list, NodeParser<T, K> nodeParser) {
|
|
|
|
|
if (CollUtil.isEmpty(list)) {
|
|
|
|
|
return null;
|
|
|
|
|
}
|
|
|
|
|
K k = ReflectUtils.invokeGetter(list.get(0), "parentId");
|
|
|
|
|
return TreeUtil.build(list, k, DEFAULT_CONFIG, nodeParser);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|