From 53d014edd19f57125c355abe71ec5b478500e610 Mon Sep 17 00:00:00 2001
From: 无关风月 <443237572@qq.com>
Date: 星期二, 05 八月 2025 09:57:40 +0800
Subject: [PATCH] 保洁巡检所有代码
---
ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TCleanerController.java | 115 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 115 insertions(+), 0 deletions(-)
diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TCleanerController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TCleanerController.java
index aa377d1..080b109 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TCleanerController.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TCleanerController.java
@@ -1,21 +1,49 @@
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>
@@ -28,9 +56,12 @@
@Api(tags = "保洁员")
@RestController
@RequestMapping("/t-cleaner")
+@Slf4j
public class TCleanerController {
@Resource
private TCleanerService cleanerService;
+ @Resource
+ private TProjectDeptService projectDeptService;
@ApiOperation(value = "保洁员分页列表")
@@ -65,5 +96,89 @@
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();
+ }
+ }
+ }
}
--
Gitblit v1.7.1