#
luxiaotao1123
2025-01-07 e14f937fac9ba2c9af1fffd4d37f0eba86292575
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
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.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 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.concurrent.ExecutorService;
import java.util.concurrent.Executors;
 
/**
 * Created by vincent on 10/14/2024
 */
@Slf4j
@Component
public class MapDataWsScheduler {
 
    public static final int WEBSOCKET_BROADCAST_INTERVAL = 600;
 
    private ExecutorService singleThreadExecutor;
 
    @Autowired
    private AgvService agvService;
    @Autowired
    private AgvDetailService agvDetailService;
    @Autowired
    private AgvModelService agvModelService;
    @Autowired
    private CodeService codeService;
    @Autowired
    private MapDataDispatcher mapDataDispatcher;
 
    @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)
        );
        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());
            vo.setBackpack(GsonUtils.fromJsonToList(agvDetail.getBackpack(), BackpackDto.class));
            vo.setBattery(agvDetail.getSoc());
            vo.setError(agvDetail.realError());
 
            vo.setDynamicRoute(mapDataDispatcher.queryCodeListFromDynamicNode(null, agv.getUuid()));
 
            agvVos.add(vo);
        }
 
        return agvVos;
    }
 
    @PreDestroy
    public void destroy() {
        this.singleThreadExecutor.shutdownNow();
    }
 
}