From 6cbb420754e6e29fa134a6afca4514b8dfd62918 Mon Sep 17 00:00:00 2001
From: vincentlu <t1341870251@gmail.com>
Date: 星期二, 13 一月 2026 16:14:49 +0800
Subject: [PATCH] #

---
 zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/TaskServiceImpl.java |  228 ++++++++++++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 209 insertions(+), 19 deletions(-)

diff --git a/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/TaskServiceImpl.java b/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/TaskServiceImpl.java
index f37814e..1f0ed1e 100644
--- a/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/TaskServiceImpl.java
+++ b/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/TaskServiceImpl.java
@@ -6,9 +6,13 @@
 import com.zy.acs.framework.common.Cools;
 import com.zy.acs.framework.common.SnowflakeIdWorker;
 import com.zy.acs.framework.exception.CoolException;
+import com.zy.acs.manager.common.constant.Constants;
+import com.zy.acs.manager.common.domain.BaseParam;
+import com.zy.acs.manager.common.domain.PageParam;
+import com.zy.acs.manager.common.domain.PageResult;
 import com.zy.acs.manager.common.exception.BusinessException;
-import com.zy.acs.manager.core.domain.Lane;
-import com.zy.acs.manager.core.service.LaneService;
+import com.zy.acs.manager.core.domain.LaneDto;
+import com.zy.acs.manager.core.service.LaneBuilder;
 import com.zy.acs.manager.manager.entity.*;
 import com.zy.acs.manager.manager.enums.*;
 import com.zy.acs.manager.manager.mapper.TaskMapper;
@@ -25,6 +29,8 @@
 public class TaskServiceImpl extends ServiceImpl<TaskMapper, Task> implements TaskService {
 
     @Autowired
+    private BusService busService;
+    @Autowired
     private CodeService codeService;
     @Autowired
     private LocService locService;
@@ -35,7 +41,20 @@
     @Autowired
     private SnowflakeIdWorker snowflakeIdWorker;
     @Autowired
-    private LaneService laneService;
+    private LaneBuilder laneBuilder;
+    @Autowired
+    private StaReserveService staReserveService;
+    @Autowired
+    private SegmentService segmentService;
+    @Autowired
+    private ActionService actionService;
+    @Autowired
+    private TravelService travelService;
+
+    @Override
+    public PageResult<Task> pageRel(PageParam<Task, BaseParam> pageParam) {
+        return new PageResult<>(this.baseMapper.selectPageRel(pageParam, pageParam.checkoutMap()), pageParam.getTotal());
+    }
 
     @Override
     public synchronized String generateSeqNum() {
@@ -86,13 +105,36 @@
         if (null == task) {
             return Boolean.FALSE;
         }
-        this.maintainLocSts(task, Boolean.TRUE);
-
+        Date now = new Date();
+        // stock
+        this.maintainLocAndStaHandler(task, Boolean.TRUE);
+        // task
         task.setTaskSts(TaskStsType.COMPLETE.val());
-        task.setUpdateTime(new Date());
+        task.setUpdateTime(now);
         task.setUpdateBy(userId);
+        task.setMemo(Constants.HANDLE + " " + TaskStsType.COMPLETE);
         if (!this.updateById(task)) {
             throw new CoolException(BaseRes.ERROR);
+        }
+        // bus
+        busService.checkoutComplete(task.getBusId());
+        // segment
+        List<Segment> segments = segmentService.list(new LambdaQueryWrapper<Segment>().eq(Segment::getTaskId, taskId));
+        if (!Cools.isEmpty(segments)) {
+            for (Segment segment : segments) {
+                if (segment.getState().equals(SegmentStateType.FINISH.toString())) {
+                    continue;
+                }
+                segment.setState(SegmentStateType.FINISH.toString());
+                segment.setUpdateTime(now);
+                segment.setUpdateBy(userId);
+                segment.setMemo(Constants.HANDLE);
+                if (!segmentService.updateById(segment)) {
+                    throw new CoolException(BaseRes.ERROR);
+                }
+            }
+            // travel
+            travelService.checkFinish(segments.get(0).getTravelId());
         }
         return Boolean.TRUE;
     }
@@ -104,19 +146,41 @@
         if (null == task) {
             return Boolean.FALSE;
         }
-        this.maintainLocSts(task, Boolean.FALSE);
-
+        Date now = new Date();
+        // stock
+        this.maintainLocAndStaHandler(task, Boolean.FALSE);
+        // task
         task.setTaskSts(TaskStsType.CANCEL.val());
-        task.setUpdateTime(new Date());
+        task.setUpdateTime(now);
         task.setUpdateBy(userId);
+        task.setMemo(Constants.HANDLE + " " + TaskStsType.CANCEL);
         if (!this.updateById(task)) {
             throw new CoolException(BaseRes.ERROR);
+        }
+        busService.checkoutComplete(task.getBusId());
+        // segment
+        List<Segment> segments = segmentService.list(new LambdaQueryWrapper<Segment>().eq(Segment::getTaskId, taskId));
+        if (!Cools.isEmpty(segments)) {
+            for (Segment segment : segments) {
+                if (segment.getState().equals(SegmentStateType.FINISH.toString())) {
+                    continue;
+                }
+                segment.setState(SegmentStateType.FINISH.toString());
+                segment.setUpdateTime(now);
+                segment.setUpdateBy(userId);
+                segment.setMemo(Constants.HANDLE);
+                if (!segmentService.updateById(segment)) {
+                    throw new CoolException(BaseRes.ERROR);
+                }
+            }
+            // travel
+            travelService.checkFinish(segments.get(0).getTravelId());
         }
         return Boolean.TRUE;
     }
 
     @Override
