From f287dfb5249618bf821626e76b7ce8d8f74d6b99 Mon Sep 17 00:00:00 2001
From: 无关风月 <443237572@qq.com>
Date: 星期三, 02 七月 2025 16:15:12 +0800
Subject: [PATCH] 保洁巡检本周代码

---
 ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TProjectDeptServiceImpl.java |   31 +++++++++++++++++++++++++++++++
 1 files changed, 31 insertions(+), 0 deletions(-)

diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TProjectDeptServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TProjectDeptServiceImpl.java
index 620f750..21b036e 100644
--- a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TProjectDeptServiceImpl.java
+++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TProjectDeptServiceImpl.java
@@ -1,10 +1,20 @@
 package com.ruoyi.system.service.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.ruoyi.common.basic.PageInfo;
+import com.ruoyi.common.core.domain.entity.SysUser;
+import com.ruoyi.system.mapper.SysUserMapper;
 import com.ruoyi.system.mapper.TProjectDeptMapper;
 import com.ruoyi.system.model.TProjectDept;
+import com.ruoyi.system.query.DeptListQuery;
 import com.ruoyi.system.service.TProjectDeptService;
+import com.ruoyi.system.vo.system.ProjectDeptListVO;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+
+import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * <p>
@@ -17,4 +27,25 @@
 @Service
 public class TProjectDeptServiceImpl extends ServiceImpl<TProjectDeptMapper, TProjectDept> implements TProjectDeptService {
 
+    @Autowired
+    private SysUserMapper sysUserMapper;
+    @Override
+    public PageInfo<ProjectDeptListVO> pageList(DeptListQuery query) {
+        List<SysUser> sysUsers = sysUserMapper.selectAllList();
+        PageInfo<ProjectDeptListVO> pageInfo = new PageInfo<>(query.getPageNum(), query.getPageSize());
+        List<ProjectDeptListVO> list = this.baseMapper.pageList(query,pageInfo);
+        List<TProjectDept> projectDepts = this.baseMapper.selectList(new LambdaQueryWrapper<TProjectDept>()
+                .ne(TProjectDept::getParentId, 0));
+        for (ProjectDeptListVO projectDeptListVO : list) {
+            List<TProjectDept> collect = projectDepts.stream().filter(e -> e.getParentId().equals(projectDeptListVO.getId()))
+                    .collect(Collectors.toList());
+            for (TProjectDept tProjectDept : collect) {
+                List<SysUser> collect1 = sysUsers.stream().filter(e -> e.getDeptId().equals(tProjectDept.getId())).collect(Collectors.toList());
+                tProjectDept.setUserCount(collect1.size());
+            }
+            projectDeptListVO.setChildren(collect);
+        }
+        pageInfo.setRecords(list);
+        return pageInfo;
+    }
 }

--
Gitblit v1.7.1