From 01d08428dd6cd9dc518a9a2ecbcb1f0e82788068 Mon Sep 17 00:00:00 2001
From: Fixiaobai <fixiaobai@163.com>
Date: 星期四, 09 十一月 2023 13:54:47 +0800
Subject: [PATCH] Changes11

---
 mes-technology/src/main/java/com/chinaztt/mes/technology/controller/StructureController.java |   72 +++++++++++++++---------------------
 1 files changed, 30 insertions(+), 42 deletions(-)

diff --git a/mes-technology/src/main/java/com/chinaztt/mes/technology/controller/StructureController.java b/mes-technology/src/main/java/com/chinaztt/mes/technology/controller/StructureController.java
index 67f390b..cace20d 100644
--- a/mes-technology/src/main/java/com/chinaztt/mes/technology/controller/StructureController.java
+++ b/mes-technology/src/main/java/com/chinaztt/mes/technology/controller/StructureController.java
@@ -32,9 +32,11 @@
 import com.chinaztt.ztt.admin.api.feign.RemoteDictService;
 import com.chinaztt.ztt.common.core.util.R;
 import com.chinaztt.ztt.common.log.annotation.SysLog;
+import com.chinaztt.ztt.common.security.util.SecurityUtils;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import lombok.AllArgsConstructor;
+import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.security.access.prepost.PreAuthorize;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
@@ -60,6 +62,8 @@
 	private final StructureService structureService;
 	private final DictUtils dictUtils;
 
+	private RedisTemplate redisTemplate;
+
 	/**
 	 * 妯潃瀵煎叆鐨勫鍏ユ柟寮�
 	 *
@@ -68,29 +72,13 @@
 	 */
 	@PostMapping("/uploadExt")
 	public R uploadExt(@RequestParam("file") MultipartFile file) {
-		//ExcelReader excelReader = null;
-		//try {
-		//	ExcelReaderBuilder excelReaderBuilder = EasyExcel.read(file.getInputStream());
-		//	excelReader = excelReaderBuilder.build();
-		//	List<ReadSheet> sheets = excelReader.excelExecutor().sheetList();
-		//	List<ReadSheet> readSheetList = new ArrayList<>();
-		//	for (ReadSheet sheet : sheets) {
-		//		ReadSheet readSheet = EasyExcel.readSheet(sheet.getSheetName()).registerReadListener(new StructureUploadExtListener(structureService)).build();
-		//		readSheetList.add(readSheet);
-		//	}
-		//	excelReader.read(readSheetList);
-		//} catch (IOException e) {
-		//	e.printStackTrace();
-		//	return R.failed(e.getMessage());
-		//} finally {
-		//	if (excelReader != null) {
-		//		// 杩欓噷鍗冧竾鍒繕璁板叧闂紝璇荤殑鏃跺�欎細鍒涘缓涓存椂鏂囦欢锛屽埌鏃剁鐩樹細宕╃殑
-		//		excelReader.finish();
-		//	}
-		//}
+		String key = "syncStructure_lock" + SecurityUtils.getUser().getId();
+		if (redisTemplate.hasKey(key)) {
+			throw new RuntimeException("鏈夊鍏ヤ换鍔℃鍦ㄥ鐞�");
+		}
 		ExcelReader excelReader = null;
 		try {
-			excelReader = EasyExcel.read(file.getInputStream(), StructureData.class, new StructureUploadListener(structureService, dictUtils)).build();
+			excelReader = EasyExcel.read(file.getInputStream(), StructureData.class, new StructureUploadListener(structureService)).build();
 			ReadSheet readSheet = EasyExcel.readSheet(0).build();
 			ReadSheet readSheet2 = EasyExcel.readSheet(1).build();
 			excelReader.read(readSheet,readSheet2);
@@ -110,27 +98,27 @@
 	 * 绔栫潃瀵煎叆
 	 *
 	 * @param file
-	 * @return
-	 */
-	@PostMapping("/upload")
-	public R simpleRead(@RequestParam("file") MultipartFile file) {
-		ExcelReader excelReader = null;
-		try {
-			excelReader = EasyExcel.read(file.getInputStream(), StructureData.class, new StructureUploadListener(structureService, dictUtils)).build();
-			ReadSheet readSheet = EasyExcel.readSheet(0).build();
-			ReadSheet readSheet2 = EasyExcel.readSheet(1).build();
-			excelReader.read(readSheet,readSheet2);
-		} catch (IOException e) {
-			e.printStackTrace();
-			return R.failed(e.getMessage());
-		} finally {
-			if (excelReader != null) {
-				// 杩欓噷鍗冧竾鍒繕璁板叧闂紝璇荤殑鏃跺�欎細鍒涘缓涓存椂鏂囦欢锛屽埌鏃剁鐩樹細宕╃殑
-				excelReader.finish();
-			}
-		}
-		return R.ok();
-	}
+//	 * @return
+//	 */
+//	@PostMapping("/upload")
+//	public R simpleRead(@RequestParam("file") MultipartFile file) {
+//		ExcelReader excelReader = null;
+//		try {
+//			excelReader = EasyExcel.read(file.getInputStream(), StructureData.class, new StructureUploadListener(structureService)).build();
+//			ReadSheet readSheet = EasyExcel.readSheet(0).build();
+//			ReadSheet readSheet2 = EasyExcel.readSheet(1).build();
+//			excelReader.read(readSheet,readSheet2);
+//		} catch (IOException e) {
+//			e.printStackTrace();
+//			return R.failed(e.getMessage());
+//		} finally {
+//			if (excelReader != null) {
+//				// 杩欓噷鍗冧竾鍒繕璁板叧闂紝璇荤殑鏃跺�欎細鍒涘缓涓存椂鏂囦欢锛屽埌鏃剁鐩樹細宕╃殑
+//				excelReader.finish();
+//			}
+//		}
+//		return R.ok();
+//	}
 
 	/**
 	 * 鏍规嵁闆朵欢鍙锋煡璇㈤粯璁OM

--
Gitblit v1.9.3