From 1313906bb1eb983d3beece810035e7fc28d6a92f Mon Sep 17 00:00:00 2001
From: zhang <zc857179121@qq.com>
Date: 星期二, 20 五月 2025 16:57:11 +0800
Subject: [PATCH] 新增充电桩

---
 zy-acs-manager/src/main/java/com/zy/acs/manager/core/scheduler/KernelScheduler.java |  112 ++++++++++++++++++++++++++++++++++++++++++++-----------
 1 files changed, 89 insertions(+), 23 deletions(-)

diff --git a/zy-acs-manager/src/main/java/com/zy/acs/manager/core/scheduler/KernelScheduler.java b/zy-acs-manager/src/main/java/com/zy/acs/manager/core/scheduler/KernelScheduler.java
index a2cf33c..43d4ac0 100644
--- a/zy-acs-manager/src/main/java/com/zy/acs/manager/core/scheduler/KernelScheduler.java
+++ b/zy-acs-manager/src/main/java/com/zy/acs/manager/core/scheduler/KernelScheduler.java
@@ -2,23 +2,26 @@
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.zy.acs.manager.core.domain.AgvTaskDto;
+import com.zy.acs.manager.core.service.MainLockWrapService;
 import com.zy.acs.manager.core.service.MainService;
 import com.zy.acs.manager.core.service.TrafficService;
 import com.zy.acs.manager.manager.entity.Bus;
 import com.zy.acs.manager.manager.entity.Segment;
 import com.zy.acs.manager.manager.entity.Task;
+import com.zy.acs.manager.manager.enums.ActionStsType;
 import com.zy.acs.manager.manager.enums.BusStsType;
 import com.zy.acs.manager.manager.enums.SegmentStateType;
 import com.zy.acs.manager.manager.enums.TaskStsType;
-import com.zy.acs.manager.manager.service.ActionService;
-import com.zy.acs.manager.manager.service.BusService;
-import com.zy.acs.manager.manager.service.SegmentService;
-import com.zy.acs.manager.manager.service.TaskService;
+import com.zy.acs.manager.manager.service.*;
+import com.zy.acs.manager.system.service.ConfigService;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.context.event.ApplicationReadyEvent;
+import org.springframework.context.event.EventListener;
 import org.springframework.scheduling.annotation.Scheduled;
 import org.springframework.stereotype.Component;
 
+import javax.annotation.PreDestroy;
 import java.util.ArrayList;
 import java.util.Date;
 import java.util.List;
@@ -33,9 +36,18 @@
 @Component
 public class KernelScheduler {
 
+    public static int CORE_SCAN_FREQUENCY_MILLISECOND = 15;
+
     private static final int LOCK_TIMEOUT = 5;
+
     private final ReentrantLock lock = new ReentrantLock(Boolean.TRUE);
 
+    private Thread trafficCalcThread;
+
+    private Thread actionPublicThread;
+
+    @Autowired
+    private AgvService agvService;
     @Autowired
     private BusService busService;
     @Autowired
@@ -43,23 +55,32 @@
     @Autowired
     private MainService mainService;
     @Autowired
+    private MainLockWrapService mainLockWrapService;
+    @Autowired
     private ActionService actionService;
     @Autowired
     private SegmentService segmentService;
     @Autowired
     private TrafficService trafficService;
+    @Autowired
+    private ConfigService configService;
+    @Autowired
+    private LocService locService;
+    @Autowired
+    private AgvModelService agvModelService;
 
-    @Scheduled(cron = "0/3 * * * * ? ")
+    @Scheduled(cron = "0/1 * * * * ? ")
     private void startupBus() throws InterruptedException {
+//        if (!configService.getVal("TaskAssignMode", Boolean.class)) { return; }
         if (!this.lock.tryLock(LOCK_TIMEOUT, TimeUnit.SECONDS)) { return; }
         List<Bus> busList = busService.selectBySts(BusStsType.RECEIVE);
         for (Bus bus : busList) {
-            mainService.infuseAgvForTask(bus);
+            mainService.allocateTask(bus);
         }
         this.lock.unlock();
     }
 
-    @Scheduled(cron = "0/3 * * * * ? ")
+    @Scheduled(cron = "0/1 * * * * ? ")
     private void calculateSeg() throws InterruptedException {
         if (!this.lock.tryLock(LOCK_TIMEOUT, TimeUnit.SECONDS)) { return; }
         List<Task> taskList = taskService.selectBySts(TaskStsType.WAITING);
@@ -77,28 +98,54 @@
         }
 
         for (AgvTaskDto dto : taskDtoList) {
-            mainService.mergeMajorTask(dto.getAgvId(), dto.getTaskList());
+            mainLockWrapService.buildMajorTask(dto.getAgvId(), dto.getTaskList());
         }
         this.lock.unlock();
     }
 
