相信你也遇到过这种场景，判断二级目录属于哪个一级目录，一个员工属于哪个上级员工领导...

当Mybatis遇上目录树，有哪些解决方法？

一般来说，有xml直接实现和java代码递归赋值实现。

方式一：xml直接实现

这里列出category数据表数据

表结构如下

type表示分类类型，也就是目录级别，1表示一级目录，3表示三级目录

大家就不要关注数据类型规范了，比如这里id应该bigint，type明明可以tinyint之类的，我抓我看到的例子直接讲解。

  目录为甜点/蛋糕的id为1，而蛋糕和点心的father_id为1，目录为饼干/膨化的id为2，饼干、薯片、虾条的father_id就是2，一级目录id对应二级子目录的father_id，这就是所属对应关系，可以理解为父子关系。

实体类是mybatis-generator插件自动生成的

public class Category {
   
   
    private Integer id;
    private String name;
    private Integer type;
    private Integer fatherId;
    private String logo;
    private String slogan;
    private String catImage;
    private String bgColor;
    //=====篇幅原因，省掉Getter和Setter方法======
    ......
}

一般我们看到的商城，鼠标放到一级分类目录就会展示出二级分类目录。我们的需求是当鼠标移动到一级分类，我们需要提供二级分类和三级分类。

这里贴出需要返回给前端的聚合模型view object数据

/**
 * 二级分类VO
 */
public class CategoryVO {
   
   
    private Integer id;
    private String name;
    private String type;
    private Integer fatherId;
    // 三级分类vo list
    private List<SubCategoryVO> subCatList;
    //=====篇幅原因，省掉Getter和Setter方法======
    ......
}

public class SubCategoryVO {
   
   
    private Integer subId;
    private String subName;
    private String subType;
    private Integer subFatherId;
    //=====篇幅原因，省掉Getter和Setter方法======
    ......
}

这就涉及到自连接查询子目录的技巧了，我们试试查找father_id是1的子分类数据，也就是查询甜点/蛋糕分类下面的二级和三级分类，执行如下语句

SELECT
    f.id AS id,
    f.`name` AS `name`,
    f.type AS type,
    f.father_id AS fatherId,
    c.id AS subId,
    c.`name` AS subName,
    c.type AS subType,
    c.father_id AS subFatherId 
FROM
    category f
    LEFT JOIN category c ON f.id = c.father_id 
WHERE
    f.father_id = 1

结果如下

可以看到二级分类为蛋糕、点心时，有哪些对应的三级分类可以提供给前端，便于展示。

  我这里分为CategoryVO、SubCategoryVO ，而不是把所有属性放在一个VO，是为了便于理解。如果不用List集合，而把所有属性放在一个VO，前端收到的数据形式和你此时在数据库查询出来的一样，有多条蛋糕记录，底下对应着不同具体食品，这让前端不好处理也不符合逻辑，正常逻辑应该是只有一个蛋糕分类，然后这个分类里面有数组去装着蛋糕对应子分类才对。

  这里其实只用一个CategoryVO里面也可以处理，在后面第二种方式用java代码处理多级目录时，你会看到我只用了一个CategoryVO就能处理。

注意，二级分类的实体类CategoryVO有个

private List<SubCategoryVO> subCatList;

这个subCatList是为了存放三级分类的vo list，在xml中三级分类用了collection对应这个list

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd" >
<mapper namespace="com.me.mapper.CategoryMapperCustom" >
  <resultMap id="myCategoryVO" type="com.me.pojo.vo.CategoryVO">
    <id column="id" property="id"/>
    <result column="name" property="name"/>
    <result column="type" property="type"/>
      <!--
       column一定要在sql语句中找到，property一定要在对应实体类中找到
       因为sql用as写了别名，所以column才能用fatherId，如果不用别名，还是得写father_id
       -->
    <result column="fatherId" property="fatherId"/>

    <!--
      collection 标签：用于定义关联的list集合类型的封装规则
      property：对应三级分类的list属性名
      ofType：集合的类型，三级分类的vo
    -->
    <collection property="subCatList" ofType="com.me.pojo.vo.SubCategoryVO">
      <id column="subId" property="subId"/>
      <result column="subName" property="subName"/>
      <result column="subType" property="subType"/>
      <result column="subFatherId" property="subFatherId"/>
    </collection>
  </resultMap>

  <select id="getSubCatList" resultMap="myCategoryVO" parameterType="int">
    SELECT
        f.id as id,
        f.`name` as `name`,
        f.type as type,
        f.father_id as fatherId,
        c.id as subId,
        c.`name` as subName,
        c.type as subType,
        c.father_id as subFatherId
    FROM
        category f
    LEFT JOIN
        category c
    on
        f.id = c.father_id
    WHERE
        f.father_id = #{rootCatId}
  </select>
