From 7a268ed852ca462347cca383a6d0306ff54e1363 Mon Sep 17 00:00:00 2001
From: zss <zss@example.com>
Date: 星期一, 04 十一月 2024 15:05:21 +0800
Subject: [PATCH] 添加模版

---
 inspect-server/src/main/java/com/yuanchu/mom/controller/StandardTreeController.java |   25 +++++++++++++++++++++----
 1 files changed, 21 insertions(+), 4 deletions(-)

diff --git a/inspect-server/src/main/java/com/yuanchu/mom/controller/StandardTreeController.java b/inspect-server/src/main/java/com/yuanchu/mom/controller/StandardTreeController.java
index 648812a..7284046 100644
--- a/inspect-server/src/main/java/com/yuanchu/mom/controller/StandardTreeController.java
+++ b/inspect-server/src/main/java/com/yuanchu/mom/controller/StandardTreeController.java
@@ -11,11 +11,16 @@
 import com.yuanchu.mom.service.StandardProductListService;
 import com.yuanchu.mom.service.StandardTreeService;
 import com.yuanchu.mom.vo.Result;
-import io.swagger.annotations.*;
+import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiImplicitParam;
+import io.swagger.annotations.ApiImplicitParams;
+import io.swagger.annotations.ApiOperation;
 import lombok.AllArgsConstructor;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
 
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
 import java.util.Map;
 
 @RestController
@@ -37,6 +42,7 @@
         return Result.success(standardTreeService.selectStandardTreeList());
     }
 
+    @ValueAuth
     @ApiOperation(value = "鑾峰彇鏍囧噯鏍�(妫�楠屼笅鍗�)")
     @GetMapping("/selectStandardTreeList2")
     public Result selectStandardTreeList2() {
@@ -167,12 +173,23 @@
 
     @ValueClassify("鏍囧噯搴�")
     @ApiOperation("瀵煎叆鏍囧噯搴�")
-    @PostMapping("/inExcelOfTree")
-    public Result inExcelOfTree(@RequestParam("file") MultipartFile file) {
-        standardTreeService.inExcelOfTree(file);
+    @PostMapping("/inExcelOfTree/{isEquipment}")
+    public Result inExcelOfTree(@RequestParam("file") MultipartFile file, @PathVariable("isEquipment") Boolean isEquipment) {
+        if (!isEquipment) {
+            standardTreeService.inExcelOfTree(file);
+        } else {
+            standardTreeService.importWorkstationExcel(file);
+        }
         return Result.success();
     }
 
+    @ValueClassify("鏍囧噯搴�")
+    @ApiOperation("瀵煎嚭鏍囧噯搴�")
+    @GetMapping("/exportStandTree")
+    public void exportStandTree(HttpServletResponse response) throws IOException  {
+        standardTreeService.exportStandTree(response);
+    }
+
     @ValueAuth
     @ApiOperation("閲嶇疆鏍囧噯搴撳崟浠�")
     @PostMapping("/resetTreeOfPrice")

--
Gitblit v1.9.3