From 8d425ae26f1070043f4e81e7ebce1e6f39d5d013 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@163.com>
Date: 星期一, 18 十一月 2024 10:21:52 +0800
Subject: [PATCH] #

---
 zy-acs-manager/src/main/java/com/zy/acs/manager/core/scheduler/MapDataWsScheduler.java |   39 +++++++++++++++++++++++++++++++++------
 1 files changed, 33 insertions(+), 6 deletions(-)

diff --git a/zy-acs-manager/src/main/java/com/zy/acs/manager/core/scheduler/MapDataWsScheduler.java b/zy-acs-manager/src/main/java/com/zy/acs/manager/core/scheduler/MapDataWsScheduler.java
index ab38279..43f55bb 100644
--- a/zy-acs-manager/src/main/java/com/zy/acs/manager/core/scheduler/MapDataWsScheduler.java
+++ b/zy-acs-manager/src/main/java/com/zy/acs/manager/core/scheduler/MapDataWsScheduler.java
@@ -8,17 +8,22 @@
 import com.zy.acs.manager.core.service.astart.MapDataDispatcher;
 import com.zy.acs.manager.manager.entity.Agv;
 import com.zy.acs.manager.manager.entity.AgvDetail;
+import com.zy.acs.manager.manager.entity.AgvModel;
 import com.zy.acs.manager.manager.entity.Code;
 import com.zy.acs.manager.manager.service.AgvDetailService;
+import com.zy.acs.manager.manager.service.AgvModelService;
 import com.zy.acs.manager.manager.service.AgvService;
 import com.zy.acs.manager.manager.service.CodeService;
 import com.zy.acs.manager.manager.websocket.MapWebSocket;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.scheduling.annotation.Scheduled;
 import org.springframework.stereotype.Component;
 
+import javax.annotation.PostConstruct;
+import javax.annotation.PreDestroy;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
 
 /**
  * Created by vincent on 10/14/2024
@@ -26,20 +31,34 @@
 @Component
 public class MapDataWsScheduler {
 
+    public static final int WEBSOCKET_BROADCAST_INTERVAL = 200;
+
+    private ExecutorService singleThreadExecutor;
+
     @Autowired
     private AgvService agvService;
     @Autowired
     private AgvDetailService agvDetailService;
     @Autowired
+    private AgvModelService agvModelService;
+    @Autowired
     private CodeService codeService;
     @Autowired
     private MapDataDispatcher mapDataDispatcher;
 
-    @Scheduled(cron = "0/1 * * * * ? ")
-    public void broadcast() {
-        MapWsVo mapWsVo = new MapWsVo();
-        mapWsVo.setAgvVos(syncAgv());
-        MapWebSocket.broadcast(GsonUtils.toJson(mapWsVo));
+    @PostConstruct
+    public void init() {
+        this.singleThreadExecutor = Executors.newSingleThreadExecutor();
+        this.singleThreadExecutor.execute(() -> {
+            try { Thread.sleep(200); } catch (InterruptedException ignore) {}
+            while (!Thread.currentThread().isInterrupted()) {
+
+                MapWsVo mapWsVo = new MapWsVo();
+                mapWsVo.setAgvVos(syncAgv());
+                MapWebSocket.broadcast(GsonUtils.toJson(mapWsVo));
+                try { Thread.sleep(WEBSOCKET_BROADCAST_INTERVAL); } catch (InterruptedException ignore) {}
+            }
+        });
     }
 
     private List<MapWsAgvVo> syncAgv() {
@@ -50,12 +69,15 @@
         );
         for (Agv agv : agvList) {
             AgvDetail agvDetail = agvDetailService.selectByAgvId(agv.getId());
+            AgvModel agvModel = agvModelService.getById(agv.getAgvModel());
             if (null == agvDetail) { continue; }
             Long recentCode = agvDetail.getRecentCode();
             if (null == recentCode) { continue; }
             Code code = codeService.getById(recentCode);
+            if (null == code) { continue; }
 
             MapWsAgvVo vo = new MapWsAgvVo();
+            vo.setAgvModel(agvModel.getType());
             vo.setAgvNo(agv.getUuid());
             vo.setCode(code.getData());
             vo.setDirection(agvDetail.getAgvAngle());
@@ -70,4 +92,9 @@
         return agvVos;
     }
 
+    @PreDestroy
+    public void destroy() {
+        this.singleThreadExecutor.shutdownNow();
+    }
+
 }

--
Gitblit v1.9.1