控制层
@GetMapping
public ResponseServer queryType() {
List<Map<String, Object>> treeData = typeService.queryType();
return ResponseServer.success(treeData);
}
service:
@Override
public List<Map<String, Object>> queryType() {
List<Type> list = typeDao.queryTypeTree();
return getTreeList(0, list);
}
private List<Map<String, Object>> getTreeList(Integer pid, List list) {
List<Map<String, Object>> treeList = new ArrayList<Map<String, Object>>();
list.forEach(type -> {
Map<String, Object> map = null;
if (pid.equals(type.getPid())) {
map = new HashMap<>();
map.put(“id”, type.getId());
map.put(“name”, type.getName());
map.put(“pid”, type.getPid());
map.put(“children”, getTreeList(type.getId(), list));
}
if (map != null) {
treeList.add(map);
}
});
return treeList;
}