From ebd2f4397a92c6a5096de1b86d59154363344720 Mon Sep 17 00:00:00 2001
From: vincentlu <t1341870251@gmail.com>
Date: 星期二, 13 五月 2025 08:48:15 +0800
Subject: [PATCH] #

---
 zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/TaskServiceImpl.java |  250 +++++++++++++++++++++++++++++++++++++------------
 1 files changed, 189 insertions(+), 61 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 2e9a36d..07c84af 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,17 +6,16 @@
 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.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.manager.entity.FuncSta;
-import com.zy.acs.manager.manager.entity.Loc;
-import com.zy.acs.manager.manager.entity.Sta;
-import com.zy.acs.manager.manager.entity.Task;
+import com.zy.acs.manager.core.domain.Lane;
+import com.zy.acs.manager.core.service.LaneService;
+import com.zy.acs.manager.manager.entity.*;
 import com.zy.acs.manager.manager.enums.*;
 import com.zy.acs.manager.manager.mapper.TaskMapper;
-import com.zy.acs.manager.manager.service.CodeService;
-import com.zy.acs.manager.manager.service.FuncStaService;
-import com.zy.acs.manager.manager.service.LocService;
-import com.zy.acs.manager.manager.service.TaskService;
+import com.zy.acs.manager.manager.service.*;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
@@ -29,13 +28,24 @@
 public class TaskServiceImpl extends ServiceImpl<TaskMapper, Task> implements TaskService {
 
     @Autowired
+    private BusService busService;
+    @Autowired
     private CodeService codeService;
     @Autowired
     private LocService locService;
     @Autowired
+    private StaService staService;
+    @Autowired
     private FuncStaService funcStaService;
     @Autowired
     private SnowflakeIdWorker snowflakeIdWorker;
+    @Autowired
+    private LaneService laneService;
+
+    @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() {
@@ -75,9 +85,143 @@
     }
 
     @Override
+    public List<Map<String, Object>> selectStatByLastSevenDays() {
+        return this.baseMapper.selectStatByLastSevenDays();
+    }
+
+    @Override
     @Transactional
-    public void operateTaskSts(Integer taskId, Boolean finish) {
+    public Boolean complete(Long taskId, Long userId) {
         Task task = this.getById(taskId);
+        if (null == task) {
+            return Boolean.FALSE;
+        }
+        this.maintainLocSts(task, Boolean.TRUE);
+
+        task.setTaskSts(TaskStsType.COMPLETE.val());
+        task.setUpdateTime(new Date());
+        task.setUpdateBy(userId);
+        if (!this.updateById(task)) {
+            throw new CoolException(BaseRes.ERROR);
+        }
+        busService.checkoutComplete(task.getBusId());
+        return Boolean.TRUE;
+    }
+
+    @Override
+    @Transactional
+    public Boolean cancel(Long taskId, Long userId) {
+        Task task = this.getById(taskId);
+        if (null == task) {
+            return Boolean.FALSE;
+        }
+        this.maintainLocSts(task, Boolean.FALSE);
+
+        task.setTaskSts(TaskStsType.CANCEL.val());
+        task.setUpdateTime(new Date());
+        task.setUpdateBy(userId);
+        if (!this.updateById(task)) {
+            throw new CoolException(BaseRes.ERROR);
+        }
+        busService.checkoutComplete(task.getBusId());
+        return Boolean.TRUE;
+    }
+
+    @Override
+    public Lane checkoutOriginLane(Task task) {
+        Long codeId = null;
+        TaskTypeType typeType = TaskTypeType.get(task.getTaskTypeEl());
+        switch (Objects.requireNonNull(typeType)) {
+            case LOC_TO_LOC:
+            case LOC_TO_STA:
+                codeId = locService.getById(task.getOriLoc()).getCode();
+                break;
+            case STA_TO_LOC:
+            case STA_TO_STA:
+                codeId = staService.getById(task.getOriSta()).getCode();
+                break;
+            default:
+                break;
+        }
+
+        if (null == codeId) {
+            return null;
+        }
+        return laneService.search(codeService.getCacheById(codeId).getData());
+    }
+
+    @Override
+    public Lane checkoutDestinationLane(Task task) {
+        Long codeId = null;
+        TaskTypeType typeType = TaskTypeType.get(task.getTaskTypeEl());
+        switch (Objects.requireNonNull(typeType)) {
+            case LOC_TO_LOC:
+            case STA_TO_LOC:
+                codeId = locService.getById(task.getDestLoc()).getCode();
+                break;
+            case LOC_TO_STA:
+            case STA_TO_STA:
+                codeId = staService.getById(task.getDestSta()).getCode();
+                break;
+            default:
+                break;
+        }
+
+        if (null == codeId) {
+            return null;
+        }
+        return laneService.search(codeService.getCacheById(codeId).getData());
+    }
+
+    @Override
+    public List<Task> findRunningTasksByLaneHash(String laneHash) {
+        if (Cools.isEmpty(laneHash)) {
+            return new ArrayList<>();
+        }
+        return this.list(new LambdaQueryWrapper<Task>()
+                .in(Task::getTaskSts, TaskStsType.WAITING.val(), TaskStsType.ASSIGN.val(), TaskStsType.PROGRESS.val())
+                .and(i -> {
+                    i.eq(Task::getOriLaneHash, laneHash).or().eq(Task::getDestLaneHash, laneHash);
+                })
+
+        );
+    }
+
+    @Override
+    public List<Task> findTransportTasksByAgv(Long agvId) {
+        if (null == agvId) {
+            return new ArrayList<>();
+        }
+        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.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);
+    }
+
+    @Transactional
+    public void maintainLocSts(Task task, Boolean complete) {
         Loc oriLoc = null; Loc destLoc = null;
         Sta oriSta = null; Sta destSta = null;
         Date now = new Date();
@@ -87,40 +231,67 @@
                 oriLoc = locService.getById(task.getOriLoc());
                 destLoc = locService.getById(task.getDestLoc());
 
-                oriLoc.setLocSts(finish?LocStsType.IDLE.val():LocStsType.STOCK.val());
+                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(finish?LocStsType.STOCK.val():LocStsType.IDLE.val());
+                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:
                 oriLoc = locService.getById(task.getOriLoc());
-                oriLoc.setLocSts(finish?LocStsType.IDLE.val():LocStsType.STOCK.val());
+                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());
+                destSta.setStaSts(complete?StaStsType.STOCK.val():StaStsType.IDLE.val());
+                destSta.setUpdateTime(now);
+                if (!staService.updateById(destSta)) {
+                    throw new BusinessException("Sta [" + destSta.getStaNo() + "] 绔欑偣淇敼鐘舵�佸け璐� 锛侊紒锛�");
                 }
                 break;
             case STA_TO_LOC:
+                oriSta = staService.getById(task.getOriSta());
+                oriSta.setStaSts(complete?StaStsType.IDLE.val():StaStsType.STOCK.val());
+                oriSta.setUpdateTime(now);
+                if (!staService.updateById(oriSta)) {
+                    throw new BusinessException("Sta [" + oriSta.getStaNo() + "] 绔欑偣淇敼鐘舵�佸け璐� 锛侊紒锛�");
+                }
+
                 destLoc = locService.getById(task.getDestLoc());
-                destLoc.setLocSts(finish?LocStsType.STOCK.val():LocStsType.IDLE.val());
+                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());
+                oriSta.setStaSts(complete?StaStsType.IDLE.val():StaStsType.STOCK.val());
+                oriSta.setUpdateTime(now);
+                if (!staService.updateById(oriSta)) {
+                    throw new BusinessException("Sta [" + oriSta.getStaNo() + "] 绔欑偣淇敼鐘舵�佸け璐� 锛侊紒锛�");
+                }
+
+                destSta = staService.getById(task.getDestSta());
+                destSta.setStaSts(complete?StaStsType.STOCK.val():StaStsType.IDLE.val());
+                destSta.setUpdateTime(now);
+                if (!staService.updateById(destSta)) {
+                    throw new BusinessException("Sta [" + destSta.getStaNo() + "] 绔欑偣淇敼鐘舵�佸け璐� 锛侊紒锛�");
+                }
                 break;
             case TO_CHARGE:
             case TO_STANDBY:
                 FuncSta funcSta = funcStaService.getByCodeAndType(task.getDestCode(), FuncStaType.query(typeType).toString());
