From 3e28500e76ffd71b7da90f40d1f3126c00a0e5f0 Mon Sep 17 00:00:00 2001
From: Junjie <fallin.jie@qq.com>
Date: 星期一, 13 四月 2026 14:49:41 +0800
Subject: [PATCH] #站点运行优化

---
 src/main/java/com/zy/core/thread/impl/ZyStationV5Thread.java |  144 ++++++++++++++++++++++++++++++++++-------------
 1 files changed, 103 insertions(+), 41 deletions(-)

diff --git a/src/main/java/com/zy/core/thread/impl/ZyStationV5Thread.java b/src/main/java/com/zy/core/thread/impl/ZyStationV5Thread.java
index 3762364..d563265 100644
--- a/src/main/java/com/zy/core/thread/impl/ZyStationV5Thread.java
+++ b/src/main/java/com/zy/core/thread/impl/ZyStationV5Thread.java
@@ -21,6 +21,8 @@
 import com.zy.core.network.ZyStationConnectDriver;
 import com.zy.core.network.entity.ZyStationStatusEntity;
 import com.zy.core.service.StationTaskLoopService;
+import com.zy.core.task.BasStationOptAsyncPublisher;
+import com.zy.core.thread.impl.v5.StationV5RuntimeConfigProvider;
 import com.zy.core.thread.impl.v5.StationV5RunBlockReroutePlanner;
 import com.zy.core.thread.impl.v5.StationV5SegmentExecutor;
 import com.zy.core.thread.impl.v5.StationV5StatusReader;
@@ -34,35 +36,54 @@
 import java.util.List;
 import java.util.Map;
 import java.util.Objects;
-import java.util.concurrent.ExecutorService;
-import java.util.concurrent.Executors;
+import java.util.concurrent.ArrayBlockingQueue;
+import java.util.concurrent.RejectedExecutionException;
 import java.util.concurrent.ThreadPoolExecutor;
