用递归调用函数来把存在父子级别的数据封装成一个树状结构

对于菜单,以及省市区联动这种数据, 要从数据库取出来传给前端,然后前端通过ztree之类的框架进行展示,那么这里我写了一个将数据库中的数据,转成dto的树状结构的工具方法

　代码

这里用了lombok框架,所以需要导lombok依赖

 <dependency>
    <groupId>org.projectlombok</groupId>
     <artifactId>lombok</artifactId>
     <version>1.16.20</version>
 </dependency>

这样只需要一个@Data注解就可以省去getter和setter了
idea还需要加一个插件LomBok plugin

　1.数据库对应的实体类

package com.zgd.demo.entity;

import lombok.Data;

/**
 * 菜单
 * @author zgd
 */
@Data
public class Menu {

    private Integer id;

    private String name;

    private Integer parentId;

    private Integer level;

    public Menu(Integer id, String name, Integer parentId, Integer level) {
        this.id = id;
        this.name = name;
        this.parentId = parentId;
        this.level = level;
    }
}

2. menu的dto类

package com.zgd.demo.dto;

import lombok.Data;

import java.util.List;

/**
 * menu的dto类
 * @author zgd
 */
@Data
public class MenuDto {

    private Integer id;

    private String name;

    private Integer parentId;

    private Integer level;

    private List<MenuDto> children;
}

4.工具类

package com.zgd.demo.util.bean;

import com.zgd.demo.dto.MenuDto;
import com.zgd.demo.dto.UserDto;
import com.zgd.demo.entity.Menu;
import com.zgd.demo.entity.User;
import org.springframework.beans.BeanUtils;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;

/**
 * 将数据库查出来的list数据抓成树状结构
 * @author zgd
 */
public class List2TreeUtil {

    /**
     * 根级父ID
     */
    private static final Integer ROOT_PARENT_ID = 0;

    /**
     * 将普通的entity的集合转成存在树状结构的集合
     * @param list
     * @return
     */
    public static List<MenuDto> getDtoTreeList(List<Menu> list){
        List<MenuDto> dtos = new ArrayList<>();
        if (list == null || list.isEmpty()) {
            return dtos;
        }
        for (Menu menu : list) {
            if (ROOT_PARENT_ID.equals(menu.getParentId())){
                //找到顶层目录
                MenuDto menuDto = new MenuDto();
                //将实体类的数据转成dto
                BeanUtils.copyProperties(menu,menuDto);
                //获取该实体类下的所有子级
                dtos.add(findChildren(menuDto,list));
            }
        }
        return dtos;
    }

    /**
     * 获取dto类下面的所有子级,存到dto中的children中
     * @param menuDto
     * @param list
     * @return
     */
    private static MenuDto findChildren(MenuDto menuDto, List<Menu> list) {
        for (Menu menu : list) {
            if (menuDto.getId().equals(menu.getParentId())){
                //此时menu是menuDto下的子级
                if (menuDto.getChildren() == null){
                    menuDto.setChildren(new ArrayList<MenuDto>());
                }
                MenuDto dto = new MenuDto();
                BeanUtils.copyProperties(menu,dto);
                //递归
                menuDto.getChildren().add(findChildren(dto,list));
            }
        }
        return menuDto;
    }

}

　4.测试

package com.zgd.demo;

import com.google.gson.Gson;
import com.zgd.demo.entity.Menu;
import com.zgd.demo.util.bean.List2TreeUtil;
import jdk.nashorn.internal.runtime.linker.LinkerCallSite;

import java.util.ArrayList;
import java.util.List;

/**
 * 运行demo
 * @author zgd
 */
public class DemoTest {

    public static void main(String[] args) {
        Gson gson = new Gson();
        List<Menu> list = new ArrayList<>();
        list.add(new Menu(1,"顶级目录",0,1));
        list.add(new Menu(2,"二级目录-a",1,2));
        list.add(new Menu(3,"二级目录-b",1,2));
        list.add(new Menu(4,"三级目录-a-a",2,3));
        list.add(new Menu(5,"三级目录-a-b",2,3));
        list.add(new Menu(6,"三级目录-a-c",2,3));
        list.add(new Menu(7,"三级目录-b-a",3,3));

        System.out.println(gson.toJson(List2TreeUtil.getDtoTreeList(list)));
    }
}

将打印出来的字符串进行格式化后