springboot 整合 mybatis plus -- 树状图

先根据 springboot 整合 mybatis plus  这篇生成各种文件

controller：

package com.lifan.controller;


import com.lifan.service.AcUserTreeService;
import com.lifan.vo.AcUserVo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;
import java.util.List;

/**
 * <p>
 *  前端控制器
 * </p>
 *
 * @author lifan
 * @since 2021-07-09
 */
@RestController
public class AcUserTreeController {

    @Autowired
    private AcUserTreeService acUserTreeService;

    @PostMapping("/treeQuery")
    public List<AcUserVo> SspFxProQuery() {
        List<AcUserVo> fxCompanyVos = acUserTreeService.formulaClothClassGetTree();

        return fxCompanyVos;
    }

}

service：

package com.lifan.service;

import com.lifan.vo.AcUserVo;
import java.util.List;

/**
 * <p>
 *  服务类
 * </p>
 *
 * @author lifan
 * @since 2021-07-09
 */
public interface AcUserTreeService {

    List<AcUserVo> formulaClothClassGetTree();
}

AcUserTreeServiceImpl ：

package com.lifan.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.lifan.entity.AcUserTree;
import com.lifan.mapper.AcUserTreeMapper;
import com.lifan.service.AcUserTreeService;
import com.lifan.vo.AcUserVo;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import java.util.ArrayList;
import java.util.List;

/**
 * <p>
 *  服务实现类
 * </p>
 *
 * @author lifan
 * @since 2021-07-09
 */
@Service
public class AcUserTreeServiceImpl implements AcUserTreeService {

    @Autowired
    private AcUserTreeMapper acUserTreeMapper;

    public List<AcUserVo> formulaClothClassGetTree() {
        QueryWrapper queryWrapper = new QueryWrapper();
        //只查询 启用状态的
//        queryWrapper.eq("`show`",'Y');
        List<AcUserTree> list = acUserTreeMapper.selectList(queryWrapper);
        List<AcUserVo> list1 = new ArrayList();
        for(AcUserTree sspFxCompany : list){
            AcUserVo fxCompanyVos = new AcUserVo();
            if(sspFxCompany.getPid().equals("1")){
                BeanUtils.copyProperties(sspFxCompany, fxCompanyVos);
                fxCompanyVos.setChildren(formulaClothClassNextTree(list,(sspFxCompany.getId())));
                list1.add(fxCompanyVos);

            }
        }
        return list1;
    }

    public List<AcUserVo> formulaClothClassNextTree(List<AcUserTree> list, Long formulaClothClassId) {
        List<AcUserVo> list1 = new ArrayList();
        for(AcUserTree sspFxCompany : list){
            AcUserVo fxCompanyVos = new AcUserVo();
            if(sspFxCompany.getPid().equals(formulaClothClassId.toString())){
                BeanUtils.copyProperties(sspFxCompany, fxCompanyVos);
                fxCompanyVos.setChildren(formulaClothClassNextTree1(list,sspFxCompany.getId()));
                list1.add(fxCompanyVos);
            }
        }
        return list1;
    }

    public List<AcUserVo> formulaClothClassNextTree1(List<AcUserTree> list, Long formulaClothClassId) {
        List<AcUserVo> list1 = new ArrayList();
        for(AcUserTree sspFxCompany : list){
            AcUserVo fxCompanyVos = new AcUserVo();
            if(sspFxCompany.getPid().equals(formulaClothClassId.toString())){
                BeanUtils.copyProperties(sspFxCompany, fxCompanyVos);
                fxCompanyVos.setChildren(formulaClothClassNextTree1(list,sspFxCompany.getId()));
                list1.add(fxCompanyVos);
            }
        }
        return list1;
    }

}

实体类：

package com.lifan.entity;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import java.io.Serializable;
import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.experimental.Accessors;

/**
 *
 * @author lifan
 * @since 2021-07-09
 */
@Data
@EqualsAndHashCode(callSuper = false)
@Accessors(chain = true)
public class AcUserTree implements Serializable {

    private static final long serialVersionUID = 1L;

    @TableId(value = "id", type = IdType.ID_WORKER_STR)
    private Long id;

    private String name;

    private String pid;

    //show 是数据库关键字，起个别名
    @TableField("`show`")
    private String show;
}

接受请求类：

package com.lifan.request;

import lombok.Data;


/**
 *
 * @author lifan
 * @since 2021-07-09
 */
