在我实习过程中,有一次碰到了一个遍历需求,大概是这样的:
我现在拿到了广州这个节点的id,然后我要把整棵树查出来,并且返回方式要和前端匹配,以父子嵌套方式返回,就是这样
然后这个问题折磨了我很久
在同事的帮助下才解决
首先,原对象字段是没有child字段的,是只有parentId进行关联,所以我要先创建一个DTO对象,假如有图示的两个字段要返回(因为真正的数据库表对象中是真的没有child字段的),那么我再加一个child字段
这样:
注意,要自己生成get和set方法
然后开始实现
要明白,根据传进来的id查询全部节点很简单,sql:
SELECT
id
name
parentId
FROM
[原表]
WHERE
tree_name = ( SELECT tree_name FROM [原表] WHERE id ='xxx' )
order by tree_level asc;
大概就是这么个意思
每个人需求不一样。
查出来很简单,但是进行构造有点复杂。
就是要构造一个数据结构。
public List<CustShowDto> showWithAll(String id) {
//list是查出来的所有节点,没有父子节点
List<Map<String, Object>> list = fmCgCustomInfoMapper.selectWithAll(id);
//l是节点的属性集合,就是类似于广州
List<CustShowDto> l = new ArrayList<>();
//
Map<String, List<CustShowDto>> parentMap = new HashMap<>();
//遍历集合,把所有的节点的父节点的parentid给赋值
for (int i = 0; i < list.size(); i++) {
CustShowDto custShowDto = JSONObject.parseObject(JSONObject.toJSONString(list.get(i)), CustShowDto.class);
l.add(custShowDto);
if (StringUtils.isNotEmpty(custShowDto.getCustTreeParentId())) {
if (parentMap.get(custShowDto.getCustTreeParentId()) == null) {
parentMap.put(custShowDto.getCustTreeParentId(), new ArrayList<>());
}
parentMap.get(custShowDto.getCustTreeParentId()).add(custShowDto);
}
}
List<CustShowDto> resList = new ArrayList<>();
//把赋值的节点添加到返回集合中
for (int i = 0; i < l.size(); i++) {
if (StringUtils.isEmpty(l.get(i).getCustTreeParentId())) {
resList.add(l.get(i));
}
if (parentMap.get(l.get(i).getCustId()) != null) {
l.get(i).setChildren(parentMap.get(l.get(i).getCustId()));
}
}
return resList;
}
主要是构造一个parentMap
,用parentId
作为key
,然后根据parentId
把子节点查出来,这个子节点是parentId
下面挂靠的节点,就像是:
然后把查出来的所有的子节点放入list中,这样就形成了
Map<String, List<CustShowDto>> parentMap = new HashMap<>();
这样大概的数据结构就出来了
然后构建一个
List<CustShowDto> resList = new ArrayList<>();
这个是返回的集合。
集合里面放的啥?Map呀
就是list [map1,map2,map3,...]
意思就是这么个意思,随缘看懂。
记录一下,免得下次又要做这种自己又不会!
全部代码:
public List<CustShowDto> showWithAll(String id) {
List<Map<String, Object>> list = fmCgCustomInfoMapper.selectWithAll(id);
List<CustShowDto> l = new ArrayList<>();
Map<String, List<CustShowDto>> parentMap = new HashMap<>();
for (int i = 0; i < list.size(); i++) {
CustShowDto custShowDto = JSONObject.parseObject(JSONObject.toJSONString(list.get(i)), CustShowDto.class);
l.add(custShowDto);
if (StringUtils.isNotEmpty(custShowDto.getCustTreeParentId())) {
if (parentMap.get(custShowDto.getCustTreeParentId()) == null) {
parentMap.put(custShowDto.getCustTreeParentId(), new ArrayList<>());
}
parentMap.get(custShowDto.getCustTreeParentId()).add(custShowDto);
}
}
List<CustShowDto> resList = new ArrayList<>();
for (int i = 0; i < l.size(); i++) {
if (StringUtils.isEmpty(l.get(i).getCustTreeParentId())) {
resList.add(l.get(i));
}
if (parentMap.get(l.get(i).getCustId()) != null) {
l.get(i).setChildren(parentMap.get(l.get(i).getCustId()));
}
}
return resList;
}
文章仅供参考!