mybatis-plus树形结构查询

MyBatis-Plus是一个MyBatis的增强工具，提供了许多实用的功能，其中就包括树形结构查询的支持。

在使用MyBatis-Plus进行树形结构查询时，需要使用到两个重要的注解：

- @TableName：用于指定实体类对应的数据表名。
- @TableField：用于指定实体类中的属性对应的数据表中的列名。

假设我们有一个表t_menu，它的结构如下：

| id | parent_id | name |
|----|-----------|----------|
| 1 | 0 | 菜单管理 |
| 2 | 1 | 添加菜单 |
| 3 | 1 | 修改菜单 |
| 4 | 2 | 添加一级菜单 |
| 5 | 2 | 添加二级菜单 |

我们需要查询出所有的菜单，并按照树形结构进行展示。代码如下：

首先，我们需要定义一个实体类Menu，用于映射t_menu表的数据。

@Data
@TableName("t_menu")
public class Menu implements Serializable {
    private static final long serialVersionUID = 1L;

    @TableId("id")
    private Long id;

    @TableField("name")
    private String name;

    @TableField("parent_id")
    private Long parentId;

    @TableField(exist = false)
    private List<Menu> children;
}

在这个实体类中，我们使用了@TableName和@TableField注解来指定实体类对应的数据表名以及属性对应的数据表中的列名。其中，@TableField(exist = false)表示这个属性不是数据表中的列，不需要映射。

接下来，我们可以使用MyBatis-Plus提供的Wrapper接口来构建查询条件，并使用selectList方法查询出所有的菜单。

@Service
public class MenuServiceImpl extends ServiceImpl<MenuMapper, Menu> implements MenuService {

    @Override
    public List<Menu> getMenuTree() {
        List<Menu> menuList = baseMapper.selectList(null);
        Map<Long, List<Menu>> menuMap = menuList.stream().collect(Collectors.groupingBy(Menu::getParentId));
        List<Menu> rootMenuList = menuMap.getOrDefault(0L, Collections.emptyList());
        for (Menu rootMenu : rootMenuList) {
            setChildren(rootMenu, menuMap);
        }
        return rootMenuList;
    }

    private void setChildren(Menu menu, Map<Long, List<Menu>> menuMap) {
        List<Menu> children = menuMap.getOrDefault(menu.getId(), Collections.emptyList());
        menu.setChildren(children);
        for (Menu child : children) {
            setChildren(child, menuMap);
        }
    }
}

在getMenuTree方法中，我们首先调用baseMapper.selectList(null)方法查询出所有的菜单。然后，使用Java 8的Stream API将菜单按照parentId进行分组。接着，从分组后的结果中取出parentId为0的菜单，这些菜单就是根菜单。最后，我们通过递归的方式将每个菜单的子菜单设置进去，从而构建出完整的树形结构。