菜单子节点的写法
- 1.测试数据
- 2.实现代码
- 3.获取父ID层级
1.测试数据
1.表结构SQL
CREATE TABLE `test` (
`id` int DEFAULT NULL,
`u_id` int DEFAULT NULL,
`p_u_id` int DEFAULT NULL
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci;
2.数据SQL
INSERT INTO test (id,u_id,p_u_id) VALUES
(1,1,0),
(2,2,0),
(3,3,0),
(4,11,1),
(5,12,1),
(6,21,2),
(7,22,2),
(8,211,21),
(9,221,22),
(10,222,22),
(11,223,22),
(12,2231,223),
(13,2232,223),
(14,0,-1);
3.查询表数据
select * from test;
2.实现代码
1.pojo代码
@TableName("test")
@Data
public class Test implements Serializable {
private Integer id;
private Integer uId;
private Integer pUId;
@TableField(exist = false)
private List<Test> testChains;
@TableField(exist =false)
private String treeId;
}
2.service层代码
@Override
public List<String> treeIds() {
List<Test> list = this.list();
return list.stream()
.filter(e -> e.getPUId() == -1)
.peek((tId) -> {
tId.setTestChains(getChildren(tId, list));
})
.toList();
}
private List<Test> getChildren(Test test, List<Test> all) {
return all.stream().filter(x->x.getPUId() != null).filter(categoryEntity -> {
return categoryEntity.getPUId().equals(test.getUId());
}).peek(treeId -> {
treeId.setTestChains(getChildren(treeId, all));
}).collect(Collectors.toList());
}
}
3.测试结果
[
{
"id": 14,
"testChains": [
{
"id": 1,
"testChains": [
{
"id": 4,
"testChains": [],
"treeId": null,
"puid": 1,
"uid": 11
},
{
"id": 5,
"testChains": [],
"treeId": null,
"puid": 1,
"uid": 12
}
],
"treeId": null,
"puid": 0,
"uid": 1
},
{
"id": 2,
"testChains": [
{
"id": 6,
"testChains": [
{
"id": 8,
"testChains": [],
"treeId": null,
"puid": 21,
"uid": 211
}
],
"treeId": null,
"puid": 2,
"uid": 21
},
{
"id": 7,
"testChains": [
{
"id": 9,
"testChains": [],
"treeId": null,
"puid": 22,
"uid": 221
},
{
"id": 10,
"testChains": [],
"treeId": null,
"puid": 22,
"uid": 222
},
{
"id": 11,
"testChains": [
{
"id": 12,
"testChains": [],
"treeId": null,
"puid": 223,
"uid": 2231
},
{
"id": 13,
"testChains": [],
"treeId": null,
"puid": 223,
"uid": 2232
}
],
"treeId": null,
"puid": 22,
"uid": 223
}
],
"treeId": null,
"puid": 2,
"uid": 22
}
],
"treeId": null,
"puid": 0,
"uid": 2
},
{
"id": 3,
"testChains": [],
"treeId": null,
"puid": 0,
"uid": 3
}
],
"treeId": null,
"puid": -1,
"uid": 0
}
]
3.获取父ID层级
1.pojo代码
@TableName("test")
@Data
public class Test implements Serializable {
private Integer id;
private Integer uId;
private Integer pUId;
@TableField(exist = false)
private List<Test> testChains;
@TableField(exist =false)
private String treeId;
}
2.service层代码
@Override
public List<String> treeIds() {
List<Test> list = this.list();
List<Test> collect = list.stream()
.peek((tId) -> {
tId.setTreeId(tId.getUId().toString());
tId.setTestChains(getChildren(tId, list));
})
.toList();
return collect.stream().map(Test::getTreeId).toList();
}
private List<Test> getChildren(Test test, List<Test> all) {
return all.stream().filter(x->x.getPUId() != null).filter(categoryEntity -> {
return categoryEntity.getPUId().equals(test.getUId());
}).peek(treeId -> {
treeId.setTreeId((test.getTreeId() == null? test.getUId() : test.getTreeId())+"/"+treeId.getUId());
treeId.setTestChains(getChildren(treeId, all));
}).collect(Collectors.toList());
}
3.测试结果
[
"0/1",
"0/2",
"0/3",
"0/1/11",
"0/1/12",
"0/2/21",
"0/2/22",
"0/2/21/211",
"0/2/22/221",
"0/2/22/222",
"0/2/22/223",
"0/2/22/223/2231",
"0/2/22/223/2232",
"0"
]
