java后端樹形菜單結構

課程分類的建表語句

CREATE TABLE `edu_subject` (
  `id` char(19) NOT NULL COMMENT '課程類別ID',
  `title` varchar(10) NOT NULL COMMENT '類別名稱',
  `parent_id` char(19) NOT NULL DEFAULT '0' COMMENT '父ID',
  `sort` int(10) unsigned NOT NULL DEFAULT '0' COMMENT '排序字段',
  `gmt_create` datetime NOT NULL COMMENT '創建時間',
  `gmt_modified` datetime NOT NULL COMMENT '更新時間',
  PRIMARY KEY (`id`),
  KEY `idx_parent_id` (`parent_id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 ROW_FORMAT=COMPACT COMMENT='課程科目';

#
# Data for table "edu_subject"
#

INSERT INTO `edu_subject` VALUES ('1178214681118568449','后端開發','0',1,'2019-09-29 15:47:25','2019-09-29 15:47:25'),('1178214681139539969','Java','1178214681118568449',1,'2019-09-29 15:47:25','2019-09-29 15:47:25'),('1178214681181483010','前端開發','0',3,'2019-09-29 15:47:25','2019-09-29 15:47:25'),('1178214681210843137','JavaScript','1178214681181483010',4,'2019-09-29 15:47:25','2019-09-29 15:47:25'),('1178214681231814658','雲計算','0',5,'2019-09-29 15:47:25','2019-09-29 15:47:25'),('1178214681252786178','Docker','1178214681231814658',5,'2019-09-29 15:47:25','2019-09-29 15:47:25'),('1178214681294729217','Linux','1178214681231814658',6,'2019-09-29 15:47:25','2019-09-29 15:47:25'),('1178214681324089345','系統/運維','0',7,'2019-09-29 15:47:25','2019-09-29 15:47:25'),('1178214681353449473','Linux','1178214681324089345',7,'2019-09-29 15:47:25','2019-09-29 15:47:25'),('1178214681382809602','Windows','1178214681324089345',8,'2019-09-29 15:47:25','2019-09-29 15:47:25'),('1178214681399586817','數據庫','0',9,'2019-09-29 15:47:25','2019-09-29 15:47:25'),('1178214681428946945','MySQL','1178214681399586817',9,'2019-09-29 15:47:25','2019-09-29 15:47:25'),('1178214681454112770','MongoDB','1178214681399586817',10,'2019-09-29 15:47:25','2019-09-29 15:47:25'),('1178214681483472898','大數據','0',11,'2019-09-29 15:47:25','2019-09-29 15:47:25'),('1178214681504444418','Hadoop','1178214681483472898',11,'2019-09-29 15:47:25','2019-09-29 15:47:25'),('1178214681529610242','Spark','1178214681483472898',12,'2019-09-29 15:47:25','2019-09-29 15:47:25'),('1178214681554776066','人工智能','0',13,'2019-09-29 15:47:25','2019-09-29 15:47:25'),('1178214681584136193','Python','1178214681554776066',13,'2019-09-29 15:47:25','2019-09-29 15:47:25'),('1178214681613496321','編程語言','0',14,'2019-09-29 15:47:25','2019-09-29 15:47:25'),('1178214681626079234','Java','1178214681613496321',14,'2019-09-29 15:47:25','2019-09-29 15:47:25'),('1178585108407984130','Python','1178214681118568449',2,'2019-09-30 16:19:22','2019-09-30 16:19:22'),('1178585108454121473','HTML/CSS','1178214681181483010',3,'2019-09-30 16:19:22','2019-09-30 16:19:22');

課程對應的實體

package com.atguigu.eduservice.entity;

import com.baomidou.mybatisplus.annotation.IdType;
import java.util.Date;
import com.baomidou.mybatisplus.annotation.TableId;
import java.io.Serializable;
import java.util.List;

import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.experimental.Accessors;

/**
 * <p>
 * 課程科目
 * </p>
 *
 * @author wangjie
 * @since 2020-07-08
 */
@Data
@EqualsAndHashCode(callSuper = false)
@Accessors(chain = true)
@ApiModel(value="EduSubject對象", description="課程科目")
public class EduSubject implements Serializable {

    private static final long serialVersionUID = 1L;

    @ApiModelProperty(value = "課程類別ID")
    @TableId(value = "id", type = IdType.ID_WORKER_STR)
    private String id;

    @ApiModelProperty(value = "類別名稱")
    private String title;

    @ApiModelProperty(value = "父ID")
    private String parentId;

    @ApiModelProperty(value = "排序字段")
    private Integer sort;

    @ApiModelProperty(value = "創建時間")
    private Date gmtCreate;

    @ApiModelProperty(value = "更新時間")
    private Date gmtModified;

    @ApiModelProperty(value = "子菜單")
    private List<EduSubject> children;


}

　　獲取樹形結構的主要代碼

package com.atguigu.eduservice.service.impl;

import com.atguigu.eduservice.entity.EduSubject;
import com.atguigu.eduservice.mapper.EduSubjectMapper;
import com.atguigu.eduservice.service.EduSubjectService;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springframework.stereotype.Service;

import java.util.*;

/**
 * <p>
 * 課程科目 服務實現類
 * </p>
 *
 * @author wangjie
 * @since 2020-07-08
 */
@Service
public class EduSubjectServiceImpl extends ServiceImpl<EduSubjectMapper, EduSubject> implements EduSubjectService {

    @Override
    public List<EduSubject> findAllSubjectList() {
        return findTree();
    }

    /*
     * 排序,根據order排序
     */
    public Comparator<EduSubject> order() {
        Comparator<EduSubject> comparator = new Comparator<EduSubject>() {
            @Override
            public int compare(EduSubject o1, EduSubject o2) {
                if (o1.getSort() != o2.getSort()) {
                    return o1.getSort() - o2.getSort();
                }
                return 0;
            }
        };
        return comparator;
    }

    /**
     * 生成樹
     *
     * @return
     */
    public List<EduSubject> findTree() {
        //最終的樹形結構
        List<EduSubject> data = new ArrayList<>();
        try {//查詢所有菜單
            QueryWrapper<EduSubject>  subjectWrapper = new QueryWrapper<>();
            subjectWrapper.select(EduSubject.class,info->!info.getColumn().equals("children"));
            List<EduSubject> allMenu = baseMapper.selectList(subjectWrapper);
            //根節點
            List<EduSubject> rootMenu = new ArrayList<EduSubject>();
            for (EduSubject nav : allMenu) {
                if (nav.getParentId().equals("0")) {//父節點是0的，為根節點。
                    rootMenu.add(nav);
                }
            }
            /* 根據Menu類的order排序 */
            Collections.sort(rootMenu, order());
            //為根菜單設置子菜單，getClild是遞歸調用的
            for (EduSubject nav : rootMenu) {
                /* 獲取根節點下的所有子節點 使用getChild方法*/
                List<EduSubject> childList = getChild(nav.getId(), allMenu);
                nav.setChildren(childList);//給根節點設置子節點
            }
            data.addAll(rootMenu);
            return data;
        } catch (Exception e) {
            e.printStackTrace();
            return data;
        }
    }

    /**
     * 獲取子節點
     *
     * @param id      父節點id
     * @param allMenu 所有菜單列表
     * @return 每個根節點下，所有子菜單列表
     */
    public List<EduSubject> getChild(String id, List<EduSubject> allMenu) {
        //子菜單
        List<EduSubject> childList = new ArrayList<EduSubject>();
        for (EduSubject nav : allMenu) {
            // 遍歷所有節點，將所有菜單的父id與傳過來的根節點的id比較
            //相等說明：為該根節點的子節點。
            if (nav.getParentId().equals(id)) {
                childList.add(nav);
            }
        }
        //遞歸
        for (EduSubject nav : childList) {
            nav.setChildren(getChild(nav.getId(), allMenu));
        }
        Collections.sort(childList, order());//排序
        //如果節點下沒有子節點，返回一個空List（遞歸退出）
        if (childList.size() == 0) {
            return new ArrayList<EduSubject>();
        }
        return childList;
    }
}