Java之使用递归查询多级树形结构数据(2)

业务层逻辑有所改动，先将所有数据存到List，再递归遍历到另一个List中返回

@Service
public class TestServiceImpl implements TestService {
@Autowired
private BaseMapper<Menu> baseMapper;
@Override
public List<Menu> findMenu() {
//数据查询
List<Menu> list = baseMapper.selectList(new QueryWrapper());
//新建一个用于接收数据的list
List<Menu> resultList = new ArrayList<>();
for (Menu result : list) {
if (result.getParentID() == 0) {
//调用方法给子类添加数据
resultList.add(getMenuTree(result, list));
}
}
return resultList;
}
private Menu getMenuTree(Menu result, List<Menu> list) {
for (Menu menu : list) {
//如果父类主键等于传过来实体类的ID
if (menu.getParentID() == result.getID()) {
if (result.getChildMenu() == null) {
result.setChildMenu(new ArrayList<>());
}
// 递归调用
result.getChildMenu().add(getMenuTree(menu, list));
}
}
return result;
}
}


sql语句

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


测试一下
使用postman发送请求

可以看到数据也成功以多级形势返回了