递归构建下拉树

public List<TreeSelect> buildMenuTreeSelect(List<PrivilegeDTO> menus) {
    List<PrivilegeDTO> menuTrees = new ArrayList<>();
    List<Integer> tempList = menus.stream().map(PrivilegeDTO::getId).collect(Collectors.toList());
    for (PrivilegeDTO menu : menus) {
        // 如果是顶级节点, 遍历该父节点的所有子节点
        if (!tempList.contains(menu.getParentId())) {
            recursionFn(menus, menu);
            menuTrees.add(menu);
        }
    }
    if (menuTrees.isEmpty()) {
        menuTrees = menus;
    }
    return menuTrees.stream().map(TreeSelect::new).collect(Collectors.toList());
}

/**
 * 递归列表,风暴大脑
 *
 * @param list       分类表
 * @param child      子节点
 */
private void recursionFn(List<PrivilegeDTO> list, PrivilegeDTO child) {
    // 得到子节点列表
    List<PrivilegeDTO> childList = getChildList(list, child);
    child.setChildren(childList);
    for (PrivilegeDTO p : childList) {
        //判断是否有子节点
        if (!getChildList(list, p).isEmpty()) {
            recursionFn(list, p);
        }
    }
}

/**
 * 得到子节点列表
 */
private List<PrivilegeDTO> getChildList(List<PrivilegeDTO> list, PrivilegeDTO child) {
    List<PrivilegeDTO> tlist = new ArrayList<>();
    for (PrivilegeDTO privilegeDTO : list) {
        if (Objects.equals(privilegeDTO.getParentId(), child.getId())) {
            tlist.add(privilegeDTO);
        }
    }
    return tlist;
}

附实体treeSelect结构

public class TreeSelect implements Serializable {
    private static final long serialVersionUID = 1L;

    @ApiModelProperty("id")
    private Integer id;

    @ApiModelProperty("权限名称")
    private String name;

    @ApiModelProperty("子权限列表")

    //此注解可以使空值时此属性不返回于前端
    @JsonInclude(JsonInclude.Include.NON_EMPTY)
    private List<TreeSelect> children;

    public TreeSelect(PrivilegeDTO menu)
    {
        this.id = menu.getId();
        this.name = menu.getName();
        this.children = menu.getChildren().stream().map(TreeSelect::new).collect(Collectors.toList());
    }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值