修改单位树视频数量统计

amos-boot-module/amos-boot-module-biz/amos-boot-module-equip-biz/src/main/java/com/yeejoin/equipmanage/service/impl/BuildingServiceImpl.java

@@ -1245,7 +1245,11 @@ public class BuildingServiceImpl extends ServiceImpl<BuildingMapper, Building> i
             vo.setTotal(countMap.get(orgUsrDto.getBizOrgCode()) == null ? 0L : countMap.get(orgUsrDto.getBizOrgCode()));
             return vo;
         }).collect(Collectors.toList());
-        return companyList.stream().filter(a -> companyList.stream().noneMatch(c -> c.getId().toString().equals(a.getParentId()))).peek(b -> b.setChildren(getCompanyBuildingChildrenTree(b.getId().toString(), companyList))).collect(Collectors.toList());
+        return companyList.stream().filter(a -> companyList.stream().noneMatch(c -> c.getId().toString().equals(a.getParentId())))
+                .peek(b -> {
+                    b.setChildren(getCompanyBuildingChildrenTree(b.getId().toString(), companyList));
+                    b.setTotal(b.getChildren().stream().filter(c -> c.getTotal() != null).mapToLong(BuildingTreeVo::getTotal).sum());
+                }).collect(Collectors.toList());
 
     }