工作中可能会碰到一个表中存在父子关系,需要查询多级结构的树形数据场景(如图1-1),因此我们可以使用递归来实现

1-1

首先我建了一个测试的菜单表:

在这里插入图片描述
其中最顶级的菜单的父类ID是用0表示的,下面我们就来查询这张表

代码演示

建一个返回菜单数据的实体类

public class Menu {
    /** 主键id */
    private long ID;
    /** 父类主键 */
    private long parentid;
    /** 菜单名称 */
    private String name;
    /** 菜单类型 */
    private String type;
    /** 菜单编码 */
    private String code;
    /** 子类菜单数据 */
    @TableField(exist = false)
    private List<Menu> childMenu;
}

这里省略了get,set方法,childMenu是用于装子类数据的;@TableField(exist = false)表示该属性不为数据库表字段,但是必须使用

控制层代码:

@RestController
@RequestMapping("/test")
public class TestController {

    @Autowired
    public TestService testService;

    @GetMapping("/menu")
    public List<Menu> findMenu(@RequestParam long parentID) {
        return testService.findMenu(parentID);
    }
}

业务实现层代码:

public interface TestService {
    /**
     * 获取菜单多级信息
     * @param parentID 父类主键
     * @return List<Menu>
     */
    List<Menu> findMenu(long parentID);
}
@Service
public class TestServiceImpl implements TestService {
    @Autowired
    private BaseMapper<Menu> baseMapper;

    @Override
    public List<Menu> findMenu(long parentID) {
        QueryWrapper queryWrapper = new QueryWrapper<>();
        //查询条件
        queryWrapper.eq("parentid", parentID);
        List<Menu> list = baseMapper.selectList(queryWrapper);
        for (Menu menu : list) {
        	//递归子类数据
            menu.setChildMenu(findMenu(menu.getID()));
        }
        return list;
    }
}

先查询指定父类主键的数据,然后在循环中再次调用此方法查询此次数据是否存在子类数据;我这里持久层使用的是mybatisPlus,如果没引入需要另写sql
sql语句:

select id,`name`,`type`,`code`,parentID from menu where parentID = ?

测试

使用postman发送请求,参数为0,表示我要查询所有顶级菜单的数据
在这里插入图片描述
可以看到数据是以多级形势返回成功

[
    {
        "name": "商品一级菜单",
        "type": "商品",
        "code": "0001",
        "childMenu": [
            {
                "name": "商品二级菜单",
                "type": "商品",
                "code": "1001",
                "childMenu": [
                    {
                        "name": "商品三级菜单",
                        "type": "商品",
                        "code": "1101",
                        "childMenu": [],
                        "id": 6,
                        "parentID": 3
                    }
                ],
                "id": 3,
                "parentID": 1
            },
            {
                "name": "商品二级菜单(2)",
                "type": "商品",
                "code": "1002",
                "childMenu": [],
                "id": 4,
                "parentID": 1
            }
        ],
        "id": 1,
        "parentID": 0
    },
    {
        "name": "店铺一级菜单",
        "type": "店铺",
        "code": "0002",
        "childMenu": [
            {
                "name": "店铺二级菜单",
                "type": "店铺",
                "code": "2002",
                "childMenu": [],
                "id": 5,
                "parentID": 2
            }
        ],
        "id": 2,
        "parentID": 0
    }
]

快去试试吧

2022/07/06
注:后来想了想觉得循环访问数据库并不推荐,于是做了一些优化,具体可以看下一篇文章
地址:Java之使用递归查询多级树形结构数据(2)

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