From 52225bdefba7d99bc07952e461f2a841948e46b2 Mon Sep 17 00:00:00 2001
From: 无关风月 <443237572@qq.com>
Date: 星期五, 11 七月 2025 21:10:49 +0800
Subject: [PATCH] 保洁巡检本周代码

---
 ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TTaskController.java |  297 ++++++++++++++++++++++++++++++++++++++++++++++++++++-------
 1 files changed, 261 insertions(+), 36 deletions(-)

diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TTaskController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TTaskController.java
index 8b7926b..36cd49f 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TTaskController.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TTaskController.java
@@ -1,8 +1,17 @@
 package com.ruoyi.web.controller.api;
-import com.google.common.collect.Lists;
+import java.io.IOException;
+import java.math.BigDecimal;
+import java.net.URLEncoder;
 import java.time.LocalDateTime;
 
+import cn.afterturn.easypoi.excel.ExcelExportUtil;
+import cn.afterturn.easypoi.excel.entity.ExportParams;
+import cn.afterturn.easypoi.excel.entity.TemplateExportParams;
+import cn.hutool.core.io.resource.ClassPathResource;
 import com.ruoyi.common.core.domain.BaseModel;
+import com.ruoyi.common.utils.CodeGenerateUtils;
+import com.ruoyi.common.utils.DateUtils;
+import com.ruoyi.common.utils.WebUtils;
 import com.ruoyi.framework.web.service.TokenService;
 import com.ruoyi.system.dto.TTaskAuditBatchDTO;
 import com.ruoyi.system.dto.TTaskAuditDTO;
@@ -16,22 +25,23 @@
 import com.ruoyi.common.enums.BusinessType;
 import com.ruoyi.common.utils.bean.BeanUtils;
 import com.ruoyi.system.dto.TTaskDTO;
-import com.ruoyi.system.dto.TTemplateDTO;
 import com.ruoyi.system.model.*;
-import com.ruoyi.system.query.ProjectDeptListQuery;
 import com.ruoyi.system.query.TaskListQuery;
 import com.ruoyi.system.service.*;
 import com.ruoyi.system.vo.system.*;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
+import org.apache.poi.ss.usermodel.Workbook;
 import org.springframework.util.StringUtils;
+import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
 
 import javax.annotation.Resource;