-    @Scheduled(cron = "0/1 * * * * ? ")
-    private void traffic() {
-        List<Segment> segments = segmentService.list(new LambdaQueryWrapper<Segment>()
-                .eq(Segment::getState, SegmentStateType.WAITING.toString())
-        );
-        for (Segment segment : segments) {
-            trafficService.trigger(segment);
-        }
+    @EventListener(ApplicationReadyEvent.class)
+    public void init() {
+        try { Thread.sleep(1200); } catch (InterruptedException ignore) {}
+        // traffic calculate
+        this.trafficCalcThread = new Thread(() -> {
+            while (!Thread.currentThread().isInterrupted()) {
+                try {
+                    Thread.sleep(CORE_SCAN_FREQUENCY_MILLISECOND);
+
+                    List<Segment> segments = segmentService.list(new LambdaQueryWrapper<Segment>()
+                            .eq(Segment::getState, SegmentStateType.WAITING.toString())
+                    );
+                    for (Segment segment : segments) {
+                        long startTime = System.currentTimeMillis();
+                        trafficService.trigger(segment);
+//                        log.info("traffic calculation spend {} ms", System.currentTimeMillis() - startTime);
+                    }
+
+                } catch (Exception e) {
+                    log.error("KernelScheduler.trafficCalcThread fail", e);
+                }
+            }
+        });
+        this.trafficCalcThread.start();
+        // public action
+        this.actionPublicThread = new Thread(() -> {
+            while (!Thread.currentThread().isInterrupted()) {
+                try {
+                    Thread.sleep(CORE_SCAN_FREQUENCY_MILLISECOND);
+
+                    List<String> actionGroupIds = actionService.selectGroupNo(ActionStsType.PREPARE);
+                    for (String actionGroupId : actionGroupIds) {
+                        mainService.publishAction(actionGroupId);
+                    }
+
+                } catch (Exception e) {
+                    log.error("KernelScheduler.actionPublicThread fail", e);
+                }
+            }
+        });
+        this.actionPublicThread.start();
     }
 
-    @Scheduled(cron = "0/1 * * * * ? ")
-    private void publishAction(){
-        List<String> actionGroupIds = actionService.selectPrepareGroup();
-        for (String actionGroupId : actionGroupIds) {
-            mainService.publishAction(actionGroupId);
-        }
-    }
 
     // patch ----------------------------------------------------------------------------------------------
 
@@ -122,6 +169,25 @@
                     log.error("Bus [{}] 鏇存柊澶辫触 锛侊紒锛�", bus.getUuid());
                 }
             }
+            long cancelNum = taskList.stream().filter(task -> TaskStsType.CANCEL.val() == task.getTaskSts()).count();
+            if (cancelNum == taskList.size()) {
+                bus.setBusSts(BusStsType.CANCEL.val());
+                bus.setUpdateTime(new Date());
+                if (!busService.updateById(bus)) {
+                    log.error("Bus [{}] 鏇存柊澶辫触 锛侊紒锛�", bus.getUuid());
+                }
+            }
+        }
+    }
+
+
+    @PreDestroy
+    public void shutDown(){
+        if (this.trafficCalcThread != null) {
+            this.trafficCalcThread .interrupt();
+        }
+        if (this.actionPublicThread != null) {
+            this.actionPublicThread .interrupt();
         }
     }
 

--
Gitblit v1.9.1