@Data
public class AcUserTreeReq {

    private Long id;

    private String name;

    private String pid;

    private String show;

}

响应扩展类：

package com.lifan.vo;

import lombok.Data;

import java.util.List;

/**
 *
 * @author lifan
 * @since 2021-07-09
 */
@Data
public class AcUserVo {

    private Long id;

    private String name;

    private String pid;

    private String show;

    private List<AcUserVo> children;

}

Mapper 接口：

package com.lifan.mapper;

import com.lifan.entity.AcUserTree;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import org.apache.ibatis.annotations.Mapper;

/**
 * <p>
 *  Mapper 接口
 * </p>
 *
 * @author lifan
 * @since 2021-07-09
 */
@Mapper
public interface AcUserTreeMapper extends BaseMapper<AcUserTree> {

}

mapper.xml

<?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.lifan.mapper.AcUserTreeMapper">

</mapper>

测试：

数据库：

pid为1 是一级结构，一级结构的id对应二级结构的pid, 下表最多的是5级

SQL ：

/*
 Navicat Premium Data Transfer

 Source Server         : localhost
 Source Server Type    : MySQL
 Source Server Version : 80023
 Source Host           : localhost:3306
 Source Schema         : ac-new

 Target Server Type    : MySQL
 Target Server Version : 80023
 File Encoding         : 65001

 Date: 09/07/2021 18:46:34
*/

SET NAMES utf8mb4;
SET FOREIGN_KEY_CHECKS = 0;

-- ----------------------------
-- Table structure for ac_user_tree
-- ----------------------------
DROP TABLE IF EXISTS `ac_user_tree`;
CREATE TABLE `ac_user_tree`  (
  `id` bigint(0) NOT NULL,
  `name` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL,
  `pid` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL,
  `show` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL,
  PRIMARY KEY (`id`) USING BTREE
) ENGINE = InnoDB CHARACTER SET = utf8 COLLATE = utf8_general_ci ROW_FORMAT = Dynamic;

-- ----------------------------
-- Records of ac_user_tree
-- ----------------------------
INSERT INTO `ac_user_tree` VALUES (1001, 'li', '1', 'Y');
INSERT INTO `ac_user_tree` VALUES (1002, 'wang', '1', 'Y');
INSERT INTO `ac_user_tree` VALUES (1003, 'zhao', '1', 'Y');
INSERT INTO `ac_user_tree` VALUES (1004, 'li-1', '1001', 'Y');
INSERT INTO `ac_user_tree` VALUES (1005, 'wang-1', '1002', 'Y');
INSERT INTO `ac_user_tree` VALUES (1006, 'wang-2', '1002', 'Y');
INSERT INTO `ac_user_tree` VALUES (1007, 'wang-3', '1002', 'Y');
INSERT INTO `ac_user_tree` VALUES (1008, 'zhao-1', '1003', 'Y');
INSERT INTO `ac_user_tree` VALUES (1009, 'zhao-2', '1003', 'Y');
INSERT INTO `ac_user_tree` VALUES (1010, 'li-1-1', '1004', 'Y');
INSERT INTO `ac_user_tree` VALUES (1011, 'li-1-2', '1004', 'Y');
INSERT INTO `ac_user_tree` VALUES (1012, 'li-1-1-1', '1010', 'Y');
INSERT INTO `ac_user_tree` VALUES (1013, 'li-1-1-2', '1010', 'Y');
INSERT INTO `ac_user_tree` VALUES (1014, 'li-1-1-3', '1010', 'Y');
INSERT INTO `ac_user_tree` VALUES (1015, 'li-1-2-1', '1011', 'Y');
INSERT INTO `ac_user_tree` VALUES (1016, 'li-1-2-2', '1011', 'Y');
INSERT INTO `ac_user_tree` VALUES (1017, 'li-1-2-3', '1011', 'Y');
INSERT INTO `ac_user_tree` VALUES (1018, 'wang-1-1', '1005', 'Y');
INSERT INTO `ac_user_tree` VALUES (1019, 'wang-1-2', '1005', 'Y');
INSERT INTO `ac_user_tree` VALUES (1020, 'wang-1-1-1', '1018', 'Y');
INSERT INTO `ac_user_tree` VALUES (1021, 'li-1-1-1-1', '1012', 'Y');
INSERT INTO `ac_user_tree` VALUES (1022, 'li-1-1-1-2', '1012', 'Y');

SET FOREIGN_KEY_CHECKS = 1;