From bcd9cb83cbbb36217cb99ced7bf1fa3cb26bd53c Mon Sep 17 00:00:00 2001
From: zjj <3272660260@qq.com>
Date: 星期一, 28 四月 2025 16:59:51 +0800
Subject: [PATCH] #wms端生成任务优化

---
 rsf-server/src/main/java/com/vincent/rsf/server/manager/controller/AsnOrderItemController.java |   62 ++++++++++++++++++++++++++-----
 1 files changed, 52 insertions(+), 10 deletions(-)

diff --git a/rsf-server/src/main/java/com/vincent/rsf/server/manager/controller/AsnOrderItemController.java b/rsf-server/src/main/java/com/vincent/rsf/server/manager/controller/AsnOrderItemController.java
index 650a15d..014c228 100644
--- a/rsf-server/src/main/java/com/vincent/rsf/server/manager/controller/AsnOrderItemController.java
+++ b/rsf-server/src/main/java/com/vincent/rsf/server/manager/controller/AsnOrderItemController.java
@@ -1,6 +1,8 @@
 package com.vincent.rsf.server.manager.controller;
 
 
+import com.alibaba.fastjson.JSONArray;
+import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.vincent.rsf.framework.common.Cools;
@@ -11,13 +13,18 @@
 import com.vincent.rsf.server.common.domain.BaseParam;
 import com.vincent.rsf.server.common.domain.KeyValVo;
 import com.vincent.rsf.server.common.domain.PageParam;
+import com.vincent.rsf.server.manager.controller.params.BatchUpdateParam;
 import com.vincent.rsf.server.manager.entity.AsnOrder;
 import com.vincent.rsf.server.manager.entity.AsnOrderItem;
+import com.vincent.rsf.server.manager.entity.Companys;
 import com.vincent.rsf.server.manager.entity.excel.AsnOrderTemplate;
+import com.vincent.rsf.server.manager.enums.CompanysType;
 import com.vincent.rsf.server.manager.service.AsnOrderItemService;
+import com.vincent.rsf.server.manager.service.CompanysService;
 import com.vincent.rsf.server.system.controller.BaseController;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
+import org.apache.commons.lang3.StringUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.security.access.prepost.PreAuthorize;
 import org.springframework.web.bind.annotation.*;
@@ -32,6 +39,9 @@
 
     @Autowired
     private AsnOrderItemService asnOrderItemService;
+
+    @Autowired
+    private CompanysService companysService;
 
     @PreAuthorize("hasAuthority('manager:asnOrderItem:list')")
     @ApiOperation("鍒嗛〉鑾峰彇鍒楄〃")
@@ -74,8 +84,8 @@
             return R.error("Save Fail");
         }
         return R.ok("Save Success");
-
     }
+
 
     @PreAuthorize("hasAuthority('manager:asnOrderItem:update')")
     @OperationLog("Update ASN鍗曟嵁")
@@ -83,6 +93,14 @@
     public R update(@RequestBody AsnOrderItem asnOrderItem) {
         asnOrderItem.setUpdateBy(getLoginUserId());
         asnOrderItem.setUpdateTime(new Date());
+        if (!Objects.isNull(asnOrderItem.getSplrName()) && StringUtils.isNotBlank(asnOrderItem.getSplrName())) {
+            Companys companys = companysService.getOne(new LambdaQueryWrapper<Companys>()
+                            .eq(Companys::getType, CompanysType.COMPANYS_TYPE_SUPPLIER.val)
+                    .eq(Companys::getId, asnOrderItem.getSplrName()));
+            if (!Objects.isNull(companys)) {
+                asnOrderItem.setSplrCode(companys.getCode()).setSplrName(companys.getName());
+            }
+        }
         if (!asnOrderItemService.updateById(asnOrderItem)) {
             return R.error("Update Fail");
         }
@@ -117,10 +135,21 @@
     @PostMapping("/asnOrderItem/export")
     @ApiOperation("瀵煎嚭鏀惰揣閫氱煡鍗曟槑缁�")
     public void export(@RequestBody Map<String, Object> map, HttpServletResponse response) throws Exception {
-        AsnOrderTemplate template = ExcelUtil.mockData(AsnOrderTemplate.class);
-        List<AsnOrderTemplate> list = new ArrayList<>();
-        list.add(template);
-        ExcelUtil.build(ExcelUtil.create(list, AsnOrderTemplate.class, true), response);
+        List<AsnOrderItem> orderItems = new ArrayList<>();
+        if (!Objects.isNull(map.get("ids"))) {
+            List<Long> ids = JSONArray.parseArray(JSONObject.toJSONString(map.get("ids")), Long.class);
+            if (!ids.isEmpty()) {
+                orderItems = asnOrderItemService.list(new LambdaQueryWrapper<AsnOrderItem>()
+                        .in(AsnOrderItem::getId, ids)
+                        .eq(AsnOrderItem::getStatus, 1));
+            } else {
+                orderItems = asnOrderItemService.list(new LambdaQueryWrapper<>());
+            }
+        } else {
+            orderItems = asnOrderItemService.list(new LambdaQueryWrapper<>());
+        }
+
+        ExcelUtil.build(ExcelUtil.create(orderItems, AsnOrderItem.class, true), response);
     }
 
 
@@ -144,13 +173,26 @@
     @PreAuthorize("hasAuthority('manager:asnOrderItem:update')")
     public R importExcel(@RequestParam(value = "file") MultipartFile file, @RequestParam String asnId) throws Exception {
         if (Objects.isNull(file)) {
-            throw new CoolException("鏂囦欢涓嶈兘涓虹┖锛侊紒");
+            R.error("鏂囦欢涓嶈兘涓虹┖锛侊紒");
         }
         HashMap<String, Object> hashMap = new HashMap<>();
-        if (!Objects.isNull(asnId)) {
-            hashMap.put("asnId", asnId);
-        }
-        return asnOrderItemService.excelImport(file, hashMap);
+        return asnOrderItemService.excelImport(file, hashMap, getLoginUserId());
+    }
+
+    /**
+     * @author Ryan
+     * @description 涓嬭浇妯℃澘
+     * @param
+     * @return
+     * @time 2025/4/18 08:17
+     */
+    @PostMapping("/asnOrderItem/template/download")
+    @ApiOperation("涓嬭浇鏀惰揣鍗曟ā鏉�")
+    @PreAuthorize("hasAuthority('manager:asnOrderItem:update')")
+    public void downloadTemplate(@RequestBody Map<String, Object> map, HttpServletResponse response) throws Exception {
+        AsnOrderTemplate template = ExcelUtil.mockData(AsnOrderTemplate.class);
+        List<AsnOrderTemplate> list = Arrays.asList(template);
+        ExcelUtil.build(ExcelUtil.create(list, AsnOrderTemplate.class, true), response);
     }
 
 }

--
Gitblit v1.9.1