树节点操作(返回树形结构)

public class ListTreeUtils {

    public static <E> List<E> getTreeList(List<String> topId, List<E> entityList,String idKey, String parentIdKey, String childListKey) {
        List<E> resultList = new ArrayList<>();
        // 获取顶层元素集合
        String id;
        for (E entity : entityList) {
            id = getValueByPropName(entity, idKey);
            if (topId.contains(id)) {
                resultList.add(entity);
            }

        }
        // 获取每个顶层元素的子数据集合
        for (E entity : resultList) {
            setValueByPropName(entity,childListKey, getSubList(getValueByPropName(entity, idKey).toString(), entityList,idKey,parentIdKey,childListKey));
        }
        return resultList;
    }








    public static <E> List<E> getTreeList(String topId, List<E> entityList,String idKey, String parentIdKey, String childListKey) {
        List<E> resultList = new ArrayList<>();
        // 获取顶层元素集合
        String parentId;
        for (E entity : entityList) {
            parentId = getValueByPropName(entity, parentIdKey);
            if (parentId == null || topId.equals(parentId)) {
                resultList.add(entity);
            }

        }
        // 获取每个顶层元素的子数据集合
        for (E entity : resultList) {
            setValueByPropName(entity,childListKey, getSubList(getValueByPropName(entity, idKey).toString(), entityList,idKey,parentIdKey,childListKey));
        }
        return resultList;
    }

    /**
     * * 获取子数据集合
     *
     * @param id
     * @param entityList
     * @return
     */
    private static <E> List<E> getSubList(String id, List<E> entityList, String idKey, String parentIdKey, String childListKey) {
        List<E> childList = new ArrayList<>();
        String parentId;
        // 子集的直接子对象
        for (E entity : entityList) {
            parentId = getValueByPropName(entity, parentIdKey);
            if (id.equals(parentId)) {
                childList.add(entity);
            }
        }
        // 子集的间接子对象
        for (E entity : childList) {
            setValueByPropName(entity,childListKey, getSubList(getValueByPropName(entity, idKey).toString(), entityList,idKey,parentIdKey,childListKey));
        }
        // 递归退出条件
        if (childList.size() == 0) {
            return null;
        }
        return childList;
    }


    // 通过属性获取传入对象的指定属性的值
    public static <T> T getValueByPropName(Object object, String propName) {
        T value = null;
        try {
            // 通过属性获取对象的属性
            Field field = object.getClass().getDeclaredField(propName);
            // 对象的属性的访问权限设置为可访问
            field.setAccessible(true);
            // 获取属性的对应的值
            value = (T)field.get(object);
        } catch (Exception e) {
            return null;
        }

        return value;
    }

    // 通过属性设置传入对象的指定属性的值
    public static <U> void setValueByPropName(Object object, String propName, U updateValue) {

        try {
            // 通过属性获取对象的属性
            Field field = object.getClass().getDeclaredField(propName);
            // 对象的属性的访问权限设置为可访问
            field.setAccessible(true);
            // 设置属性的对应的值
            field.set(object, updateValue);
        } catch (Exception e) {

        }


    }

}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值