public List<SmsTag> selectList() {
List<SmsTag> smsTagList = this.list(new QueryWrapper<SmsTag>().eq("is_del", SmsTag.IS_USE_OK));
List<SmsTag> returnSmsTagList = new ArrayList<>();
for(SmsTag smsTag : smsTagList){
Long id = smsTag.getId();
Long parentId = smsTag.getParentId();
if (parentId.compareTo(0L) == 0){
List<SmsTag> smsTags = parentSmsTag(id, smsTagList);
smsTag.setSmsTagList(smsTags);
returnSmsTagList.add(smsTag);
}
}
return returnSmsTagList;
}
//递归
private List<SmsTag> parentSmsTag(Long parentId,List<SmsTag> smsTagList){
List<SmsTag> returnList = new ArrayList<>();
for (SmsTag smsTag : smsTagList){
if (parentId.compareTo(smsTag.getParentId()) == 0){
smsTag.setSmsTagList(this.parentSmsTag(smsTag.getId(),smsTagList));
returnList.add(smsTag);
}
}
return returnList
}
jdk1.8通过流来处理
public void test01() {
// 1、收集出 第一集的 数据
List<StatisticsCenter> collect = list.stream()
.filter((c) -> c.getParentNum() == 0)
// 与map功能一致,对list中每一项循环转化,封装childrenList
.peek((c) -> c.setChildren(getChildrenList(c.getNum(), list)))
//排序操作
.sorted(Comparator.comparingInt(StatisticsCenter::getTotal))
.collect(Collectors.toList());
collect.forEach(System.out::println);
}
//2、获取子节点集合
public List<StatisticsCenter> getChildrenList(Integer pNum, List<StatisticsCenter> old) {
return old.stream()
.filter((c) -> pNum.equals(c.getParentNum()))
.peek((c) -> c.setChildren(getChildrenList(c.getNum(), old)))
.sorted(Comparator.comparingInt(StatisticsCenter::getTotal))
.collect(Collectors.toList());
}
@Data
@Accessors(chain = true)
class StatisticsCenter implements Serializable {
private static final long serialVersionUID = 1L;
private Integer id;
private Integer num;
private String title;
private Integer parentNum;
private Integer total;
private List<StatisticsCenter> children;
}