java 多级tree

经常会面临这样的需求,返回一个多级的tree。在此记录一下希望能帮助到大家。

首先先创建一个TreeModel 。

@Data
public class TreeModel {
    private  String id;
    private String name;
    private String pid;
    private String type;

    public TreeModel(String id, String name, String pid, String type) {
        this.id = id;
        this.name = name;
        this.pid = pid;
        this.type = type;
    }

    private List<TreeModel> treeModels;

    public static List<TreeModel> buildTree(List<TreeModel> treeModels,String id){
        for (TreeModel treeModel :
            treeModels) {
            treeModel.setTreeModels(treeModels.stream().filter(p->p.getPid().equals(treeModel.getId())).collect(Collectors.toList()));
        }
        return treeModels.stream().filter(p->p.id.equals(id)).collect(Collectors.toList());
    }

    @Override
    public String toString() {
        return "TreeModel{" +
            "id='" + id + '\'' +
            ", name='" + name + '\'' +
            ", pid='" + pid + '\'' +
            ", type='" + type + '\'' +
            '}';
    }
}

2、引用

// 创建一个 TreeModel 集合
List<TreeModel> list = new ArrayList<>();
// 获取到 部门 List
List<SysDept> depts= ....

// 获取到 下一级 List
List<Line> lines = ...

// 多级 ....


// 进行调用
for (SysDept dept : depts) {
            list.add(new TreeModel("dept_" + dept.getDeptId(), dept.getDeptName(), "dept_" + dept.getParentId(), "Dept"));
        }
for (TbPatrolLine line:lines) {
            list.add(new TreeModel("line_" + line.getId(), line.getName(), "dept_" + line.getDeptId(), "Line"));
        }

// 排序  其中 dept_100  100 是我部门的最高级,这里建议你 查出 最高级别
 List<TreeModel> result = TreeModel.buildTree(list, "dept_100");

// 返回
return  result;

一个简单的多级tree 就封装好了。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值