From bfac8bfa522289b8d2c0b66c50c4bf8b8d75c0e1 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@163.com>
Date: 星期二, 29 十月 2024 14:27:02 +0800
Subject: [PATCH] #

---
 zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/FuncStaServiceImpl.java |  145 +++++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 142 insertions(+), 3 deletions(-)

diff --git a/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/FuncStaServiceImpl.java b/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/FuncStaServiceImpl.java
index 680bfde..ae89023 100644
--- a/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/FuncStaServiceImpl.java
+++ b/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/FuncStaServiceImpl.java
@@ -2,16 +2,37 @@
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
-import com.zy.acs.manager.manager.entity.FuncSta;
-import com.zy.acs.manager.manager.mapper.FuncStaMapper;
-import com.zy.acs.manager.manager.service.FuncStaService;
+import com.zy.acs.common.enums.AgvStatusType;
 import com.zy.acs.framework.common.Cools;
+import com.zy.acs.manager.manager.entity.*;
+import com.zy.acs.manager.manager.enums.FuncStaType;
+import com.zy.acs.manager.manager.enums.StatusType;
+import com.zy.acs.manager.manager.enums.TaskStsType;
+import com.zy.acs.manager.manager.enums.TaskTypeType;
+import com.zy.acs.manager.manager.mapper.FuncStaMapper;
+import com.zy.acs.manager.manager.service.*;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.Comparator;
 import java.util.List;
+import java.util.stream.Collectors;
 
 @Service("funcStaService")
 public class FuncStaServiceImpl extends ServiceImpl<FuncStaMapper, FuncSta> implements FuncStaService {
+
+    @Autowired
+    private CodeService codeService;
+    @Autowired
+    private AgvService agvService;
+    @Autowired
+    private AgvDetailService agvDetailService;
+    @Autowired
+    private AgvModelService agvModelService;
+    @Autowired
+    private TaskService taskService;
 
     @Override
     public FuncSta getByCodeAndType(Long codeId, String type) {
@@ -32,4 +53,122 @@
         return list.stream().findFirst().orElse(null);
     }
 
+    @Override
+    public List<FuncSta> findInIdleStatus(FuncStaType type, Long agvId) {
+        LambdaQueryWrapper<FuncSta> wrapper = new LambdaQueryWrapper<FuncSta>()
+                .eq(FuncSta::getType, type).eq(FuncSta::getStatus, StatusType.ENABLE.val);
+        if (null != agvId) {
+            wrapper.eq(FuncSta::getAgvId, agvId);
+        }
+        List<FuncSta> funcStaList = this.list(wrapper);
+        funcStaList = funcStaList.stream().filter(funcSta -> {
+            if (null != funcSta.getAgvId()) {
+                if (!funcSta.getAgvId().equals(agvId)) {
+                    return false;
+                }
+            }
+
+            return true;
+        }).collect(Collectors.toList());
+
+        if (Cools.isEmpty(funcStaList)) {
+            return new ArrayList<>();
+        }
+
+        Collections.shuffle(funcStaList);
+        // filter idle
+        funcStaList = funcStaList.stream().filter(funcSta -> {
+
+            Long code = funcSta.getCode();
+            if (null == code) {
+                return false;
+            }
+
+            Agv existAgv = agvService.findByPosition(code);
+
+            if (funcSta.getType().equals(FuncStaType.CHARGE.toString())) {
+                // if the type of this funSta is charge and the existing agv is in charge status, then that means this funSta is occupied
+                if (null != existAgv) {
+                    AgvDetail agvDetail = agvDetailService.selectByAgvId(existAgv.getId());
+                    AgvModel agvModel = agvModelService.getById(existAgv.getAgvModel());
+                    if (agvDetail.getAgvStatus().equals(AgvStatusType.CHARGE)) {
+                        if (agvDetail.getVol() < agvModel.getQuaBattery()) {
+                            return false;
+                        }
+                    }
+                }
+
+                // if there is a running task whose destination is this funSta, then that means this funSta is occupied
+                if (0 < taskService.count(new LambdaQueryWrapper<Task>()
+                        .eq(Task::getTaskType, TaskTypeType.TO_CHARGE.val())
+                        .in(Task::getTaskSts, TaskStsType.ASSIGN.val(), TaskStsType.PROGRESS.val())
+                        .eq(Task::getDestCode, code)
+                )) {
+                    return false;
+                }
+
+            }
+
+            if (funcSta.getType().equals(FuncStaType.STANDBY.toString())) {
+
+                if (null == existAgv) {
+                    // if there is a running task whose destination is this funSta, then that means this funSta is occupied
+                    if (0 < taskService.count(new LambdaQueryWrapper<Task>()
+                            .eq(Task::getTaskType, TaskTypeType.TO_STANDBY.val())
+                            .in(Task::getTaskSts, TaskStsType.ASSIGN.val(), TaskStsType.PROGRESS.val())
+                            .eq(Task::getDestCode, code)
+                    )) {
+                        return false;
+                    }
+
+                } else {
+                    // if there is an agv on the code of this funSta, should we let this agv leave?
+                    // we need to judge whether the agv went to this funSta based on a task which in GO_STANDBY type
+
+//                    Task latestTaskByAgv = taskService.findLatestTask(existAgv.getId());
+
+                }
+            }
+
+            return true;
+        }).collect(Collectors.toList());
+
+        return funcStaList;
+    }
+
+    @Override
+    public FuncSta checkoutClosestFunSta(Long codeId, List<FuncSta> funcStaList) {
+        if (Cools.isEmpty(funcStaList)) {
+            return null;
+        }
+
+        if (null != codeId) {
+            Code currCode = codeService.getById(codeId);
+            Double[] startPos = new Double[]{currCode.getX(), currCode.getY()};
+
+            // checkout one funSta which is the closest
+
+            // compare => compare杩斿洖璐熸暟锛屽垯鎺掑湪闆嗗悎鍓嶉潰 锛坅sc锛�
+            funcStaList.sort(new Comparator<FuncSta>() {
+                @Override
+                public int compare(FuncSta o1, FuncSta o2) {
+                    Code o1Code = codeService.getById(o1.getCode());
+                    int o1Distance = calcPositionDistance(startPos, new Double[]{o1Code.getX(), o1Code.getY()});
+
+                    Code o2Code = codeService.getById(o2.getCode());
+                    int o2Distance = calcPositionDistance(startPos, new Double[]{o2Code.getX(), o2Code.getY()});
+
+                    return o1Distance - o2Distance;
+                }
+            });
+
+        }
+
+        return funcStaList.stream().findFirst().orElse(null);
+    }
+
+    private int calcPositionDistance(Double[] from, Double[] to) {
+        return (int) (Math.abs(to[0] - from[0]) + Math.abs(to[1] - from[1]));
+    }
+
 }

--
Gitblit v1.9.1