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