From 89c7f6e5bcc21b0e8f83a2bc6d680e2ffe431e6f Mon Sep 17 00:00:00 2001 From: vincentlu <t1341870251@gmail.com> Date: 星期一, 13 一月 2025 09:38:29 +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..771fe7f 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.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()); + 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