package com.ruoyi.web.controller.api;
|
|
|
import cn.afterturn.easypoi.excel.ExcelExportUtil;
|
import cn.afterturn.easypoi.excel.ExcelImportUtil;
|
import cn.afterturn.easypoi.excel.entity.ExportParams;
|
import cn.afterturn.easypoi.excel.entity.ImportParams;
|
import cn.hutool.json.JSONObject;
|
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.core.exception.ServiceException;
|
import com.ruoyi.common.enums.BusinessType;
|
import com.ruoyi.common.utils.WebUtils;
|
import com.ruoyi.system.dto.CleanerDTO;
|
import com.ruoyi.system.importExcel.TCleanerImportExcel;
|
import com.ruoyi.system.importExcel.TLocationImportExcel;
|
import com.ruoyi.system.model.TCleaner;
|
import com.ruoyi.system.model.TLocation;
|
import com.ruoyi.system.model.TLocationType;
|
import com.ruoyi.system.model.TProjectDept;
|
import com.ruoyi.system.query.CleanerListQuery;
|
import com.ruoyi.system.service.TCleanerService;
|
import com.ruoyi.system.service.TProjectDeptService;
|
import com.ruoyi.system.vo.system.CleanerListVO;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.poi.ss.usermodel.Workbook;
|
import org.springframework.transaction.annotation.Propagation;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.util.CollectionUtils;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.annotation.Resource;
|
import javax.servlet.ServletOutputStream;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.IOException;
|
import java.io.InputStream;
|
import java.net.URLEncoder;
|
import java.util.ArrayList;
|
import java.util.Arrays;
|
import java.util.List;
|
import java.util.Objects;
|
|
/**
|
* <p>
|
* 保洁员 前端控制器
|
* </p>
|
*
|
* @author xiaochen
|
* @since 2025-05-28
|
*/
|
@Api(tags = "保洁员")
|
@RestController
|
@RequestMapping("/t-cleaner")
|
@Slf4j
|
public class TCleanerController {
|
@Resource
|
private TCleanerService cleanerService;
|
@Resource
|
private TProjectDeptService projectDeptService;
|
|
|
@ApiOperation(value = "保洁员分页列表")
|
@PostMapping(value = "/pageList")
|
public R<PageInfo<CleanerListVO>> pageList(@RequestBody CleanerListQuery query) {
|
return R.ok(cleanerService.pageList(query));
|
}
|
@Log(title = "新增保洁员", businessType = BusinessType.INSERT)
|
@ApiOperation(value = "新增保洁员")
|
@PostMapping(value = "/add")
|
public R<Boolean> add(@RequestBody CleanerDTO dto) {
|
cleanerService.save(dto);
|
return R.ok();
|
}
|
@Log(title = "编辑保洁员", businessType = BusinessType.UPDATE)
|
@ApiOperation(value = "编辑保洁员")
|
@PostMapping(value = "/edit")
|
public R<Boolean> edit(@RequestBody CleanerDTO dto) {
|
cleanerService.updateById(dto);
|
return R.ok();
|
}
|
@ApiOperation(value = "详情保洁员")
|
@GetMapping(value = "/detail")
|
public R<TCleaner> detail(@RequestParam String id) {
|
return R.ok(cleanerService.getById(id));
|
}
|
@Log(title = "批量删除保洁员", businessType = BusinessType.DELETE)
|
@ApiOperation(value = "批量删除保洁员")
|
@DeleteMapping(value = "/delete")
|
public R<Boolean> edit(@RequestParam String ids) {
|
String[] split = ids.split(",");
|
cleanerService.removeBatchByIds(Arrays.asList(split));
|
return R.ok();
|
}
|
@Log(title = "保洁员导入", businessType = BusinessType.INSERT)
|
@ApiOperation(value = "保洁员导入")
|
@Transactional(rollbackFor = Exception.class,propagation = Propagation.REQUIRES_NEW)
|
@PostMapping("/importCleaner")
|
public R<String> importCleaner(@RequestPart("file") MultipartFile file) {
|
ImportParams params = new ImportParams();
|
// params.setTitleRows(1); // 标题行数
|
params.setHeadRows(1); //表头行数
|
InputStream inputStream = null;
|
// List<CustomerImportFailedData> failedData = new ArrayList<>();
|
List<TCleanerImportExcel> locationExcelList;
|
try {
|
inputStream = file.getInputStream();
|
locationExcelList = ExcelImportUtil.importExcel(inputStream,
|
TCleanerImportExcel.class, params);
|
} catch (Exception e) {
|
e.printStackTrace();
|
log.error("保洁员导入失败:{}", e.getMessage());
|
throw new ServiceException("保洁员导入失败!");
|
} finally {
|
try {
|
inputStream.close();
|
} catch (IOException e) {
|
throw new ServiceException(e.getMessage());
|
}
|
}
|
if (CollectionUtils.isEmpty(locationExcelList)) {
|
throw new ServiceException("保洁员数据为空!");
|
}
|
JSONObject result = new JSONObject();
|
List<TProjectDept> projects = projectDeptService.list();
|
for (TCleanerImportExcel locationExcel : locationExcelList) {
|
System.err.println(locationExcel);
|
TCleaner tCleaner = new TCleaner();
|
TProjectDept projectDept = projects.stream().filter(e -> e.getProjectName().equals(locationExcel.getProjectName())).findFirst().orElse(null);
|
if (projectDept==null){
|
result.append("保洁员：[", locationExcel.getCleanerName()+"]未查询到部门");
|
continue;
|
}else{
|
if (projectDept.getParentId().equals("0")){
|
result.append("保洁员：[", locationExcel.getCleanerName()+"]所属片区不能指定为项目部");
|
continue;
|
}
|
}
|
tCleaner.setCleanerCode(locationExcel.getCleanerCode());
|
tCleaner.setCleanerName(locationExcel.getCleanerName());
|
tCleaner.setProjectId(projectDept.getId());
|
tCleaner.setDeptCode(locationExcel.getDeptCode());
|
tCleaner.setProjectCode(locationExcel.getProjectCode());
|
cleanerService.save(tCleaner);
|
}
|
if(!result.isEmpty()){
|
return R.ok(result.toString());
|
}
|
return R.ok();
|
}
|
@ApiOperation(value = "保洁员导入模板下载")
|
@GetMapping("/import-cleaner")
|
public void importCleaner() {
|
List<TCleanerImportExcel> locationImportExcels = new ArrayList<>();
|
TCleanerImportExcel tLocationImportExcel = new TCleanerImportExcel();
|
locationImportExcels.add(tLocationImportExcel);
|
Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), TCleanerImportExcel.class, locationImportExcels);
|
HttpServletResponse response = WebUtils.response();
|
ServletOutputStream outputStream = null;
|
try {
|
String fileName = URLEncoder.encode("保洁员导入模板.xls", "utf-8");
|
response.setHeader("Content-Disposition", "attachment;filename=" + fileName);
|
response.setHeader("content-Type", "application/vnd.ms-excel");
|
response.setHeader("Pragma", "no-cache");
|
response.setHeader("Cache-Control", "no-cache");
|
outputStream = response.getOutputStream();
|
workbook.write(outputStream);
|
} catch (IOException e) {
|
e.printStackTrace();
|
System.out.println("保洁员导入模板下载失败!");
|
} finally {
|
try {
|
outputStream.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
}
|
