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;
|
|
/**
|
* <p>
|
* 督察记录 前端控制器
|
* </p>
|
*
|
* @author xiaochen
|
* @since 2025-05-28
|
*/
|
@Api(tags = "督察记录")
|
@RestController
|
@RequestMapping("/t-inspector")
|
public class TInspectorController {
|
|
@Resource
|
private TLocationService locationService;
|
@Resource
|
private TProjectDeptService projectDeptService;
|
|
@Resource
|
private TTaskDetailService tTaskDetailService;
|
@Resource
|
private TDeptService deptService;
|
@Resource
|
private ISysUserService sysUserService;
|
|
@Resource
|
private TInspectorService inspectorService;
|
@ApiOperation(value = "督察记录分页列表")
|
@PostMapping(value = "/pageList")
|
public R<PageInfo<InspectorListVO>> pageList(@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()){
|
return R.ok(new PageInfo<>());
|
}
|
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()){
|
return R.ok(new PageInfo<>());
|
}
|
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());
|
}
|
query.setPatrolInspectorIds(patrolInspectorIds);
|
if (patrolInspectorIds.isEmpty()){
|
return R.ok(new PageInfo<>());
|
}
|
}
|
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()){
|
return R.ok(new PageInfo<>());
|
}
|
}
|
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()){
|
return R.ok(new PageInfo<>());
|
}
|
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()){
|
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<>());
|
}
|
query.setLocationIds(collect);
|
}
|
}
|
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")
|
public R<Boolean> deleteByIds(@RequestParam String ids) {
|
String[] split = ids.split(",");
|
inspectorService.removeBatchByIds(Arrays.asList(split));
|
return R.ok();
|
}
|
@Log(title = "删除督察任务", businessType = BusinessType.DELETE)
|
@ApiOperation(value = "删除督察任务")
|
@DeleteMapping(value = "/deleteById")
|
public R<Boolean> deleteById(@RequestParam String id) {
|
inspectorService.removeById(id);
|
return R.ok();
|
}
|
}
|
