diff --git a/hz-pm-api/src/main/java/com/hz/pm/api/workbench/manage/WorkbenchManage.java b/hz-pm-api/src/main/java/com/hz/pm/api/workbench/manage/WorkbenchManage.java index 502ac73..14c4e19 100644 --- a/hz-pm-api/src/main/java/com/hz/pm/api/workbench/manage/WorkbenchManage.java +++ b/hz-pm-api/src/main/java/com/hz/pm/api/workbench/manage/WorkbenchManage.java @@ -818,6 +818,11 @@ public class WorkbenchManage { } List records = projectEarlyWarningService.list(queryWrapper); + if (Objects.nonNull(req.getUnitId())){ + List viewUnitIds = mhUnitCache.getViewChildIdsRecursion(req.getUnitId()); + List queryMhUnitIdList = CollUtils.convert(viewUnitIds, String::valueOf); + records = records.stream().filter(record -> queryMhUnitIdList.contains(record.getBuildOrgCode())).collect(Collectors.toList()); + } if (CollUtil.isNotEmpty(records)) { List projectCodes = CollUtils.fieldList(records, ProjectEarlyWarning::getProjectCode);