From 5989a593451ce3a64d3e1a66d6472c1c38175541 Mon Sep 17 00:00:00 2001
From: xuhy <3313886187@qq.com>
Date: 星期四, 03 七月 2025 15:38:17 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'

---
 ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TInspectorController.java |  107 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 107 insertions(+), 0 deletions(-)

diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TInspectorController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TInspectorController.java
index 8b06389..848afcd 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TInspectorController.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TInspectorController.java
@@ -1,25 +1,35 @@
 package com.ruoyi.web.controller.api;
 
 
+import cn.afterturn.easypoi.excel.ExcelExportUtil;
+import cn.afterturn.easypoi.excel.entity.ExportParams;
 import com.ruoyi.common.annotation.Log;
 import com.ruoyi.common.basic.PageInfo;
 import com.ruoyi.common.core.domain.R;
 import com.ruoyi.common.core.domain.entity.SysUser;
 import com.ruoyi.common.enums.BusinessType;
+import com.ruoyi.common.utils.WebUtils;
 import com.ruoyi.framework.web.service.TokenService;
 import com.ruoyi.system.model.TDept;
 import com.ruoyi.system.model.TLocation;
 import com.ruoyi.system.model.TProjectDept;
 import com.ruoyi.system.model.TTaskDetail;
 import com.ruoyi.system.query.InsepectorListQuery;
+import com.ruoyi.system.query.TaskListQuery;
 import com.ruoyi.system.service.*;
 import com.ruoyi.system.vo.system.InspectorListVO;
+import com.ruoyi.system.vo.system.TaskListVO;
 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.web.bind.annotation.*;
 
 import javax.annotation.Resource;
+import javax.servlet.ServletOutputStream;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.net.URLEncoder;
 import java.util.Arrays;
 import java.util.List;
 import java.util.stream.Collectors;
@@ -121,6 +131,103 @@
         }
         return R.ok(inspectorService.pageList(query));
     }
+
+    @Log(title = "督察记录导出", businessType = BusinessType.EXPORT)
+    @ApiOperation(value = "督察记录导出")
+    @PostMapping(value = "/exportExcel")
+    public void exportExcel(@RequestBody InsepectorListQuery query) {
+        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()
+                    .stream().map(TProjectDept::getId).collect(Collectors.toList());
+            List<String> deptIds = deptService.lambdaQuery().like(TDept::getDeptName, query.getDeptName()).list()
+                    .stream().map(TDept::getId).collect(Collectors.toList());
+            projectIds.addAll(deptIds);
+            if (projectIds.isEmpty()){
+                projectIds.add("0")    ;
+            }
+            if (StringUtils.hasLength(query.getPhonenumber())){
+                List<Long> patrolInspectorIds = sysUsers.stream().filter(sysUser ->
+                        sysUser.getPhonenumber().equals(query.getPhonenumber())
+                                && projectIds.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().equals(query.getPhonenumber())
+            ).map(SysUser::getUserId).collect(Collectors.toList());
+            if (!query.getPatrolInspectorIds().isEmpty()){
+                // 取交集
+                patrolInspectorIds = patrolInspectorIds.stream().filter(query.getPatrolInspectorIds()::contains).collect(Collectors.toList());
+            }
+            if (patrolInspectorIds.isEmpty()){
+                patrolInspectorIds.add(0L);
+            }
+            query.setPatrolInspectorIds(patrolInspectorIds);
+
+        }
+        if (query.getClearStatus()!=null){
+            List<String> collect = tTaskDetailService.lambdaQuery().eq(TTaskDetail::getClearStatus, query.getClearStatus())
+                    .list().stream().distinct().map(TTaskDetail::getTaskId).collect(Collectors.toList());
+            if (collect.isEmpty()){
+                collect.add("-1");
+            }
+            query.setTaskIds(collect);
+
+        }
+        if (query.getLocationType()!=null){
+            List<String> collect = locationList.stream().filter(e -> e.getLocationType().equals(query.getLocationType())).map(TLocation::getId)
+                    .collect(Collectors.toList());
+            if (collect.isEmpty()){
+                collect.add("-1");
+
+            }
+            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());
+            if (collect.isEmpty()){
+                collect.add("-1");
+
+                query.setLocationIds(collect);
+            }else{
+                collect = collect.stream().filter(query.getLocationIds()::contains).collect(Collectors.toList());
+                if (collect.isEmpty()){
+                    collect.add("-1");
+                }
+                query.setLocationIds(collect);
+            }
+        }
+        List<InspectorListVO> inspectorListVOS = inspectorService.pageListExport(query);
+        //1.获取excel模板
+        Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), InspectorListVO.class, inspectorListVOS);
+        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();
+            }
+        }
+
+    }
     @Log(title = "批量删除督察任务", businessType = BusinessType.DELETE)
     @ApiOperation(value = "批量删除督察任务")
     @DeleteMapping(value = "/deleteByIds")

--
Gitblit v1.7.1