</mapper>

首先让前端展示在首页的一级分类，前端调用一级分类接口，我们只需要查询type为1的数据返回给前端，鼠标移动到一级分类，就调用获取子分类的接口，前端传入对应一级分类的id给后端，后端将这个id作为father_id去查询子分类。最后我们可以调用getSubCatList来得到所有目录。

    @Transactional(propagation = Propagation.SUPPORTS)
    @Override
    public List<CategoryVO> getSubCatList(Integer rootCatId) {
   
   
        return categoryMapperCustom.getSubCatList(rootCatId);
    }

最后数据就是这样，如下

{
   
   
    "status": 200,
    "msg": "OK",
    "data": [{
   
   
        "id": 11,
        "name": "蛋糕",
        "type": "2", <==================type=2表示二级目录
        "fatherId": 1,
        "subCatList": [{
   
   
            "subId": 37,
            "subName": "蒸蛋糕",
            "subType": "3", <================subType=3表示3级目录
            "subFatherId": 11
        }, {
   
   
            "subId": 38,
            "subName": "软面包",
            "subType": "3",
            "subFatherId": 11
        }, {
   
   
            "subId": 39,
            "subName": "脱水蛋糕",
            "subType": "3",
            "subFatherId": 11
        }, {
   
   
            "subId": 40,
            "subName": "马卡龙",
            "subType": "3",
            "subFatherId": 11
        }, {
   
   
            "subId": 41,
            "subName": "甜甜圈",
            "subType": "3",
            "subFatherId": 11
        }, {
   
   
            "subId": 42,
            "subName": "三明治",
            "subType": "3",
            "subFatherId": 11
        }, {
   
   
            "subId": 43,
            "subName": "铜锣烧",
            "subType": "3",
            "subFatherId": 11
        }]
    }, {
   
   
        "id": 12,
        "name": "点心",
        "type": "2",
        "fatherId": 1,
        "subCatList": [{
   
   
            "subId": 44,
            "subName": "肉松饼",
            "subType": "3",
            "subFatherId": 12
        }, {
   
   
            "subId": 45,
            "subName": "华夫饼",
            "subType": "3",
            "subFatherId": 12
        }, {
   
   
            "subId": 46,
            "subName": "沙琪玛",
            "subType": "3",
            "subFatherId": 12
        }, {
   
   
            "subId": 47,
            "subName": "鸡蛋卷",
            "subType": "3",
            "subFatherId": 12
        }, {
   
   
            "subId": 48,
            "subName": "蛋饼",
            "subType": "3",
            "subFatherId": 12
        }, {
   
   
            "subId": 49,
            "subName": "凤梨酥",
            "subType": "3",
            "subFatherId": 12
        }, {
   
   
            "subId": 50,
            "subName": "手撕面包",
            "subType": "3",
            "subFatherId": 12
        }]
    }]
}

方式二：java代码递归处理二级三级目录

此刻我换一个数据库例子，但是还是和上面一个处理一级二级三级分类的例子一样
数据表如下

表结构如下

和上一个例子大同小异，type依然表示目录级别

此刻需要返回给前端的VO如下，此刻我只写了一个CategoryVO，没有写子VO，可以对比前一种方式看看，道理都是一样的。