+import javax.servlet.ServletOutputStream;
+import javax.servlet.http.HttpServletResponse;
 import javax.validation.Valid;
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.List;
+import java.time.format.DateTimeFormatter;
+import java.util.*;
 import java.util.stream.Collectors;
 
 /**
@@ -47,8 +57,11 @@
 @RequestMapping("/t-task")
 public class TTaskController {
     @Resource
-    private TTaskService taskService;
-
+    private TTaskCleanService taskCleanerService;
+    @Resource
+    private TNoticeService noticeService;
+    @Resource
+    private TNoticeSetService noticeSetService;
     @Resource
     private TTaskDetailService taskDetailService;
 
@@ -59,10 +72,7 @@
     private TLocationService locationService;
     @Resource
     private TProjectDeptService projectDeptService;
-    @Resource
-    private TCleanerService cleanerService;
-    @Resource
-    private TTaskService tTaskService;
+
     @Resource
     private TTaskDetailService tTaskDetailService;
     @Resource
@@ -73,6 +83,8 @@
     private TDictDataService dictDataService;
     @Resource
     private TokenService tokenService;
+    @Resource
+    private TCleanerService cleanerService;
 
     @ApiOperation(value = "任务记录分页列表")
     @PostMapping(value = "/pageList")
@@ -80,18 +92,30 @@
         List<SysUser> sysUsers = sysUserService.selectAllList();
         List<TLocation> locationList = locationService.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<>());
@@ -101,8 +125,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());
@@ -120,7 +145,7 @@
                 return R.ok(new PageInfo<>());
             }
         }
-        if (query.getLocationType()!=null){
+        if (StringUtils.hasLength(query.getLocationType())){
             List<String> collect = locationList.stream().filter(e -> e.getLocationType().equals(query.getLocationType())).map(TLocation::getId)
                     .collect(Collectors.toList());
             if (collect.isEmpty()){
@@ -131,8 +156,8 @@
         if (StringUtils.hasLength(query.getLocationName())){
             List<String> collect = locationList.stream().filter(e -> e.getLocationName().contains(query.getLocationName())).map(TLocation::getId)
                     .collect(Collectors.toList());
+            query.setLocationIds(collect);
             if (collect.isEmpty()){
-                query.setLocationIds(collect);
                 return R.ok(new PageInfo<>());
             }else{
                 collect = collect.stream().filter(query.getLocationIds()::contains).collect(Collectors.toList());
@@ -142,40 +167,88 @@
                 query.setLocationIds(collect);
             }
         }
-        return R.ok(taskService.pageList(query));
+        return R.ok(taskCleanerService.pageList(query));
     }
 
     @Log(title = "新增任务", businessType = BusinessType.INSERT)
     @ApiOperation(value = "新增任务")
     @PostMapping(value = "/add")
     public R<Boolean> add(@RequestBody TTaskDTO dto) {
-        taskService.save(dto);
+        String nameAndCode = CodeGenerateUtils.generateVolumeSn();
+        dto.setTaskName(nameAndCode);
+        dto.setTaskCode(nameAndCode);
+        dto.setTaskType(2);
+        String implementTime1 = dto.getImplementTime1();
+        // 转化为LocalDateTime类型
+        dto.setImplementTime(LocalDateTime.parse(implementTime1, DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")));
+        TCleaner byId = cleanerService.getById(dto.getCleanerId());
+        dto.setProjectId(byId.getProjectId());
+        taskCleanerService.save(dto);
         return R.ok();
     }
     @Log(title = "审核任务", businessType = BusinessType.UPDATE)
     @ApiOperation(value = "审核任务")
     @PostMapping(value = "/audit")
-    public R<Boolean> audit(@RequestBody @Valid TTaskAuditDTO dto) {
+    public R<Boolean> audit(@RequestBody  TTaskAuditDTO dto) {
         dto.setAuditTime(LocalDateTime.now());
+        TTask task = taskCleanerService.getById(dto.getTaskId());
+        TTaskDetail one = taskDetailService.lambdaQuery().eq(TTaskDetail::getTaskId, dto.getTaskId())
+                .eq(TTaskDetail::getHandleType,1)
+                .orderByDesc(BaseModel::getCreateTime).last("limit 1").one();
+        dto.setClearStatus(one.getClearStatus());
+        TLocation location = locationService.getById(task.getLocationId());
+        TNotice tNotice = new TNotice();
+        TNoticeSet noticeSet = noticeSetService.lambdaQuery().eq(TNoticeSet::getType, 2).last("limit 1")
+                .one();
+        tNotice.setUserId(task.getPatrolInspector());
+        tNotice.setStatus(1);
+        tNotice.setDataId(task.getId());
+        tNotice.setNoticeSetType(noticeSet.getNoticeType());
         if (dto.getAuditStatus()==2){
             dto.setHandleType(3);
             // 将任务修改为待整改
-            TTask byId = taskService.getById(dto.getTaskId());
-            byId.setStatus(4);
-            taskService.updateById(byId);
+            task.setStatus(4);
+            taskCleanerService.updateById(task);
+            // 增加消息
+            tNotice.setNoticeType(1);
+            tNotice.setNoticeContent("【"+location.getLocationName()+"】任务不合格,请重新上传!");
+
         }else{
+
             dto.setHandleType(2);
+            // 如果是初次审核就通过 将状态设置为已完成
+            List<TTaskDetail> list = tTaskDetailService.lambdaQuery()
+                    .eq(TTaskDetail::getHandleType,1)
+                    .eq(TTaskDetail::getTaskId, dto.getTaskId()).list();
+                if (list.size()<2){
+                    task.setStatus(6);
+                    taskCleanerService.updateById(task);
+                    tNotice.setNoticeType(2);
+                    tNotice.setNoticeContent("【"+location.getLocationName()+"】任务已通过");
+                }else{
+                    task.setStatus(5);
+                    taskCleanerService.updateById(task);
+                    tNotice.setNoticeType(2);
+                    tNotice.setNoticeContent("【"+location.getLocationName()+"】任务已通过");
+                }
         }
+        noticeService.save(tNotice);
         dto.setAuditPerson(tokenService.getLoginUser().getUserId()+"");
         taskDetailService.save(dto);
+        one.setAuditPerson(tokenService.getLoginUser().getUserId()+"");
+        one.setAuditTime(LocalDateTime.now());
+        one.setAuditRemark(dto.getAuditRemark());
+        one.setAuditStatus(dto.getAuditStatus());
+        taskDetailService.updateById(one);
         return R.ok();
     }
     @Log(title = "批量审核任务", businessType = BusinessType.UPDATE)
     @ApiOperation(value = "批量审核任务")
     @PostMapping(value = "/auditBatch")
-    public R<Boolean> auditBatch(@RequestBody @Valid TTaskAuditBatchDTO dto) {
+    public R<Boolean> auditBatch(@RequestBody  TTaskAuditBatchDTO dto) {
         List<TTaskDetail> tTaskDetails = new ArrayList<>();
         for (String s : dto.getTaskIds().split(",")) {
+            TTask byId = taskCleanerService.getById(s);
             TTaskDetail tTaskDetail = new TTaskDetail();
             tTaskDetail.setTaskId(s);
             tTaskDetail.setAuditTime(LocalDateTime.now());
@@ -183,11 +256,19 @@
             if (dto.getAuditStatus()==2){
                 tTaskDetail.setHandleType(3);
                 // 将任务修改为待整改
-                TTask byId = taskService.getById(tTaskDetail.getTaskId());
                 byId.setStatus(4);
-                taskService.updateById(byId);
+                taskCleanerService.updateById(byId);
             }else{
                 tTaskDetail.setHandleType(2);
+                // 如果是初次审核就通过 将状态设置为已完成
+                List<TTaskDetail> list = taskDetailService.lambdaQuery().eq(TTaskDetail::getTaskId, s).list();
+                if (list.isEmpty()){
+                    byId.setStatus(6);
+                    taskCleanerService.updateById(byId);
+                }else{
+                    byId.setStatus(5);
+                    taskCleanerService.updateById(byId);
+                }
             }
             tTaskDetails.add(tTaskDetail);
         }
@@ -200,13 +281,14 @@
         List<TTaskDetail> list = taskDetailService.lambdaQuery().eq(TTaskDetail::getTaskId, id)
                 .orderByDesc(BaseModel::getCreateTime).list();
         TaskDetailVO taskDetailVO = new TaskDetailVO();
-        TTask byId = taskService.getById(id);
+        TTask byId = taskCleanerService.getById(id);
         BeanUtils.copyProperties(byId, taskDetailVO);
         TLocation byId1 = locationService.getById(byId.getLocationId());
         TLocationType byId2 = locationTypeService.getById(byId1.getLocationType());
         taskDetailVO.setLocationAddress(byId1.getLocationAddress());
         taskDetailVO.setLocationIcon(byId2.getLocationIcon());
-        taskDetailVO.setLocationName(byId2.getLocationName());
+        taskDetailVO.setLocationName(byId1.getLocationName());
+        taskDetailVO.setLocationTypeName(byId2.getLocationName());
         TTaskDetail tTaskDetail = list.stream().filter(e -> e.getHandleType() == 1).findFirst().orElse(null);
         if (tTaskDetail!=null &&  tTaskDetail.getUnqualified()!=null){
             TDictData byId3 = dictDataService.getById(tTaskDetail.getUnqualified());
@@ -214,8 +296,43 @@
                 tTaskDetail.setUnqualifiedName(byId3.getDataContent());
             }
         }
+        if (tTaskDetail!=null){
+            tTaskDetail.setFinishTime(tTaskDetail.getCreateTime());
+        }
         taskDetailVO.setTaskDetail(tTaskDetail);
         taskDetailVO.setRecords(list);
+        return R.ok(taskDetailVO);
+    }
+    @ApiOperation(value = "详情任务-根据任务编号")
+    @GetMapping(value = "/detailByTaskCode")
+    public R<TaskDetailVO> detailByTaskCode(@RequestParam String taskCode) {
+        TTask task = taskCleanerService.lambdaQuery().eq(TTask::getTaskCode, taskCode)
+                .last("limit 1").one();
+        TaskDetailVO taskDetailVO = new TaskDetailVO();
+        if (task!=null){
+            String id = task.getId();
+            List<TTaskDetail> list = taskDetailService.lambdaQuery().eq(TTaskDetail::getTaskId, id)
+                    .orderByDesc(BaseModel::getCreateTime).list();
+            TTask byId = taskCleanerService.getById(id);
+            BeanUtils.copyProperties(byId, taskDetailVO);
+            TLocation byId1 = locationService.getById(byId.getLocationId());
+            TLocationType byId2 = locationTypeService.getById(byId1.getLocationType());
+            taskDetailVO.setLocationAddress(byId1.getLocationAddress());
+            taskDetailVO.setLocationIcon(byId2.getLocationIcon());
+            taskDetailVO.setLocationName(byId2.getLocationName());
+            TTaskDetail tTaskDetail = list.stream().filter(e -> e.getHandleType() == 1).findFirst().orElse(null);
+            if (tTaskDetail!=null &&  tTaskDetail.getUnqualified()!=null){
+                TDictData byId3 = dictDataService.getById(tTaskDetail.getUnqualified());
+                if (byId3!=null){
+                    tTaskDetail.setUnqualifiedName(byId3.getDataContent());
+                }
+            }
+            taskDetailVO.setTaskDetail(tTaskDetail);
+            taskDetailVO.setRecords(list);
+        }else{
+            return R.fail("任务不存在");
+        }
+
         return R.ok(taskDetailVO);
     }
     @ApiOperation(value = "详情任务-操作记录-详情")
@@ -224,7 +341,7 @@
 
         TaskRecordDetailVO taskDetailVO = new TaskRecordDetailVO();
         TTaskDetail byId4 = taskDetailService.getById(id);
-        TTask byId = taskService.getById(byId4.getTaskId());
+        TTask byId = taskCleanerService.getById(byId4.getTaskId());
         BeanUtils.copyProperties(byId, taskDetailVO);
         TLocation byId1 = locationService.getById(byId.getLocationId());
         TLocationType byId2 = locationTypeService.getById(byId1.getLocationType());
@@ -237,9 +354,11 @@
                 byId4.setUnqualifiedName(byId3.getDataContent());
             }
         }
-        SysUser sysUser = sysUserService.selectUserById(Long.valueOf(byId4.getAuditPerson()));
-        if (sysUser!=null){
-            byId4.setAuditPersonName(sysUser.getUserName());
+        if (byId4.getAuditPerson()!=null){
+            SysUser sysUser = sysUserService.selectUserById(Long.valueOf(byId4.getAuditPerson()));
+            if (sysUser!=null){
+                byId4.setAuditPersonName(sysUser.getUserName());
+            }
         }
         taskDetailVO.setTaskDetail(byId4);
         return R.ok(taskDetailVO);
@@ -249,9 +368,115 @@
     @DeleteMapping(value = "/deleteByIds")
     public R<Boolean> deleteByIds(@RequestParam String ids) {
         String[] split = ids.split(",");
-        taskService.removeBatchByIds(Arrays.asList(split));
+        taskCleanerService.removeBatchByIds(Arrays.asList(split));
         return R.ok();
     }
 
+    @Log(title = "任务记录导出", businessType = BusinessType.OTHER)
+    @ApiOperation(value = "任务记录导出")
+    @PostMapping(value = "/exportExcel")
+    public void exportExcel(@RequestBody TaskListQuery query) {
+        List<SysUser> sysUsers = sysUserService.selectAllList();
+        List<TLocation> locationList = locationService.list();
+        if (StringUtils.hasLength(query.getDeptName())){
+            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);
+                }
+            }
+            List<String> deptIds = deptService.lambdaQuery().like(TDept::getDeptName, query.getDeptName()).list()
+                    .stream().map(TDept::getId).collect(Collectors.toList());
+            listIds.addAll(deptIds);
+            if (listIds.isEmpty()){
+                listIds.add("0");
+            }
+            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().contains(query.getPhonenumber())
+                                && listIds.contains(sysUser.getDeptId())
+                ).map(SysUser::getUserId).collect(Collectors.toList());
+                if (patrolInspectorIds.isEmpty()){
+                    patrolInspectorIds.add(0L);
+                }
+                query.setPatrolInspectorIds(patrolInspectorIds);
+            }
+        }
+        if (StringUtils.hasLength(query.getPhonenumber())){
+            List<Long> patrolInspectorIds = sysUsers.stream().filter(sysUser ->
+                    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());
+            }
+            query.setPatrolInspectorIds(patrolInspectorIds);
+            if (patrolInspectorIds.isEmpty()){
+                patrolInspectorIds.add(0L);
+            }
+        }
+        if (query.getClearStatus()!=null){
+            List<String> collect = tTaskDetailService.lambdaQuery().eq(TTaskDetail::getClearStatus, query.getClearStatus())
+                    .list().stream().distinct().map(TTaskDetail::getTaskId).collect(Collectors.toList());
+            query.setTaskIds(collect);
+            if (collect.isEmpty()){
+                collect.add("0");
+            }
+        }
+        if (StringUtils.hasLength(query.getLocationType())){
+            List<String> collect = locationList.stream().filter(e -> e.getLocationType().equals(query.getLocationType())).map(TLocation::getId)
+                    .collect(Collectors.toList());
+            if (collect.isEmpty()){
+                collect.add("0");
+            }
+            query.setLocationIds(collect);
+        }
+        if (StringUtils.hasLength(query.getLocationName())){
+            List<String> collect = locationList.stream().filter(e -> e.getLocationName().contains(query.getLocationName())).map(TLocation::getId)
+                    .collect(Collectors.toList());
+            query.setLocationIds(collect);
+            if (collect.isEmpty()){
+                collect.add("0");
+            }else{
+                collect = collect.stream().filter(query.getLocationIds()::contains).collect(Collectors.toList());
+                if (collect.isEmpty()){
+                    collect.add("0");
+                }
+                query.setLocationIds(collect);
+            }
+        }
+        List<TaskListVO> list = taskCleanerService.exportList(query);
+
+        //1.获取excel模板
+        Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), TaskListVO.class, list);
+        HttpServletResponse response = WebUtils.response();
+        response.setContentType("application/vnd.ms-excel");
+        response.setCharacterEncoding("utf-8");
+        ServletOutputStream outputStream = null;
+        try {
+            String fileName = URLEncoder.encode("任务记录列表.xls", "utf-8");
+            response.setHeader("Content-dispodition", "attachment;filename=" + fileName);
+            outputStream = response.getOutputStream();
+            workbook.write(outputStream);
+        } catch (IOException e) {
+            e.printStackTrace();
+        } finally {
+            try {
+                outputStream.close();
+            } catch (IOException e) {
+                e.printStackTrace();
+            }
+        }
+
+    }
+
 }
 

--
Gitblit v1.7.1