编程语言
首页 > 编程语言> > java返回树形结构的正确姿势

java返回树形结构的正确姿势

作者:互联网

业务场景

通常我们前端需要一个树形的导航菜单或者分类菜单,如后台权限管理中的权限树,亦或者下面例子中商城系统的商品分类多级菜单(一般为三级菜单)

数据库设计

数据库设计,采用parentId来指向自己的父级菜单,如:

CREATE TABLE `pms_category` (
  `cat_id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT '分类id',
  `name` char(50) DEFAULT NULL COMMENT '分类名称',
  `parent_cid` bigint(20) DEFAULT NULL COMMENT '父分类id',
  `cat_level` int(11) DEFAULT NULL COMMENT '层级',
  `show_status` tinyint(4) DEFAULT NULL COMMENT '是否显示[0-不显示,1显示]',
  `sort` int(11) DEFAULT NULL COMMENT '排序',
  `icon` char(255) DEFAULT NULL COMMENT '图标地址',
  `product_unit` char(50) DEFAULT NULL COMMENT '计量单位',
  `product_count` int(11) DEFAULT NULL COMMENT '商品数量',
  PRIMARY KEY (`cat_id`)
) ENGINE=InnoDB AUTO_INCREMENT=1433 DEFAULT CHARSET=utf8mb4 COMMENT='商品三级分类';

java组装树形结构

之前的做法是通过sql自连接来查出树形结构数据,但是效率不高,我们知道单表查询效率是最高的,我们可以一次查出所有数据,通过java8的新特性 stream来处理数据,stream是通过CPU计算实现,效率极高,具体用法可以参考:
Java 8新特性之 Lambd和StreamAPI

下面是处理数据的两个主要方法:

 

@Override
public List<CategoryEntity> listWithTree() {
    // 1. 先查出所有分类数据
    List<CategoryEntity> categories = baseMapper.selectList(null);
    // 2. 找出所有一级分类
    //    在映射到每个一级分类 添加它的子分类类
    return categories.stream()
            .filter(o -> o.getParentCid() == 0)
            // 给每个一级分类加子分类
            .peek(o -> o.setChildrens(getChildCategoryList(o, categories)))
            // 排序
            .sorted(Comparator.comparingInt(CategoryEntity::getSort))
            // 收集
            .collect(Collectors.toList());
}

// 根据当前分类 找出子类, 并通过递归找出子类的子类
private List<CategoryEntity> getChildCategoryList(CategoryEntity currMenu, List<CategoryEntity> categories) {
    return categories.stream().filter(o -> o.getParentCid().equals(currMenu.getCatId()))
            .peek(o -> o.setChildrens(getChildCategoryList(o, categories)))
            .sorted(Comparator.comparingInt(CategoryEntity::getSort))
            .collect(Collectors.toList());
}

实体类变动

@TableField(exist = false)
private List<CategoryEntity> childrens;

public Integer getSort() {
    return sort == null ? 0 : sort;
}

返回数据效果

 

标签:姿势,COMMENT,菜单,java,DEFAULT,分类,树形,NULL,categories
来源: https://www.cnblogs.com/shisanye/p/15432754.html