首页预警数据单位筛选修改

hz-pm-api/src/main/java/com/hz/pm/api/workbench/manage/WorkbenchManage.java

@@ -818,6 +818,11 @@ public class WorkbenchManage {
         }
 
         List<ProjectEarlyWarning> records = projectEarlyWarningService.list(queryWrapper);
+        if (Objects.nonNull(req.getUnitId())){
+            List<Long> viewUnitIds = mhUnitCache.getViewChildIdsRecursion(req.getUnitId());
+            List<String> queryMhUnitIdList = CollUtils.convert(viewUnitIds, String::valueOf);
+            records = records.stream().filter(record -> queryMhUnitIdList.contains(record.getBuildOrgCode())).collect(Collectors.toList());
+        }
         if (CollUtil.isNotEmpty(records)) {
             List<String> projectCodes = CollUtils.fieldList(records, ProjectEarlyWarning::getProjectCode);