joylink/rt-sim-training-service
6
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases 10 Wiki Activity

增加订阅设备状态时发送所有设备状态逻辑

Browse Source
This commit is contained in:
joylink_zhangsai 2022-11-08 10:16:52 +08:00
parent b4f7b305b6
commit e379a0a95f
2 changed files with 21 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
src/main/java/club/joylink/rtss/simulation/cbtc/ATS/ATSMessageCollectAndDispatcher.java
View File

@ -127,37 +127,6 @@ public class ATSMessageCollectAndDispatcher {
return totalList;
}
public void collectDeviceStatusAndSend(Simulation simulation, List<? extends MapElement> deviceList) {
if (!CollectionUtils.isEmpty(deviceList)) {
List<Object> filterList = new ArrayList<>();
SimulationDataRepository repository = simulation.getRepository();
deviceList.forEach(device -> {
DeviceStatus deviceStatus = repository.findStatusByCode(device.getCode());
if (Objects.isNull(deviceStatus)) {
deviceStatus = buildDeviceStatus(device);
if (Objects.nonNull(deviceStatus)) {
repository.addDeviceInfo(deviceStatus);
filterList.add(deviceStatus);
}
} else {
DeviceStatusVO statusVO = buildDeviceStatusVO(device);
if (deviceStatus.compareAndChange(device, statusVO)) {
filterList.add(statusVO);
}
}
});
// 构建并发送设备状态消息
if (CollectionUtils.isEmpty(filterList)) {
return;
}
SocketMessageVO deviceStatusMessage =
SocketMessageFactory.build(WebSocketMessageType.Simulation_DeviceStatus,
simulation.getId(), filterList);
Set<String> sessions = simulation.getSimulationUserIds();
stompMessageService.sendToUser(sessions, deviceStatusMessage);
}
}
private DeviceStatus buildDeviceStatus(MapElement device) {
DeviceStatus status = null;
switch (device.getDeviceType()) {

29
src/main/java/club/joylink/rtss/simulation/cbtc/message/SimulationUserWsListener.java
View File

@ -97,6 +97,10 @@ public class SimulationUserWsListener {
handleSubscribeSimulationMain(simulation, simulationUser);
break;
}
case DeviceStatus: {
handleSubscribeSimulationDeviceStatus(simulation, userId);
break;
}
case SandBox: {
applicationContext.publishEvent(new SandboxUserConnectEvent(userId, subscribeTopic.getId(destination), this));
break;
@ -114,6 +118,12 @@ public class SimulationUserWsListener {
ctcLogicLoop.sendMessageWhenSubscribeCtc(simulation, userId);
break;
}
// case Room:
// break;
// case Wgu3d:
// break;
// case YJDDZH_TRAIN_POSITION:
// break;
}
}
@ -123,14 +133,6 @@ public class SimulationUserWsListener {
private void handleSubscribeSimulationMain(club.joylink.rtss.simulation.cbtc.Simulation simulation, SimulationUser simulationUser) {
String group = simulation.getId();
String userId = simulationUser.getId();
// 将当前所有设备状态同步给用户
List<DeviceStatus> statusList = simulation.getRepository().getAllStatusList();
if (!CollectionUtils.isEmpty(statusList)) {
SocketMessageVO allStatusMessage =
SocketMessageFactory.build(WebSocketMessageType.Simulation_DeviceStatus,
group, statusList);
this.stompMessageService.sendToUser(userId, allStatusMessage);
}
// 将当前所有列车实际运行数据同步给用户
List<RealRun> realRunRecordList = simulation.getRepository().getRealRunRecordList();
if (!CollectionUtils.isEmpty(realRunRecordList)) {
@ -159,6 +161,17 @@ public class SimulationUserWsListener {
this.stompMessageService.sendToUser(userId, message);
}
private void handleSubscribeSimulationDeviceStatus(club.joylink.rtss.simulation.cbtc.Simulation simulation, String userId) {
// 将当前所有设备状态同步给用户
List<DeviceStatus> statusList = simulation.getRepository().getAllStatusList();
if (!CollectionUtils.isEmpty(statusList)) {
SocketMessageVO allStatusMessage =
SocketMessageFactory.build(WebSocketMessageType.Simulation_DeviceStatus,
simulation.getId(), statusList);
this.stompMessageService.sendToUser(userId, allStatusMessage);
}
}
@Async("nsExecutor")
@EventListener
public void handle(SimulationUserUnsubscribeEvent event) {