-                if (!finish) {
+                if (!complete) {
                     funcSta.setState(FuncStaStateType.IDLE.toString());
                     funcSta.setUpdateTime(now);
                     if (!funcStaService.updateById(funcSta)) {
@@ -131,49 +302,6 @@
             default:
                 break;
         }
-        task.setTaskSts(TaskStsType.COMPLETE.val());
-        task.setUpdateTime(now);
-        if (!this.updateById(task)) {
-            throw new BusinessException(task.getSeqNum() + "浠诲姟鏇存柊鐘舵�佸け璐�");
-        }
     }
-
-    @Override
-    public List<Map<String, Object>> selectStatByLastSevenDays() {
-        return this.baseMapper.selectStatByLastSevenDays();
-    }
-
-    @Override
-    public Boolean complete(Long taskId, Long userId) {
-        Task task = this.getById(taskId);
-        if (null == task) {
-            return Boolean.FALSE;
-        }
-
-        task.setTaskSts(TaskStsType.COMPLETE.val());
-        task.setUpdateTime(new Date());
-        task.setUpdateBy(userId);
-        if (!this.updateById(task)) {
-            throw new CoolException(BaseRes.ERROR);
-        }
-        return Boolean.TRUE;
-    }
-
-    @Override
-    public Boolean cancel(Long taskId, Long userId) {
-        Task task = this.getById(taskId);
-        if (null == task) {
-            return Boolean.FALSE;
-        }
-
-        task.setTaskSts(TaskStsType.CANCEL.val());
-        task.setUpdateTime(new Date());
-        task.setUpdateBy(userId);
-        if (!this.updateById(task)) {
-            throw new CoolException(BaseRes.ERROR);
-        }
-        return Boolean.TRUE;
-    }
-
 
 }

--
Gitblit v1.9.1