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

---
 rsf-server/src/main/java/com/vincent/rsf/server/manager/service/impl/TaskServiceImpl.java |  235 +++++++++++++++++++++++++++++++++++++++++++++++++++++++---
 1 files changed, 223 insertions(+), 12 deletions(-)

diff --git a/rsf-server/src/main/java/com/vincent/rsf/server/manager/service/impl/TaskServiceImpl.java b/rsf-server/src/main/java/com/vincent/rsf/server/manager/service/impl/TaskServiceImpl.java
index cd40c71..a6ea80f 100644
--- a/rsf-server/src/main/java/com/vincent/rsf/server/manager/service/impl/TaskServiceImpl.java
+++ b/rsf-server/src/main/java/com/vincent/rsf/server/manager/service/impl/TaskServiceImpl.java
@@ -1,25 +1,33 @@
 package com.vincent.rsf.server.manager.service.impl;
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
+import com.vincent.rsf.framework.common.Cools;
 import com.vincent.rsf.server.api.entity.enums.OrderType;
 import com.vincent.rsf.server.api.entity.enums.TaskStsType;
 import com.vincent.rsf.framework.common.R;
 import com.vincent.rsf.framework.exception.CoolException;
 import com.vincent.rsf.server.api.entity.enums.TaskType;
+import com.vincent.rsf.server.api.utils.LocUtils;
+import com.vincent.rsf.server.manager.controller.params.GenerateTaskParams;
 import com.vincent.rsf.server.manager.entity.*;
+import com.vincent.rsf.server.manager.enums.PakinIOStatus;
 import com.vincent.rsf.server.manager.mapper.TaskMapper;
 import com.vincent.rsf.server.manager.service.*;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.vincent.rsf.server.manager.utils.LocManageUtil;
 import com.vincent.rsf.server.system.constant.SerialRuleCode;
+import com.vincent.rsf.server.system.enums.LocStsType;
 import com.vincent.rsf.server.system.utils.SerialRuleUtils;
 import org.apache.commons.lang3.StringUtils;
 import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Map;
 import java.util.Objects;
 import java.util.stream.Collectors;
 
@@ -32,7 +40,34 @@
     private TaskItemService taskItemService;
     @Autowired
     private WaitPakinItemService waitPakinItemService;
+    @Autowired
+    private LocService locService;
+    @Autowired
+    private PurchaseService purchaseService;
+    @Autowired
+    private AsnOrderService asnOrderService;
+    @Autowired
+    private AsnOrderItemService asnOrderItemService;
+    @Autowired
+    private StockService stockService;
+    @Autowired
+    private StockItemService stockItemService;
+    @Autowired
+    private DeviceSiteService deviceSiteService;
 
+    @Autowired
+    private OrderService orderService;
+
+    @Autowired
+    private OrderItemService orderItemService;
+
+    @Autowired
+    private LocItemService locItemService;
+
+    @Autowired
+    private DeviceBindService deviceBindService;
+    @Autowired
+    private WarehouseAreasService warehouseAreasService;
 
 
     /**
@@ -44,15 +79,37 @@
      * @time 2025/3/29 15:59
      */
     @Override
