From fd6d26f8ffe6a37533e741337f3eca9134d934c7 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@163.com>
Date: 星期三, 15 一月 2025 10:42:59 +0800
Subject: [PATCH] #

---
 zy-acs-manager/src/main/java/com/zy/acs/manager/core/scheduler/MapDataWsScheduler.java |  101 ++++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 96 insertions(+), 5 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 7aadde7..860ee60 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
@@ -1,21 +1,112 @@
 package com.zy.acs.manager.core.scheduler;
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.zy.acs.common.utils.GsonUtils;
+import com.zy.acs.manager.core.domain.BackpackDto;
+import com.zy.acs.manager.core.domain.MapWsAgvVo;
 import com.zy.acs.manager.core.domain.MapWsVo;
+import com.zy.acs.manager.core.service.MapService;
+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.scheduling.annotation.Scheduled;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
+
+import javax.annotation.PostConstruct;
+import javax.annotation.PreDestroy;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
+import java.util.stream.Collectors;
 
 /**
  * Created by vincent on 10/14/2024
  */
+@Slf4j
 @Component
 public class MapDataWsScheduler {
 
-    @Scheduled(cron = "0/1 * * * * ? ")
-    public void broadcast() {
-        MapWsVo mapWsVo = new MapWsVo();
+    public static final int WEBSOCKET_BROADCAST_INTERVAL = 300;
 
-        MapWebSocket.broadcast("123");
+    private ExecutorService singleThreadExecutor;
+
+    @Autowired
+    private AgvService agvService;
+    @Autowired
+    private AgvDetailService agvDetailService;
+    @Autowired
+    private AgvModelService agvModelService;
+    @Autowired
+    private CodeService codeService;
+    @Autowired
+    private MapService mapService;
+
+    @PostConstruct
+    public void init() {
+        this.singleThreadExecutor = Executors.newSingleThreadExecutor();
+        this.singleThreadExecutor.execute(() -> {
+            try { Thread.sleep(200); } catch (InterruptedException ignore) {}
+            while (!Thread.currentThread().isInterrupted()) {
+                try {
+                    MapWsVo mapWsVo = new MapWsVo();
+                    mapWsVo.setAgvVos(syncAgv());
+                    MapWebSocket.broadcast(GsonUtils.toJson(mapWsVo));
+                    Thread.sleep(WEBSOCKET_BROADCAST_INTERVAL / 3);
+                } catch (Exception e) {
+                    log.error("MapDataWsScheduler.init", e);
+                }
+            }
+        });
+    }
+
+    private List<MapWsAgvVo> syncAgv() {
+        List<MapWsAgvVo> agvVos = new ArrayList<>();
+
+        List<Agv> agvList = agvService.list(new LambdaQueryWrapper<Agv>()
+//                .eq(Agv::getStatus, 1)
+        );
+
+        Map<String, List<String>> codeDataListMap = mapService.queryCodeListFromDynamicNode(null
+                , agvList.stream().map(Agv::getUuid).collect(Collectors.toSet()));
+
+        for (Agv agv : agvList) {
+            AgvDetail agvDetail = agvDetailService.selectByAgvId(agv.getId());
+            AgvModel agvModel = agvModelService.getByAgvId(agv.getId());
+            if (null == agvDetail) { continue; }
+            Long recentCode = agvDetail.getRecentCode();
+            if (null == recentCode) { continue; }
+            Code code = codeService.getCacheById(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());
+            vo.setBackpack(GsonUtils.fromJsonToList(agvDetail.getBackpack(), BackpackDto.class));
+            vo.setBattery(agvDetail.getSoc());
+            vo.setError(agvDetail.realError());
+
+            vo.setDynamicRoute(codeDataListMap.get(agv.getUuid()));
+
+            agvVos.add(vo);
+        }
+
+        return agvVos;
+    }
+
+    @PreDestroy
+    public void destroy() {
+        this.singleThreadExecutor.shutdownNow();
     }
 
 }

--
Gitblit v1.9.1