+import java.util.concurrent.TimeUnit;
 
 @Data
 @Slf4j
 public class ZyStationV5Thread implements Runnable, com.zy.core.thread.StationThread {
 
-    private static final int SEGMENT_EXECUTOR_POOL_SIZE = 64;
+    private static final int DEFAULT_SEGMENT_EXECUTOR_POOL_SIZE = 128;
+    private static final int DEFAULT_SEGMENT_EXECUTOR_QUEUE_CAPACITY = 512;
     private static final int EXECUTOR_QUEUE_WARN_THRESHOLD = 20;
     private static final int EXECUTOR_ACTIVE_WARN_THRESHOLD = 48;
     private static final long SEGMENT_EXECUTE_WARN_MS = 10_000L;
+    private static final int QUEUE_DRAIN_BATCH_SIZE = 32;
+    private static final long QUEUE_IDLE_SLEEP_MS = 20L;
 
     private DeviceConfig deviceConfig;
     private RedisUtil redisUtil;
     private ZyStationConnectDriver zyStationConnectDriver;
-    private final ExecutorService executor = Executors.newFixedThreadPool(SEGMENT_EXECUTOR_POOL_SIZE);
+    private final ThreadPoolExecutor executor;
     private StationV5SegmentExecutor segmentExecutor;
     private final RecentStationArrivalTracker recentArrivalTracker;
     private final StationV5StatusReader statusReader;
     private final StationV5RunBlockReroutePlanner runBlockReroutePlanner;
+    private final int executorQueueCapacity;
+    private final BasStationOptAsyncPublisher basStationOptAsyncPublisher;
 
     public ZyStationV5Thread(DeviceConfig deviceConfig, RedisUtil redisUtil) {
         this.deviceConfig = deviceConfig;
         this.redisUtil = redisUtil;
+        StationV5RuntimeConfigProvider configProvider = SpringUtils.getBean(StationV5RuntimeConfigProvider.class);
+        int poolSize = configProvider == null ? DEFAULT_SEGMENT_EXECUTOR_POOL_SIZE : configProvider.getSegmentExecutorPoolSize();
+        this.executorQueueCapacity = configProvider == null ? DEFAULT_SEGMENT_EXECUTOR_QUEUE_CAPACITY : configProvider.getSegmentExecutorQueueCapacity();
+        this.executor = new ThreadPoolExecutor(
+                poolSize,
+                poolSize,
+                0L,
+                TimeUnit.MILLISECONDS,
+                new ArrayBlockingQueue<>(executorQueueCapacity)
+        );
         this.recentArrivalTracker = new RecentStationArrivalTracker(redisUtil);
         this.segmentExecutor = new StationV5SegmentExecutor(deviceConfig, redisUtil, this::sendCommand);
         this.statusReader = new StationV5StatusReader(deviceConfig, redisUtil, recentArrivalTracker);
         this.runBlockReroutePlanner = new StationV5RunBlockReroutePlanner(redisUtil);
+        this.basStationOptAsyncPublisher = SpringUtils.getBean(BasStationOptAsyncPublisher.class);
+        log.info("鍒濆鍖朧5杈撻�佺嚎绋嬫睜锛宒eviceNo={}, poolSize={}, queueCapacity={}",
+                deviceConfig == null ? null : deviceConfig.getDeviceNo(), poolSize, executorQueueCapacity);
     }
 
     @Override
@@ -85,8 +106,10 @@
         Thread processThread = new Thread(() -> {
             while (true) {
                 try {
-                    pollAndDispatchQueuedCommand();
-                    Thread.sleep(100);
+                    int dispatchedCount = pollAndDispatchQueuedCommandBatch();
+                    if (dispatchedCount <= 0) {
+                        Thread.sleep(QUEUE_IDLE_SLEEP_MS);
+                    }
                 } catch (Exception e) {
                     log.error("StationV5Process Fail", e);
                 }
@@ -130,38 +153,76 @@
         return map;
     }
 
-    private void pollAndDispatchQueuedCommand() {
-        Task task = MessageQueue.poll(SlaveType.Devp, deviceConfig.getDeviceNo());
-        if (task == null || task.getStep() == null || task.getStep() != 2) {
-            return;
-        }
-        StationCommand command = (StationCommand) task.getData();
-        logExecutorAbnormal("queue-poll", command);
-        submitSegmentCommand(command);
+    @Override
+    public List<Integer> getAllTaskNoList() {
+        return statusReader.getTaskNoList();
     }
 
-    private void submitSegmentCommand(StationCommand command) {
-        if (command == null || executor == null || segmentExecutor == null) {
-            return;
-        }
-        executor.submit(() -> {
-            long start = System.currentTimeMillis();
-            try {
-                segmentExecutor.execute(command);
-            } finally {
-                long costMs = System.currentTimeMillis() - start;
-                if (costMs >= SEGMENT_EXECUTE_WARN_MS) {
-                    log.warn("V5杈撻�佸懡浠ゅ垎娈垫墽琛岃�楁椂杩囬暱锛宒eviceNo={}, taskNo={}, stationId={}, targetStaNo={}, costMs={}",
-                            deviceConfig.getDeviceNo(),
-                            command.getTaskNo(),
-                            command.getStationId(),
-                            command.getTargetStaNo(),
-                            costMs);
-                    logExecutorAbnormal("segment-finish-slow", command);
-                }
+    private int pollAndDispatchQueuedCommandBatch() {
+        int dispatchedCount = 0;
+        while (dispatchedCount < QUEUE_DRAIN_BATCH_SIZE) {
+            if (isExecutorQueueAtWatermark()) {
+                logExecutorAbnormal("executor-watermark", null);
+                break;
             }
-        });
+            Task task = MessageQueue.peek(SlaveType.Devp, deviceConfig.getDeviceNo());
+            if (task == null || task.getStep() == null || task.getStep() != 2) {
+                break;
+            }
+            StationCommand command = (StationCommand) task.getData();
+            logExecutorAbnormal("queue-peek", command);
+            if (!submitSegmentCommand(command)) {
+                logExecutorAbnormal("submit-rejected", command);
+                break;
+            }
+            MessageQueue.poll(SlaveType.Devp, deviceConfig.getDeviceNo());
+            dispatchedCount++;
+        }
+        return dispatchedCount;
+    }
+
+    private boolean submitSegmentCommand(StationCommand command) {
+        if (command == null || executor == null || segmentExecutor == null) {
+            return false;
+        }
+        try {
+            executor.execute(() -> {
+                long start = System.currentTimeMillis();
+                try {
+                    segmentExecutor.execute(command);
+                } finally {
+                    long costMs = System.currentTimeMillis() - start;
+                    if (costMs >= SEGMENT_EXECUTE_WARN_MS) {
+                        log.warn("V5杈撻�佸懡浠ゅ垎娈垫墽琛岃�楁椂杩囬暱锛宒eviceNo={}, taskNo={}, stationId={}, targetStaNo={}, costMs={}",
+                                deviceConfig.getDeviceNo(),
+                                command.getTaskNo(),
+                                command.getStationId(),
+                                command.getTargetStaNo(),
+                                costMs);
+                        logExecutorAbnormal("segment-finish-slow", command);
+                    }
+                }
+            });
+        } catch (RejectedExecutionException e) {
+            log.error("V5杈撻�佺嚎绋嬫睜鎷掔粷鎵ц锛屼繚鐣欒澶囬槦鍒楃Н鍘嬶紝deviceNo={}, taskNo={}, stationId={}, targetStaNo={}, activeCount={}, queuedCount={}, queueCapacity={}",
+                    deviceConfig.getDeviceNo(),
+                    command.getTaskNo(),
+                    command.getStationId(),
+                    command.getTargetStaNo(),
+                    executor.getActiveCount(),
+                    executor.getQueue() == null ? 0 : executor.getQueue().size(),
+                    executorQueueCapacity);
+            return false;
+        }
         logExecutorAbnormal("after-submit", command);
+        return true;
+    }
+
+    private boolean isExecutorQueueAtWatermark() {
+        if (executor == null || executor.getQueue() == null) {
+            return false;
+        }
+        return executor.getQueue().size() >= executorQueueCapacity;
     }
 
     @Override
@@ -399,18 +460,19 @@
                     commandResponse != null && Boolean.TRUE.equals(commandResponse.getResult()) ? 1 : 0,
                     JSON.toJSONString(commandResponse)
             );
-            optService.save(basStationOpt);
+            if (basStationOptAsyncPublisher == null || !basStationOptAsyncPublisher.publish(basStationOpt)) {
+                optService.save(basStationOpt);
+            }
         }
         return commandResponse;
     }
 
     private void logExecutorAbnormal(String scene, StationCommand command) {
-        if (!(executor instanceof ThreadPoolExecutor)) {
+        if (executor == null) {
             return;
         }
-        ThreadPoolExecutor threadPoolExecutor = (ThreadPoolExecutor) executor;
-        int activeCount = threadPoolExecutor.getActiveCount();
-        int queuedCount = threadPoolExecutor.getQueue() == null ? 0 : threadPoolExecutor.getQueue().size();
+        int activeCount = executor.getActiveCount();
+        int queuedCount = executor.getQueue() == null ? 0 : executor.getQueue().size();
         if (activeCount < EXECUTOR_ACTIVE_WARN_THRESHOLD && queuedCount < EXECUTOR_QUEUE_WARN_THRESHOLD) {
             return;
         }
@@ -420,10 +482,10 @@
                 command == null ? null : command.getTaskNo(),
                 command == null ? null : command.getStationId(),
                 command == null ? null : command.getTargetStaNo(),
-                threadPoolExecutor.getPoolSize(),
+                executor.getPoolSize(),
                 activeCount,
                 queuedCount,
-                threadPoolExecutor.getCompletedTaskCount());
+                executor.getCompletedTaskCount());
     }
 
     @Override

--
Gitblit v1.9.1