public class CategoryVO {
    private Integer id;
    private String name;
    private Integer type;
    private Integer parentId;
    private Integer orderNum;
    private Date createTime;
    private Date updateTime;
    private List<CategoryVO> childCategory = new ArrayList<>();
    //=====篇幅原因，省掉Getter和Setter方法======
    ......
}

    @Override
    public List<CategoryVO> listCategoryForCustomer(Integer parentId) {
   
   
        ArrayList<CategoryVO> categoryVOList = new ArrayList<>();
        recursivelyFindCategories(categoryVOList, parentId);
        return categoryVOList;
    }
    // 以该parentId对应的目录为根节点，查询下面所有子目录信息，categoryVOList是要返回给前端展示的聚合模型数据
    private void recursivelyFindCategories(List<CategoryVO> categoryVOList, Integer parentId) {
   
   
        // 递归获取所有子类别，并组合成为一个"目录树"
        List<Category> list= categoryMapper.selectCategoriesByParentId(parentId); // 通过父id查询子分类
        if (!CollectionUtils.isEmpty(list)) {
   
   
            for (int i = 0; i < list.size(); ++i) {
   
   
                Category category = list.get(i);
                CategoryVO categoryVO = new CategoryVO();
                BeanUtils.copyProperties(category, categoryVO);
                categoryVOList.add(categoryVO);
                // 这里当前目录id作为下一次的父id，查询有没有对应的子目录，getChildCategory()方法是返回定义的List<CategoryVO> childCategory
                recursivelyFindCategories(categoryVO.getChildCategory(), categoryVO.getId());
            }
        }
    }

XML文件如下：

......
  <resultMap id="BaseResultMap" type="com.me.mall.model.pojo.Category">
    <id column="id" jdbcType="INTEGER" property="id" />
    <result column="name" jdbcType="VARCHAR" property="name" />
    <result column="type" jdbcType="INTEGER" property="type" />
    <result column="parent_id" jdbcType="INTEGER" property="parentId" />
    <result column="order_num" jdbcType="INTEGER" property="orderNum" />
    <result column="create_time" jdbcType="TIMESTAMP" property="createTime" />
    <result column="update_time" jdbcType="TIMESTAMP" property="updateTime" />
  </resultMap>
  <sql id="Base_Column_List">
    id, `name`, `type`, parent_id, order_num, create_time, update_time
  </sql>
  <select id="selectCategoriesByParentId" parameterType="int" resultMap="BaseResultMap">
    select <include refid="Base_Column_List"/>
    from category
    where parent_id = #{parentId}
  </select>
  ......

我们手动查询模拟一下递归的过程，首先查询parent_id为3的二级分类

select *
from category
where parent_id = 3

结果递归查询的时候，又会发现parent_id=4时还有数据，即还有三级分类，我们手动查询试试

select *
from category
where parent_id = 4

示例数据如下：

{
   
   
    "status": 10000,
    "msg": "SUCCESS",
    "data": [
        {
   
   
            "id": 4,
            "name": "橘子橙子",
            "type": 2, <=================代表二级目录
            "parentId": 3,
            "orderNum": 1,
            "createTime": "2019-12-17T17:17:00.000+0000",
            "updateTime": "2019-12-28T08:25:10.000+0000",
            "childCategory": [ <===============代表还有三级目录
                {
   
   
                    "id": 19,
                    "name": "果冻橙",
                    "type": 3,
                    "parentId": 4,
                    "orderNum": 1,
                    "createTime": "2019-12-17T17:17:00.000+0000",
                    "updateTime": "2020-02-10T16:37:02.000+0000",
                    "childCategory": []
                }
            ]
        },
        {
   
   
            "id": 11,
            "name": "草莓",
            "type": 2,
            "parentId": 3,
            "orderNum": 2,
            "createTime": "2019-12-17T17:17:00.000+0000",
            "updateTime": "2019-12-28T07:44:42.000+0000",
            "childCategory": []
        },
        {
   
   
            "id": 12,
            "name": "奇异果",
            "type": 2,
            "parentId": 3,
            "orderNum": 3,
            "createTime": "2019-12-17T17:17:00.000+0000",
            "updateTime": "2019-12-28T08:25:12.000+0000",
            "childCategory": []
        },
        {
   
   
            "id": 14,
            "name": "车厘子",
            "type": 2,
            "parentId": 3,
            "orderNum": 4,
            "createTime": "2019-12-17T17:17:00.000+0000",
            "updateTime": "2019-12-28T08:25:12.000+0000",
            "childCategory": []
        },
        {
   
   
            "id": 28,
            "name": "其他水果",
            "type": 2,
            "parentId": 3,
            "orderNum": 4,
            "createTime": "2019-12-17T17:17:00.000+0000",
            "updateTime": "2019-12-28T08:25:12.000+0000",
            "childCategory": []
        }
    ]
}

欢迎一键三连~



有问题请留言，大家一起探讨学习



----------------------Talk is cheap, show me the code-----------------------