From 4a129f05fb7369b7916ccd8e68e727bb182439f1 Mon Sep 17 00:00:00 2001
From: Junjie <540245094@qq.com>
Date: 星期一, 12 八月 2024 11:21:46 +0800
Subject: [PATCH] #

---
 zy-asrs-wms/src/main/java/com/zy/asrs/wms/asrs/controller/OrderController.java |   68 +++++++++++++++++++++++++++++++++-
 1 files changed, 66 insertions(+), 2 deletions(-)

diff --git a/zy-asrs-wms/src/main/java/com/zy/asrs/wms/asrs/controller/OrderController.java b/zy-asrs-wms/src/main/java/com/zy/asrs/wms/asrs/controller/OrderController.java
index 7a1dbf9..326c0ef 100644
--- a/zy-asrs-wms/src/main/java/com/zy/asrs/wms/asrs/controller/OrderController.java
+++ b/zy-asrs-wms/src/main/java/com/zy/asrs/wms/asrs/controller/OrderController.java
@@ -5,9 +5,13 @@
 import com.zy.asrs.framework.common.Cools;
 import com.zy.asrs.framework.common.R;
 import com.zy.asrs.framework.exception.CoolException;
+import com.zy.asrs.wms.asrs.entity.MatField;
 import com.zy.asrs.wms.asrs.entity.OrderType;
+import com.zy.asrs.wms.asrs.entity.enums.OrderSettleType;
 import com.zy.asrs.wms.asrs.entity.param.CreateOrderParam;
 import com.zy.asrs.wms.asrs.entity.param.UpdateOrderParam;
+import com.zy.asrs.wms.asrs.entity.template.OrderTemplate;
+import com.zy.asrs.wms.asrs.service.MatFieldService;
 import com.zy.asrs.wms.asrs.service.OrderTypeService;
 import com.zy.asrs.wms.common.annotation.OperationLog;
 import com.zy.asrs.wms.common.domain.BaseParam;
@@ -21,6 +25,7 @@
 import org.springframework.security.access.prepost.PreAuthorize;
 import org.springframework.transaction.annotation.Transactional;
 import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
 
 import javax.servlet.http.HttpServletResponse;
 import java.util.*;
@@ -33,6 +38,8 @@
     private OrderService orderService;
     @Autowired
     private OrderTypeService orderTypeService;
+    @Autowired
+    private MatFieldService matFieldService;
 
     @PreAuthorize("hasAuthority('asrs:order:list')")
     @PostMapping("/order/page")
@@ -170,8 +177,65 @@
     @PreAuthorize("hasAuthority('asrs:order:list')")
     @PostMapping("/order/exportTemplate")
     public void exportTemplate(@RequestBody Map<String, Object> map, HttpServletResponse response) throws Exception {
-        ArrayList<Order> list = new ArrayList<>();
-        ExcelUtil.build(ExcelUtil.create(list, Order.class), response);
+        ArrayList<OrderTemplate> list = new ArrayList<>();
+        List<MatField> locFields = matFieldService.getLocFields();
+        ExcelUtil.build(ExcelUtil.create(list, OrderTemplate.class, locFields), response);
+    }
+
+    @PostMapping("/order/upload")
+    public R upload(@RequestParam("file") MultipartFile file) {
+        List<OrderTemplate> list = ExcelUtil.parseExcelFile(file, OrderTemplate.class);
+
+        HashMap<String, Object> map = new HashMap<>();
+        for (OrderTemplate orderTemplate : list) {
+            if (!map.containsKey(orderTemplate.getOrderNo())) {
+                OrderType orderType = orderTypeService.getOne(new LambdaQueryWrapper<OrderType>().eq(OrderType::getName, orderTemplate.getOrderType()));
+                if (orderType == null) {
+                    throw new CoolException("璁㈠崟绫诲瀷涓嶅瓨鍦�");
+                }
+
+                HashMap<String, Object> paramMap = new HashMap<>();
+                ArrayList<HashMap<String, Object>> paramList = new ArrayList<>();
+                paramList.add(paramMap);
+
+                paramMap.put("matnr", orderTemplate.getMatnr());
+                paramMap.put("batch", orderTemplate.getBatch());
+                paramMap.put("anfme", orderTemplate.getAnfme());
+                paramMap.put("memo", orderTemplate.getMemo());
+                paramMap.putAll(orderTemplate.getDynamicFields());
+
+                CreateOrderParam orderParam = new CreateOrderParam();
+                orderParam.setOrderNo(orderTemplate.getOrderNo());
+                orderParam.setOrderType(orderType.getId());
+                orderParam.setOrderSettle(OrderSettleType.INIT.val());
+                orderParam.setList(paramList);
+
+                map.put(orderTemplate.getOrderNo(), orderParam);
+            }else {
+                CreateOrderParam orderParam = (CreateOrderParam) map.get(orderTemplate.getOrderNo());
+
+                HashMap<String, Object> paramMap = new HashMap<>();
+                paramMap.put("matnr", orderTemplate.getMatnr());
+                paramMap.put("batch", orderTemplate.getBatch());
+                paramMap.put("anfme", orderTemplate.getAnfme());
+                paramMap.put("memo", orderTemplate.getMemo());
+                paramMap.putAll(orderTemplate.getDynamicFields());
+
+                List<HashMap<String, Object>> paramList = orderParam.getList();
+                paramList.add(paramMap);
+
+                orderParam.setList(paramList);
+                map.put(orderTemplate.getOrderNo(), orderParam);
+            }
+        }
+
+        ArrayList<CreateOrderParam> orderParams = new ArrayList<>();
+        for (Map.Entry<String, Object> entry : map.entrySet()) {
+            CreateOrderParam orderParam = (CreateOrderParam) entry.getValue();
+            orderParams.add(orderParam);
+        }
+        orderService.createOrder(orderParams);
+        return R.ok();
     }
 
 }

--
Gitblit v1.9.1