|
|
@ -26,18 +26,22 @@ public class WebSocketTopicListener implements ApplicationRunner, Ordered { |
|
|
|
public void run(ApplicationArguments args) throws Exception { |
|
|
|
// 订阅WebSocket消息
|
|
|
|
WebSocketUtils.subscribeMessage((message) -> { |
|
|
|
log.info("WebSocket主题订阅收到消息session keys={} message={}", message.getSessionKeys(), message.getMessage()); |
|
|
|
// 如果key不为空就按照key发消息 如果为空就群发
|
|
|
|
if (CollUtil.isNotEmpty(message.getSessionKeys())) { |
|
|
|
message.getSessionKeys().forEach(key -> { |
|
|
|
if (WebSocketSessionHolder.existSession(key)) { |
|
|
|
try { |
|
|
|
log.info("WebSocket主题订阅收到消息session keys={} message={}", message.getSessionKeys(), message.getMessage()); |
|
|
|
// 如果key不为空就按照key发消息 如果为空就群发
|
|
|
|
if (CollUtil.isNotEmpty(message.getSessionKeys())) { |
|
|
|
message.getSessionKeys().forEach(key -> { |
|
|
|
if (WebSocketSessionHolder.existSession(key)) { |
|
|
|
WebSocketUtils.sendMessage(key, message.getMessage()); |
|
|
|
} |
|
|
|
}); |
|
|
|
} else { |
|
|
|
WebSocketSessionHolder.getSessionsAll().forEach(key -> { |
|
|
|
WebSocketUtils.sendMessage(key, message.getMessage()); |
|
|
|
} |
|
|
|
}); |
|
|
|
} else { |
|
|
|
WebSocketSessionHolder.getSessionsAll().forEach(key -> { |
|
|
|
WebSocketUtils.sendMessage(key, message.getMessage()); |
|
|
|
}); |
|
|
|
}); |
|
|
|
} |
|
|
|
} catch (Exception e) { |
|
|
|
log.error("处理消息时发生异常", e); |
|
|
|
} |
|
|
|
}); |
|
|
|
log.info("初始化WebSocket主题订阅监听器成功"); |
|
|
|