Java

[Java]用Lambda构建父子级菜单

2021-12-27  本文已影响0人  AbstractCulture
    public List<MenuDTO> getMenus() {
        List<MenuEntity> menuEntityList = list();
        if (CollectionUtils.isEmpty(menuEntityList)) {
            return Collections.emptyList();
        }
        List<MenuDTO> menuDTOList = BeanUtils.copyProperties(menuEntityList, MenuDTO.class);
        List<MenuDTO> resultDTOs = menuDTOList.stream()
                .filter(m -> m.getParentId() == 0)
                .peek(m -> m.setChildrenMenuDTOs(getChildMenus(m, menuDTOList)))
                .sorted(Comparator.comparing(MenuDTO::getSort))
                .collect(Collectors.toList());
        return resultDTOs;
    }

    private List<MenuDTO> getChildMenus(MenuDTO menuDTO, List<MenuDTO> menuDTOList) {
        return menuDTOList.stream()
                .filter(m -> Objects.equals(m.getParentId(), menuDTO.getId()))
                .peek(m -> m.setChildrenMenuDTOs(getChildMenus(m, menuDTOList)))
                .sorted(Comparator.comparing(MenuDTO::getSort))
                .collect(Collectors.toList());
    }
@Data
@NoArgsConstructor
@AllArgsConstructor
@Builder
public class MenuDTO {
    @ApiModelProperty("菜单id")
    private Long id;
    @ApiModelProperty("菜单名称")
    private String name;
    @ApiModelProperty("菜单路径")
    private String url;
    @ApiModelProperty("菜单图标")
    private String icon;
    @ApiModelProperty("前端菜单索引")
    private String frontIndex;
    @ApiModelProperty("排序")
    private Integer sort;
    @ApiModelProperty("父级ID")
    private Long parentId;
    @ApiModelProperty("子菜单")
    private List<MenuDTO> childrenMenuDTOs;
}

DDDD

上一篇 下一篇

猜你喜欢

热点阅读