java构造树形结构工具类

节点数据构造为树形结构工具类（从数据库查询出来的List集合构造为树形结构，多条树）

注意：对象中必须要有 id,pid,children属性，其中children为集合

自测没问题，如有更好方法请联系一下我…

package com.chaint.monomer_auth.utils;


import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.List;

/**
 * @Auther: xieTao
 * @Date: 2020-06-18
 * @Description: TODO
 * @version: 1.0
 */
public class TreeUtil {

    /**
     * 先根据List数据集合获取每一组树的根集合，然后将根集合和数据List集合进行递归得到树
     * @param nodes 全部数据节点
     * @param parentColumnName 对象根ID的属性名
     * @param pidColumnName 对象pid的属性名
     * @param rootValue 根的属性值为多少
     * @param childrenName 对象装子节点的属性名
     * @param <T>
     * @return
     * @throws Exception
     */
    public static<T> List<T> getTree(List<T> nodes,String parentColumnName, String pidColumnName, String rootValue,String childrenName) throws Exception {
        List<T> root = new ArrayList<T>();
        for(T t : nodes){
            Field field = t.getClass().getDeclaredField(pidColumnName);
            field.setAccessible(true);
            Object pid = field.get(t);

            if (rootValue.equals(pid)){
                root.add(t);
            }

        }
        return buildTree(nodes, root,parentColumnName,pidColumnName,childrenName);
    }

    /**
     * 循环根的集合， 将根的id和数据集合的父id进行遍历对比整合为子集合，然后继续将子集合作为根进行递归，最后将子集合存入根对象中
     * @param nodes 全部数据节点
     * @param root 根节点的集合
     * @param parentColumnName 对象根ID的属性名
     * @param pidColumnName 对象pid的属性名
     * @param childrenName 对象装子节点的属性名
     * @param <T>
     * @return
     * @throws Exception
     */
    private static<T> List<T> buildTree(List<T> nodes, List<T> root,String parentColumnName,String pidColumnName,String childrenName) throws Exception {
        for(int i = 0; i < root.size(); i++){

            T rootT = root.get(i);
            Class<?> tClass = rootT.getClass();
            Field rootField = tClass.getDeclaredField(parentColumnName);
            Field childrenNameField = tClass.getDeclaredField(childrenName);
            rootField.setAccessible(true);
            childrenNameField.setAccessible(true);
            Object id = rootField.get(rootT);

            List<T> children = new ArrayList<>();
            for (T t : nodes){
                Field field = t.getClass().getDeclaredField(pidColumnName);
                field.setAccessible(true);
                Object pid = field.get(t);

                if (id.equals(pid)){
                    children.add(t);
                }
            }
            buildTree(nodes, children,parentColumnName,pidColumnName,childrenName);
            childrenNameField.set(rootT,children);
        }
        return root;

    }

}