diff --git a/pqs-supervision/supervision-api/src/main/java/com/njcn/supervision/pojo/po/databank/LibAlgorithm.java b/pqs-supervision/supervision-api/src/main/java/com/njcn/supervision/pojo/po/databank/LibAlgorithm.java index 66566b5c0..37ab9856c 100644 --- a/pqs-supervision/supervision-api/src/main/java/com/njcn/supervision/pojo/po/databank/LibAlgorithm.java +++ b/pqs-supervision/supervision-api/src/main/java/com/njcn/supervision/pojo/po/databank/LibAlgorithm.java @@ -83,5 +83,8 @@ public class LibAlgorithm extends BaseEntity { @TableField(exist = false) private List children; + @TableField(exist = false) + private Integer level; + } \ No newline at end of file diff --git a/pqs-supervision/supervision-boot/src/main/java/com/njcn/supervision/service/databank/impl/LibAlgorithmServiceImpl.java b/pqs-supervision/supervision-boot/src/main/java/com/njcn/supervision/service/databank/impl/LibAlgorithmServiceImpl.java index 8eb117f9c..05573c232 100644 --- a/pqs-supervision/supervision-boot/src/main/java/com/njcn/supervision/service/databank/impl/LibAlgorithmServiceImpl.java +++ b/pqs-supervision/supervision-boot/src/main/java/com/njcn/supervision/service/databank/impl/LibAlgorithmServiceImpl.java @@ -93,12 +93,14 @@ public class LibAlgorithmServiceImpl extends ServiceImpl libAlgorithmList = this.lambdaQuery().eq(LibAlgorithm::getStatus,1).list(); List tree = libAlgorithmList.stream().filter(item -> item.getPid().equals("0")).peek(item -> { + item.setLevel(1); item.setChildren(getChildren(item, libAlgorithmList)); }).collect(Collectors.toList()); return tree; } private List getChildren(LibAlgorithm libAlgorithm, List all) { return all.stream().filter(item -> item.getPid().equals(libAlgorithm.getId())).peek(item -> { + item.setLevel(libAlgorithm.getLevel()+1); item.setChildren(getChildren(item, all)); }).collect(Collectors.toList()); }