From 12aac2ade617f3bb4e40165176d009ab3fa15a52 Mon Sep 17 00:00:00 2001
From: xuhy <3313886187@qq.com>
Date: 星期五, 20 六月 2025 17:52:13 +0800
Subject: [PATCH] 点位导入
---
ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TLocationController.java | 98 ++++++++++++++++++++++++++++++++++++++++++++----
1 files changed, 89 insertions(+), 9 deletions(-)
diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TLocationController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TLocationController.java
index 6ea909c..fdb8ee8 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TLocationController.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TLocationController.java
@@ -1,32 +1,46 @@
package com.ruoyi.web.controller.api;
-import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+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 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.common.utils.bean.BeanUtils;
import com.ruoyi.system.dto.LocationDTO;
-import com.ruoyi.system.dto.LocationTypeDTO;
+import com.ruoyi.system.importExcel.TLocationImportExcel;
import com.ruoyi.system.model.*;
import com.ruoyi.system.query.LocationListQuery;
import com.ruoyi.system.query.LocationListTaskQuery;
-import com.ruoyi.system.query.LocationTypeListQuery;
import com.ruoyi.system.service.*;
+import com.ruoyi.system.utils.CustomerImportFailedData;
import com.ruoyi.system.vo.system.LocationListTaskVO;
import com.ruoyi.system.vo.system.LocationListVO;
import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiImplicitParam;
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.util.StringUtils;
import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
import javax.annotation.Resource;
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.List;
-import java.util.Objects;
+import javax.servlet.ServletOutputStream;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.URLEncoder;
+import java.util.*;
import java.util.stream.Collectors;
/**
@@ -37,6 +51,7 @@
* @author xiaochen
* @since 2025-05-28
*/
+@Slf4j
@Api(tags = "点位管理")
@RestController
@RequestMapping("/t-location")
@@ -50,7 +65,7 @@
@Resource
private TCleanerService cleanerService;
@Resource
- private TTaskService tTaskService;
+ private TTaskCleanService tTaskCleanService;
@Resource
private TTaskDetailService tTaskDetailService;
@Resource
@@ -109,7 +124,7 @@
}
}
PageInfo<LocationListTaskVO> pageInfo = new PageInfo<>(query.getPageNum(), query.getPageSize());
- List<TTask> list = tTaskService.lambdaQuery()
+ List<TTask> list = tTaskCleanService.lambdaQuery()
.eq(TTask::getLocationId, query.getId())
.between(StringUtils.hasLength(query.getStartTime()), TTask::getImplementTime, query.getStartTime(), query.getEndTime())
.eq(query.getStatus() != null, TTask::getStatus, query.getStatus())
@@ -194,5 +209,70 @@
locationService.removeBatchByIds(Arrays.asList(split));
return R.ok();
}
+
+ @ApiOperation(value = "点位导入模板下载")
+ @GetMapping("/import-template")
+ public void importTemplate() {
+ List<TLocationImportExcel> locationImportExcels = new ArrayList<>();
+ Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), TLocationImportExcel.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();
+ }
+ }
+ }
+
+ @Log(title = "点位导入", businessType = BusinessType.INSERT)
+ @ApiOperation(value = "点位导入")
+ @Transactional(rollbackFor = Exception.class,propagation = Propagation.REQUIRES_NEW)
+ @PostMapping("/importLocation")
+ public R<List<CustomerImportFailedData>> importLocation(@RequestPart("file") MultipartFile file) {
+ ImportParams params = new ImportParams();
+ params.setTitleRows(2); // 标题行数
+ params.setHeadRows(1); //表头行数
+ InputStream inputStream = null;
+ List<CustomerImportFailedData> failedData = new ArrayList<>();
+ List<TLocationImportExcel> locationExcelList;
+ try {
+ inputStream = file.getInputStream();
+ locationExcelList = ExcelImportUtil.importExcel(inputStream,
+ TLocationImportExcel.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("点位数据为空!");
+ }
+ for (TLocationImportExcel locationExcel : locationExcelList) {
+ System.err.println(locationExcel);
+ }
+// locationService.saveOrUpdateBatch(saveOrUpdateList);
+
+ return R.ok(failedData);
+ }
+
}
--
Gitblit v1.7.1