-    public Lane checkoutOriginLane(Task task) {
+    public LaneDto checkoutOriginLane(Task task) {
         Long codeId = null;
         TaskTypeType typeType = TaskTypeType.get(task.getTaskTypeEl());
         switch (Objects.requireNonNull(typeType)) {
@@ -135,11 +199,11 @@
         if (null == codeId) {
             return null;
         }
-        return laneService.search(codeService.getById(codeId).getData());
+        return laneBuilder.search(codeService.getCacheById(codeId).getData());
     }
 
     @Override
-    public Lane checkoutDestinationLane(Task task) {
+    public LaneDto checkoutDestinationLane(Task task) {
         Long codeId = null;
         TaskTypeType typeType = TaskTypeType.get(task.getTaskTypeEl());
         switch (Objects.requireNonNull(typeType)) {
@@ -158,7 +222,7 @@
         if (null == codeId) {
             return null;
         }
-        return laneService.search(codeService.getById(codeId).getData());
+        return laneBuilder.search(codeService.getCacheById(codeId).getData());
     }
 
     @Override
@@ -176,7 +240,7 @@
     }
 
     @Override
-    public List<Task> findRunningTasksByAgv(Long agvId) {
+    public List<Task> findTransportTasksByAgv(Long agvId) {
         if (null == agvId) {
             return new ArrayList<>();
         }
@@ -186,8 +250,107 @@
         return this.list(wrapper);
     }
 
+    @Override
+    public Integer findTransportTasksCountByAgv(Long agvId) {
+        LambdaQueryWrapper<Task> wrapper = new LambdaQueryWrapper<Task>().eq(Task::getAgvId, agvId);
+        wrapper.in(Task::getTaskSts, TaskStsType.WAITING.val(), TaskStsType.ASSIGN.val(), TaskStsType.PROGRESS.val());
+        wrapper.notIn(Task::getTaskType, TaskTypeType.MOVE.val(), TaskTypeType.TO_CHARGE.val(), TaskTypeType.TO_STANDBY.val());
+        return this.count(wrapper);
+    }
+
+    @Override
+    public Task findLatestTask(Long agvId, TaskStsType taskSts) {
+        LambdaQueryWrapper<Task> wrapper = new LambdaQueryWrapper<Task>()
+                .orderByDesc(Task::getCreateTime)
+                .last("limit 0, 1");
+        if (null != agvId) {
+            wrapper.eq(Task::getAgvId, agvId);
+        }
+        if (null != taskSts) {
+            wrapper.eq(Task::getTaskSts, taskSts.val());
+        }
+        return this.list(wrapper).stream().findFirst().orElse(null);
+    }
+
+    @Override
+    public void maintainLocAndSta(Task task) {
+        if (null == task) {
+            return;
+        }
+        if (!task.getTaskSts().equals(TaskStsType.COMPLETE.val())) {
+            return;
+        }
+        Date now = new Date();
+        // loc status
+        Loc oriLoc = null;
+        Loc destLoc = null;
+        Sta oriSta = null;
+        Sta destSta = null;
+        switch (Objects.requireNonNull(TaskTypeType.get(task.getTaskTypeEl()))) {
+            case LOC_TO_LOC:
+                oriLoc = locService.getById(task.getOriLoc());
+                if (oriLoc.getLocSts().equals(LocStsType.PAKOUT.val())) {
+                    oriLoc.setLocSts(LocStsType.IDLE.val());
+                    oriLoc.setUpdateTime(now);
+                    if (!locService.updateById(oriLoc)) {
+                        log.error("Loc [{}] 搴撲綅淇敼鐘舵�佸け璐�", task.getOriLoc$());
+                    }
+                }
+
+                destLoc = locService.getById(task.getDestLoc());
+                if (destLoc.getLocSts().equals(LocStsType.PAKIN.val())) {
+                    destLoc.setLocSts(LocStsType.STOCK.val());
+                    destLoc.setUpdateTime(now);
+                    if (!locService.updateById(destLoc)) {
+                        log.error("Loc [{}] 搴撲綅淇敼鐘舵�佸け璐�", task.getDestLoc$());
+                    }
+                }
+                break;
+            case LOC_TO_STA:
+                oriLoc = locService.getById(task.getOriLoc());
+                if (oriLoc.getLocSts().equals(LocStsType.PAKOUT.val())) {
+                    oriLoc.setLocSts(LocStsType.IDLE.val());
+                    oriLoc.setUpdateTime(now);
+                    if (!locService.updateById(oriLoc)) {
+                        log.error("Loc [{}] 搴撲綅淇敼鐘舵�佸け璐�", task.getOriLoc$());
+                    }
+                }
+
+                destSta = staService.getById(task.getDestSta());
+                staReserveService.confirmStaReserve(destSta, task, 1, StaReserveType.IN);
+                break;
+            case STA_TO_LOC:
+                oriSta = staService.getById(task.getOriSta());
+                staReserveService.confirmStaReserve(oriSta, task, 1, StaReserveType.OUT);
+
+                destLoc = locService.getById(task.getDestLoc());
+                if (destLoc.getLocSts().equals(LocStsType.PAKIN.val())) {
+                    destLoc.setLocSts(LocStsType.STOCK.val());
+                    destLoc.setUpdateTime(now);
+                    if (!locService.updateById(destLoc)) {
+                        log.error("Loc [{}] 搴撲綅淇敼鐘舵�佸け璐�", task.getDestLoc$());
+                    }
+                }
+                break;
+            case STA_TO_STA:
+                oriSta = staService.getById(task.getOriSta());
+                staReserveService.confirmStaReserve(oriSta, task, 1, StaReserveType.OUT);
+
+                destSta = staService.getById(task.getDestSta());
+                staReserveService.confirmStaReserve(destSta, task, 1, StaReserveType.IN);
+                break;
+            case TO_CHARGE:
+            case TO_STANDBY:
+            case MOVE:
+                break;
+            default:
+                break;
+        }
+    }
+
+
     @Transactional
-    public void maintainLocSts(Task task, Boolean complete) {
+    public void maintainLocAndStaHandler(Task task, Boolean complete) {
         Loc oriLoc = null; Loc destLoc = null;
         Sta oriSta = null; Sta destSta = null;
         Date now = new Date();
@@ -200,13 +363,13 @@
                 oriLoc.setLocSts(complete?LocStsType.IDLE.val():LocStsType.STOCK.val());
                 oriLoc.setUpdateTime(now);
                 if (!locService.updateById(oriLoc)) {
-                    throw new BusinessException("Loc [" + task.getOriLoc$() + "] 搴撲綅淇敼鐘舵�佸け璐� 锛侊紒锛�");
+                    throw new BusinessException("Loc [" + oriLoc.getLocNo() + "] 搴撲綅淇敼鐘舵�佸け璐� 锛侊紒锛�");
                 }
 
                 destLoc.setLocSts(complete?LocStsType.STOCK.val():LocStsType.IDLE.val());
                 destLoc.setUpdateTime(now);
                 if (!locService.updateById(destLoc)) {
-                    throw new BusinessException("Loc [" + task.getDestLoc$() + "] 搴撲綅淇敼鐘舵�佸け璐� 锛侊紒锛�");
+                    throw new BusinessException("Loc [" + destLoc.getLocNo() + "] 搴撲綅淇敼鐘舵�佸け璐� 锛侊紒锛�");
                 }
                 break;
             case LOC_TO_STA:
@@ -214,18 +377,45 @@
                 oriLoc.setLocSts(complete?LocStsType.IDLE.val():LocStsType.STOCK.val());
                 oriLoc.setUpdateTime(now);
                 if (!locService.updateById(oriLoc)) {
-                    throw new BusinessException("Loc [" + task.getOriLoc$() + "] 搴撲綅淇敼鐘舵�佸け璐� 锛侊紒锛�");
+                    throw new BusinessException("Loc [" + oriLoc.getLocNo() + "] 搴撲綅淇敼鐘舵�佸け璐� 锛侊紒锛�");
+                }
+
+                destSta = staService.getById(task.getDestSta());
+                if (complete) {
+                    staReserveService.confirmStaReserve(destSta, task, 1, StaReserveType.IN);
+                } else {
+                    staReserveService.cancelStaReserve(destSta, task, 1, StaReserveType.IN);
                 }
                 break;
             case STA_TO_LOC:
+                oriSta = staService.getById(task.getOriSta());
+                if (complete) {
+                    staReserveService.confirmStaReserve(oriSta, task, 1, StaReserveType.OUT);
+                } else {
+                    staReserveService.cancelStaReserve(oriSta, task, 1, StaReserveType.OUT);
+                }
+
                 destLoc = locService.getById(task.getDestLoc());
                 destLoc.setLocSts(complete?LocStsType.STOCK.val():LocStsType.IDLE.val());
                 destLoc.setUpdateTime(now);
                 if (!locService.updateById(destLoc)) {
-                    throw new BusinessException("Loc [" + task.getDestLoc$() + "] 搴撲綅淇敼鐘舵�佸け璐� 锛侊紒锛�");
+                    throw new BusinessException("Loc [" + destLoc.getLocNo() + "] 搴撲綅淇敼鐘舵�佸け璐� 锛侊紒锛�");
                 }
                 break;
             case STA_TO_STA:
+                oriSta = staService.getById(task.getOriSta());
+                if (complete) {
+                    staReserveService.confirmStaReserve(oriSta, task, 1, StaReserveType.OUT);
+                } else {
+                    staReserveService.cancelStaReserve(oriSta, task, 1, StaReserveType.OUT);
+                }
+
+                destSta = staService.getById(task.getDestSta());
+                if (complete) {
+                    staReserveService.confirmStaReserve(destSta, task, 1, StaReserveType.IN);
+                } else {
+                    staReserveService.cancelStaReserve(destSta, task, 1, StaReserveType.IN);
+                }
                 break;
             case TO_CHARGE:
             case TO_STANDBY:

--
Gitblit v1.9.1