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