From 25b19e60e004290531f61fdf608d1adb5e531903 Mon Sep 17 00:00:00 2001
From: 无关风月 <443237572@qq.com>
Date: 星期六, 12 七月 2025 14:40:12 +0800
Subject: [PATCH] 保洁巡检本周代码

---
 ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TEarlyWarningController.java |   33 ++++++++++++++++++++++++---------
 1 files changed, 24 insertions(+), 9 deletions(-)

diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TEarlyWarningController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TEarlyWarningController.java
index 1e8dc3c..9c617f7 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TEarlyWarningController.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TEarlyWarningController.java
@@ -68,18 +68,30 @@
         List<TLocation> locationList = locationService.list();
         List<TTask> taskList = taskCleanerService.list();
         if (StringUtils.hasLength(query.getDeptName())){
-            List<String> projectIds = projectDeptService.lambdaQuery().like(TProjectDept::getProjectName, query.getDeptName()).list()
+            List<TProjectDept> list = projectDeptService.lambdaQuery().like(TProjectDept::getProjectName, query.getDeptName()).list();
+            List<String> listIds = projectDeptService.lambdaQuery().like(TProjectDept::getProjectName, query.getDeptName()).list()
                     .stream().map(TProjectDept::getId).collect(Collectors.toList());
+            for (TProjectDept tProjectDept : list) {
+                if (tProjectDept.getParentId().equals("0")){
+                    List<String> collect = projectDeptService.lambdaQuery().eq(TProjectDept::getParentId, tProjectDept.getId()).list()
+                            .stream().map(TProjectDept::getId).collect(Collectors.toList());
+                    listIds.addAll(collect);
+                    listIds.add(tProjectDept.getId());
+                }
+            }
             List<String> deptIds = deptService.lambdaQuery().like(TDept::getDeptName, query.getDeptName()).list()
                     .stream().map(TDept::getId).collect(Collectors.toList());
-            projectIds.addAll(deptIds);
-            if (projectIds.isEmpty()){
+            listIds.addAll(deptIds);
+            if (listIds.isEmpty()){
                 return R.ok(new PageInfo<>());
             }
+            List<Long> collect = sysUsers.stream().filter(e -> listIds.contains(e.getDeptId()))
+                    .map(SysUser::getUserId).collect(Collectors.toList());
+            query.setPatrolInspectorIds( collect);
             if (StringUtils.hasLength(query.getPhonenumber())){
                 List<Long> patrolInspectorIds = sysUsers.stream().filter(sysUser ->
-                        sysUser.getPhonenumber().equals(query.getPhonenumber())
-                                && projectIds.contains(sysUser.getDeptId())
+                        sysUser.getPhonenumber().contains(query.getPhonenumber())
+                                && listIds.contains(sysUser.getDeptId())
                 ).map(SysUser::getUserId).collect(Collectors.toList());
                 if (patrolInspectorIds.isEmpty()){
                     return R.ok(new PageInfo<>());
@@ -89,8 +101,9 @@
         }
         if (StringUtils.hasLength(query.getPhonenumber())){
             List<Long> patrolInspectorIds = sysUsers.stream().filter(sysUser ->
-                    sysUser.getPhonenumber().equals(query.getPhonenumber())
+                    sysUser.getPhonenumber().contains(query.getPhonenumber())
             ).map(SysUser::getUserId).collect(Collectors.toList());
+            query.setPatrolInspectorIds(patrolInspectorIds);
             if (!query.getPatrolInspectorIds().isEmpty()){
                 // 取交集
                 patrolInspectorIds = patrolInspectorIds.stream().filter(query.getPatrolInspectorIds()::contains).collect(Collectors.toList());
@@ -118,9 +131,11 @@
                 query.setLocationIds(collect);
                 return R.ok(new PageInfo<>());
             }else{
-                collect = collect.stream().filter(query.getLocationIds()::contains).collect(Collectors.toList());
-                if (collect.isEmpty()){
-                    return R.ok(new PageInfo<>());
+                if (query.getLocationIds()!=null && !query.getLocationIds().isEmpty()){
+                    collect = collect.stream().filter(query.getLocationIds()::contains).collect(Collectors.toList());
+                    if (collect.isEmpty()){
+                        return R.ok(new PageInfo<>());
+                    }
                 }
                 query.setLocationIds(collect);
             }

--
Gitblit v1.7.1