-    public R generateTasks(List<WaitPakin> waitPakin, Long loginUserId) {
-        if (Objects.isNull(waitPakin) || waitPakin.isEmpty()) {
+    @Transactional(rollbackFor = Exception.class)
+    public synchronized R generateTasks(GenerateTaskParams waitPakin, Long loginUserId) {
+        if (Objects.isNull(waitPakin) || waitPakin.getWaitPakins().isEmpty()) {
             throw new CoolException("鍙傛暟涓嶈兘涓虹┖锛侊紒");
         }
+
+        DeviceSite deviceSite = deviceSiteService.getById(waitPakin.getSiteId());
+        if (Objects.isNull(deviceSite)) {
+            throw new CoolException("绔欑偣涓嶅瓨鍦紒锛�");
+        }
+        DeviceBind deviceBind = deviceBindService.getById(LocUtils.getAreaType(Integer.valueOf(deviceSite.getSite())));
+        if (Cools.isEmpty(deviceBind)) {
+            throw new CoolException("搴撲綅瑙勫垯鏈煡");
+        }
+        WarehouseAreas warehouseArea = warehouseAreasService.getById(deviceBind.getTypeId());
+        if (Cools.isEmpty(warehouseArea)) {
+            throw new CoolException("鏈壘鍒版墍灞炲簱鍖轰俊鎭�");
+        }
+
+        /**鑾峰彇搴撲綅*/
+        String targetLoc = LocManageUtil.getTargetLoc(warehouseArea.getId());
+        if (Cools.isEmpty(targetLoc)) {
+            throw new CoolException("璇ョ珯鐐瑰搴斿簱鍖烘湭鎵惧埌搴撲綅");
+        }
         /**鑾峰彇缁勬嫋*/
-        List<Long> ids = waitPakin.stream().map(WaitPakin::getId).collect(Collectors.toList());
-        List<WaitPakin> waitPakins = waitPakinService.list(new LambdaQueryWrapper<WaitPakin>().in(WaitPakin::getId, ids));
+        List<Long> ids = waitPakin.getWaitPakins().stream().map(WaitPakin::getId).collect(Collectors.toList());
+        List<WaitPakin> waitPakins = waitPakinService.list(new LambdaQueryWrapper<WaitPakin>()
+                .in(WaitPakin::getId, ids)
+                .eq(WaitPakin::getIoStatus, Short.parseShort(PakinIOStatus.PAKIN_IO_STATUS_DONE.val)));
         if (waitPakins.isEmpty()) {
-            throw new CoolException("缁勬嫋淇℃伅涓嶅瓨鍦紒锛�");
+            throw new CoolException("璇锋鏌ョ粍鎷栫姸鎬佹槸鍚﹀畬鎴愶紒锛�");
         }
         waitPakins.forEach(pakin -> {
             List<TaskItem> taskItems = new ArrayList<>();
@@ -64,14 +121,30 @@
             task.setTaskCode(ruleCode)
                     .setTaskStatus(TaskStsType.GENERATE_IN.id.shortValue())
                     .setTaskType(TaskType.TASK_TYPE_IN.type.shortValue())
-                    .setTargLoc(LocManageUtil.getTargetLoc())
+                    .setTargLoc(targetLoc)
                     .setBarcode(pakin.getBarcode())
+                    .setOrgSite(deviceSite.getSite())
+                    .setTargSite(deviceSite.getDeviceCode())
                     .setCreateBy(loginUserId)
-                    .setUpdateBy(loginUserId)
-                    .setTargSite(LocManageUtil.getTargetSite());
+                    .setUpdateBy(loginUserId);
+
+            if (!Objects.isNull(waitPakin.getLocCode()) && StringUtils.isNotBlank(waitPakin.getLocCode())) {
+                List<Loc> locs = locService.list(new LambdaQueryWrapper<Loc>().eq(Loc::getCode, waitPakin.getLocCode()));
+//                if (!locs.isEmpty()) {
+//                    throw new CoolException("搴撲綅閿欒锛氱浉鍚屽簱浣嶅簲鍙竴鏉�")
+//                }
+                Loc loc = locs.stream().findFirst().get();
+                task.setTargSite(loc.getCode());
+            } else {
+                task.setTargSite(LocManageUtil.getTargetSite());
+            }
 
             if (!this.save(task)) {
                 throw new CoolException("浠诲姟淇濆瓨澶辫触锛侊紒");
+            }
+            if (!locService.update(new LambdaUpdateWrapper<Loc>().eq(Loc::getCode, task.getTargLoc())
+                    .set(Loc::getUseStatus, LocStsType.LOC_STS_TYPE_S.type).set(Loc::getBarcode, pakin.getBarcode()))) {
+                throw new CoolException("搴撲綅棰勭害澶辫触锛侊紒");
             }
             /**鑾峰彇缁勬嫋鏄庣粏**/
             List<WaitPakinItem> waitPakinItems = waitPakinItemService.list(new LambdaQueryWrapper<WaitPakinItem>().eq(WaitPakinItem::getPakinId, pakin.getId()));
@@ -81,9 +154,14 @@
             waitPakinItems.forEach(item -> {
                 TaskItem taskItem = new TaskItem();
                 BeanUtils.copyProperties(item, taskItem);
+//                AsnOrder order = asnOrderService.getOne(new LambdaQueryWrapper<AsnOrder>().eq(AsnOrder::getId, item.getAsnId()));
+//                if (Objects.isNull(order)) {
+//                    throw new CoolException("鏁版嵁閿欒: 鍗曟嵁涓嶅瓨鍦紒锛�");
+//                }
                 taskItem.setTaskId(task.getId())
                         .setOrderType(OrderType.ORDER_RECEIPT.type)
                         .setSource(item.getId())
+                        .setTrackCode(item.getTrackCode())
                         .setCreateBy(loginUserId)
                         .setUpdateBy(loginUserId)
                         .setOrderId(item.getAsnId())
@@ -93,14 +171,147 @@
             if (!taskItemService.saveBatch(taskItems)) {
                 throw new CoolException("浠诲姟鏄庣粏淇濆瓨澶辫触锛侊紒");
             }
-
-
-
         });
-        //TODO 浠诲姟鐢熸垚瀹屾垚闇�淇敼浠诲姟ASN璁㈠崟鐘舵�佷负鎵ц涓紝缁勬嫋鍒犻櫎闇�鍒ゆ柇鏄惁鏈夊瓙浠诲姟鍦ㄦ墽琛�
 
+        if (!waitPakinService.update(new LambdaUpdateWrapper<WaitPakin>()
+                .in(WaitPakin::getId, ids)
+                .set(WaitPakin::getUpdateBy, loginUserId)
+                .set(WaitPakin::getCreateBy, loginUserId)
+                .set(WaitPakin::getIoStatus, PakinIOStatus.PAKIN_IO_STATUS_TASK_EXCE.val))) {
+            throw new CoolException("缁勬嫋鐘舵�佷慨鏀瑰け璐ワ紒锛�");
+        }
         return R.ok("浠诲姟鐢熸垚瀹屾瘯锛�");
     }
 
+    /**
+     * 瀹屾垚浠诲姟 鏇存柊搴撲綅鏄庣粏淇℃伅锛屽皢鍗曟嵁搴撳瓨鏇存柊鍒板崟鎹簱瀛樿〃
+     * @param tasks
+     * @throws Exception
+     */
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public void completeTask(List<Task> tasks) throws Exception {
+        if (tasks.isEmpty()) {
+            return;
+        }
+        List<Long> list = tasks.stream().map(Task::getId).collect(Collectors.toList());
+        List<TaskItem> taskItems = taskItemService.list(new LambdaQueryWrapper<TaskItem>().in(TaskItem::getTaskId, list));
+        if (taskItems.isEmpty()) {
+            throw new CoolException("浠诲姟鏄庣粏涓嶅瓨鍦紒锛�");
+        }
+        List<String> locCodes = tasks.stream().map(Task::getTargLoc).collect(Collectors.toList());
+        Map<Long, List<TaskItem>> listMap = taskItems.stream().collect(Collectors.groupingBy(TaskItem::getTaskId));
+        /**瀵逛换鍔℃槑缁嗘寜浠诲姟涓诲崟杩涜鍒嗙粍*/
+        listMap.keySet().forEach(key -> {
+            List<TaskItem> items = listMap.get(key);
+            //鏇存柊搴撲綅鏄庣粏
+            try {
+                saveLocItem(items, key);
+            } catch (Exception e) {
+                throw new CoolException("搴撲綅鏄庣粏鏇存柊澶辫触锛侊紒");
+            }
+        });
 
+        /**瀵逛换鍔℃槑缁嗘寜璁㈠崟杩涜鍒嗙粍*/
+        Map<Long, List<TaskItem>> orderMap = taskItems.stream().collect(Collectors.groupingBy(TaskItem::getSource));
+        orderMap.keySet().forEach(key -> {
+            WaitPakinItem pakinItem = waitPakinItemService.getById(key);
+//            AsnOrder order = asnOrderService.getOne(new LambdaQueryWrapper<AsnOrder>()
+//                    .eq(AsnOrder::getId, key)
+//                    .select(AsnOrder::getId, AsnOrder::getPoCode, AsnOrder::getCode));
+            if (null == pakinItem) {
+                throw new CoolException("鏁版嵁閿欒锛氱粍鎷栨暟鎹笉瀛樺湪锛岃鑱旂郴绠$悊鍛橈紒锛�");
+            }
+            List<TaskItem> items = orderMap.get(key);
+            //淇濆瓨搴撳瓨鏄庣粏
+            try {
+                saveStockItems(items, pakinItem);
+            } catch (Exception e) {
+                throw new CoolException("搴撳瓨绠$悊淇濆瓨鏇存柊澶辫触锛侊紒");
+            }
+        });
+        /**淇敼搴撲綅鐘舵�佷负F.鍦ㄥ簱*/
+        if (!locService.update(new LambdaUpdateWrapper<Loc>().set(Loc::getUseStatus, LocStsType.LOC_STS_TYPE_F.type).in(Loc::getCode, locCodes))) {
+            throw new CoolException("搴撲綅鐘舵�佷慨鏀瑰け璐ワ紒锛�");
+        }
+        if (!this.update(new LambdaUpdateWrapper<Task>().in(Task::getId, list).set(Task::getTaskStatus, TaskStsType.UPDATED_IN.id))) {
+            throw new CoolException("浠诲姟鐘舵�佷慨鏀瑰け璐ワ紒锛�");
+        }
+    }
+
+    /**
+     * @author Ryan
+     * @description 鏇存柊搴撲綅鏄庣粏
+     * @param
+     * @return
+     * @time 2025/4/15 15:28
+     */
+    @Transactional(rollbackFor = Exception.class)
+    void saveLocItem(List<TaskItem> items, Long taskId) throws Exception {
+        Task task = this.getById(taskId);
+        if (Objects.isNull(task)) {
+            throw new CoolException("浠诲姟涓嶅瓨鍦�!!");
+        }
+        List<LocItem> locItems = new ArrayList<>();
+        items.forEach(taskItem -> {
+            Loc loc = locService.getOne(new LambdaQueryWrapper<Loc>().eq(Loc::getCode, task.getTargLoc()), false);
+            LocItem item = new LocItem();
+            BeanUtils.copyProperties(taskItem, item);
+            item.setLocCode(loc.getCode()).setId(null).setLocId(loc.getId()).setType(taskItem.getOrderType());
+            locItems.add(item);
+        });
+        if (!locItemService.saveBatch(locItems)) {
+            throw new CoolException("搴撲綅鏄庣粏鏇存柊澶辫触锛侊紒");
+        }
+    }
+
+    /**
+     * 鐢熸垚搴撳瓨鏄庣粏
+     * @param items
+     * @return
+     */
+    @Transactional(rollbackFor = Exception.class)
+    void saveStockItems(List<TaskItem> items, WaitPakinItem order) throws Exception {
+        Stock stock = new Stock();
+//        if (!Objects.isNull(order.getPoCode()) && StringUtils.isNotBlank(order.getPoCode())) {
+//            Purchase purchase = purchaseService.getOne(new LambdaQueryWrapper<Purchase>().eq(Purchase::getCode, order.getPoCode()));
+//            if (!Objects.isNull(purchase)) {
+//                stock.setPlatOrderNo(purchase.getPlatCode()).setPlatToken(purchase.getPlatId());
+//            }
+//        }
+        String ruleCode = SerialRuleUtils.generateRuleCode(SerialRuleCode.SYS_STOCK_CODE, null);
+        if (StringUtils.isBlank(ruleCode)) {
+            throw new CoolException("褰撳墠涓氬姟锛�" + SerialRuleCode.SYS_STOCK_CODE + "锛岀紪鐮佽鍒欎笉瀛樺湪锛侊紒");
+        }
+        double sum = items.stream().mapToDouble(TaskItem::getAnfme).sum();
+        stock.setSourceId(order.getId())
+                .setCode(ruleCode)
+                .setSourceCode(order.getAsnCode())
+                .setAnfme(sum)
+                .setWkType(order.getWkType())
+                .setType(order.getType());
+        if (!stockService.save(stock)) {
+            throw new CoolException("搴撳瓨淇濆瓨澶辫触锛侊紒");
+        }
+
+        List<StockItem> stockItems = new ArrayList<>();
+        for (TaskItem item : items) {
+            /**閫氳繃浠诲姟鏄庣粏涓殑taskId鏌ヨ,鑾峰彇TASK鐨勭洰鏍囧簱浣嶄俊鎭�*/
+//            AsnOrderItem orderItem = asnOrderItemService.getOne(new LambdaQueryWrapper<AsnOrderItem>().eq(AsnOrderItem::getId, item.getOrderItemId()));
+//            if (Objects.isNull(orderItem)) {
+//                throw new CoolException("鍗曟嵁鏄庣粏涓嶅瓨鍦紒锛�");
+//            }
+            StockItem stockItem = new StockItem();
+            BeanUtils.copyProperties(item, stockItem);
+            stockItem.setSourceItemId(item.getOrderItemId())
+                    .setStockCode(stock.getCode())
+                    .setSourceItemId(item.getOrderItemId())
+                    .setId(null)
+                    .setStockId(stock.getId());
+            stockItems.add(stockItem);
+        }
+        if (!stockItemService.saveBatch(stockItems)) {
+            throw new CoolException("搴撳瓨淇敼鏋跺け璐ワ紒锛�");
+        }
+    }
 }

--
Gitblit v1.9.1