From 1428a3da49f299abff59023f516deb7efa0bbdc4 Mon Sep 17 00:00:00 2001
From: vincentlu <t1341870251@gmail.com>
Date: 星期一, 05 一月 2026 17:31:19 +0800
Subject: [PATCH] #

---
 zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/StaReserveServiceImpl.java |   68 ++++++++++++++++++++++++++++++++++
 1 files changed, 68 insertions(+), 0 deletions(-)

diff --git a/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/StaReserveServiceImpl.java b/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/StaReserveServiceImpl.java
index 9dee9af..9d6081e 100644
--- a/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/StaReserveServiceImpl.java
+++ b/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/StaReserveServiceImpl.java
@@ -3,6 +3,7 @@
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.zy.acs.framework.common.Cools;
+import com.zy.acs.framework.exception.CoolException;
 import com.zy.acs.manager.common.exception.BusinessException;
 import com.zy.acs.manager.manager.entity.Sta;
 import com.zy.acs.manager.manager.entity.StaReserve;
@@ -10,6 +11,7 @@
 import com.zy.acs.manager.manager.enums.StaReserveStateType;
 import com.zy.acs.manager.manager.enums.StaReserveType;
 import com.zy.acs.manager.manager.enums.StatusType;
+import com.zy.acs.manager.manager.enums.TaskTypeType;
 import com.zy.acs.manager.manager.mapper.StaMapper;
 import com.zy.acs.manager.manager.mapper.StaReserveMapper;
 import com.zy.acs.manager.manager.service.StaReserveService;
@@ -208,6 +210,72 @@
         }
     }
 
+    @Override
+    public void allocateCallBack(Sta sta, Task task, Long agvId) {
+        if (Cools.isEmpty(sta, task, agvId)) {
+            return;
+        }
+        TaskTypeType taskType = TaskTypeType.get(task.getTaskTypeEl());
+        if (null == taskType) {
+            return;
+        }
+        Date now = new Date();
+        StaReserve inStaReserve = null;
+        StaReserve outStaReserve = null;
+        switch (taskType) {
+            case LOC_TO_STA:
+                inStaReserve = this.getOne(new LambdaQueryWrapper<StaReserve>()
+                                .eq(StaReserve::getTaskId, task.getId())
+                                .eq(StaReserve::getStaId, sta.getId())
+                                .eq(StaReserve::getType, StaReserveType.IN.toString())
+                                .eq(StaReserve::getStatus, StaReserveStateType.RESERVED.toString())
+//                        .gt(StaReserve::getExpireTime, now)
+                );
+                break;
+            case STA_TO_LOC:
+                outStaReserve = this.getOne(new LambdaQueryWrapper<StaReserve>()
+                                .eq(StaReserve::getTaskId, task.getId())
+                                .eq(StaReserve::getStaId, sta.getId())
+                                .eq(StaReserve::getType, StaReserveType.OUT.toString())
+                                .eq(StaReserve::getStatus, StaReserveStateType.RESERVED.toString())
+//                        .gt(StaReserve::getExpireTime, now)
+                );
+                break;
+            case STA_TO_STA:
+                inStaReserve = this.getOne(new LambdaQueryWrapper<StaReserve>()
+                                .eq(StaReserve::getTaskId, task.getId())
+                                .eq(StaReserve::getStaId, sta.getId())
+                                .eq(StaReserve::getType, StaReserveType.IN.toString())
+                                .eq(StaReserve::getStatus, StaReserveStateType.RESERVED.toString())
+//                        .gt(StaReserve::getExpireTime, now)
+                );
+                outStaReserve = this.getOne(new LambdaQueryWrapper<StaReserve>()
+                                .eq(StaReserve::getTaskId, task.getId())
+                                .eq(StaReserve::getStaId, sta.getId())
+                                .eq(StaReserve::getType, StaReserveType.OUT.toString())
+                                .eq(StaReserve::getStatus, StaReserveStateType.RESERVED.toString())
+//                        .gt(StaReserve::getExpireTime, now)
+                );
+                break;
+            default:
+                return;
+        }
+        if (null != inStaReserve) {
+            inStaReserve.setAgvId(agvId);
+            inStaReserve.setUpdateTime(now);
+            if (!this.updateById(inStaReserve)) {
+                throw new CoolException("Sta[" + sta.getStaNo() + "] reserve failed");
+            }
+        }
+        if (null != outStaReserve) {
+            outStaReserve.setAgvId(agvId);
+            outStaReserve.setUpdateTime(now);
+            if (!this.updateById(outStaReserve)) {
+                throw new CoolException("Sta[" + sta.getStaNo() + "] reserve failed");
+            }
+        }
+    }
+
 
     private String generateReserveUniqKey(Long staId, Long taskId, StaReserveType type) {
         return "STA:" + staId + "-TASK:" + taskId + "-T:" + type.toString();

--
Gitblit v1.9.1