Java 实现多级菜单分类

Java 实现多级菜单分类

@Override
public List<CategoryEntity> listWithTree() {
    //1.获得所有的分类
    List<CategoryEntity> list = baseMapper.selectList(null);

    //2.列表进行组装成树型结构
    return list.stream()
            .filter(item -> item.getParentCid() == 0)
            .peek(item -> item.setChildren(getChildes(item, list)))
            .sorted(Comparator.comparingInt(item -> (item.getSort() == null ? 0 : item.getSort())))
            .collect(Collectors.toList());
}

/**
 * 使用递归获取商品分类的树形结构
 * @param current 当前父分类
 * @param all 所有分类
 * @return 递归子分类
 */
private List<CategoryEntity> getChildes(CategoryEntity current, List<CategoryEntity> all) {
    return all.stream()
    		.filter(item -> item.getParentCid().equals(current.getCatId()))
            .peek(item -> item.setChildren(getChildes(item, all)))
            .sorted(Comparator.comparingInt(item -> (item.getSort() == null ? 0 : item.getSort())))
            .collect(Collectors.toList());
}