树形展示三级分类并分页：递归树形结构

Posted 2022-01-01 prode123

今天在写一个项目的时候遇到了一个问题：如何写三级分类的接口？

经过多方查询资料 我解决了该问题

前端页面如下：

 数据库：

 在实体类中添加children

package com.example.humanresources.entity;

import java.util.Date;
import java.io.Serializable;
import java.util.List;

import lombok.Data;
@Data
public class BizProductCategory implements Serializable 
    private static final long serialVersionUID = 892831452437776529L;
    /**
    * 类别id
    */
    
    private Long id;
    /**
    * 类别名称
    */
    
    private String name;
    /**
    * 备注
    */
    
    private String remark;
    /**
    * 排序
    */
    
    private Integer sort;
    
    private Date createTime;
    
    private Date modifiedTime;
    /**
    * 父级分类id
    */
    
    private Long pid;

//    @TableField
    private List<BizProductCategory> children;

在mybatis-plus中需要用到@TableField注解

由于我直接用的mybatis 所以直接将children添加为属性即可

因为只对一级分类进行分页，所以我又写了一个实体类：

package com.example.humanresources.entity;

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.List;

@Data
@NoArgsConstructor
@AllArgsConstructor
public class BizProductCategoryPlus 
    private BizProductCategory bizProductCategory;
    private List<BizProductCategory> rows;
    private int total;

service层代码：

  @Override
    public BizProductCategoryPlus listWithTree(int  pageNum, int pageSize) 
        // 1.查出所有分类
        List<BizProductCategory> bizProductCategories = bizProductCategoryDao.selectAll();
        // 2.组装成父子树形结构
        // 2.1找出所有的一级分类
        List<BizProductCategory> level1Menus = bizProductCategories.stream().filter(bizProductCategory ->
             bizProductCategory.getPid() == 0
        ).map((menu)->
            menu.setChildren(getChildrens(menu,bizProductCategories));
            return menu;
        ).sorted((menu1,menu2)->
            // 菜单的排序
            return (menu1.getSort()==null?0:menu1.getSort())-(menu2.getSort()==null?0:menu2.getSort());
        ).collect(Collectors.toList());
        //        List<BizProductCategory> level1Menus = bizProductCategories.stream().filter((bizProductCategory) -> 
        //            return bizProductCategory.getPid() == 0;
        //        ).collect(Collectors.toList());
        List<BizProductCategory> page = new ArrayList<>();
        int start = (pageNum-1)*pageSize;
        for (int i = start; i < (start+pageSize > level1Menus.size() ? level1Menus.size() : start+pageSize); i++) 
            page.add(level1Menus.get(i));
        
        BizProductCategoryPlus bizProductCategoryPlus=new BizProductCategoryPlus();
        bizProductCategoryPlus.setRows(page);
        bizProductCategoryPlus.setTotal(level1Menus.size());
        return bizProductCategoryPlus;
    

上面调用的递归方法：

//    递归查找所有菜单的子菜单
//    root为当前菜单 all是所有菜单
    private List<BizProductCategory> getChildrens(BizProductCategory root, List<BizProductCategory> all)
        List<BizProductCategory> children = all.stream().filter(bizProductCategory -> 
            return bizProductCategory.getPid() == root.getId();
        ).map(bizProductCategory->
//   找到子菜单
            bizProductCategory.setChildren(getChildrens(bizProductCategory,all));
            return bizProductCategory;
        ).sorted((menu1,menu2)->
//    菜单的排序
            return (menu1.getSort()==null?0:menu1.getSort())-(menu2.getSort()==null?0:menu2.getSort());
        ).collect(Collectors.toList());
        return children;
    

controller层：

/**
     * 分类树形结构
     * @param pageNum
     * @param pageSize
     * @return
     * @author lt
     */
    @GetMapping("categoryTree")
    public R findAll(int  pageNum,int pageSize)
        BizProductCategoryPlus bizProductCategoryPlus = bizProductCategoryService.listWithTree(pageNum, pageSize);
//        BizProductCategoryPlus bizProductCategoryPlus=new BizProductCategoryPlus();
        return R.ok().setData(bizProductCategoryPlus);

这样就实现了对一级分类的分页查询