opensource/RuoYi-Vue-Plus
2
0
Fork 2
mirror of https://gitee.com/dromara/RuoYi-Vue-Plus.git synced 2026-09-21 18:45:57 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add 增加 ruoyi-websocket 模块

Browse Source
This commit is contained in:
zendwang 2023-03-29 18:19:22 +08:00
parent 754b78cb18
commit aede0ce844
3 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ruoyi-common/ruoyi-common-websocket/src/main/java/com/ruoyi/common/websocket/dto/WebSocketMessageDto.java
View File

@ -2,7 +2,6 @@ package com.ruoyi.common.websocket.dto;
import lombok.Builder;
import lombok.Data;
import lombok.experimental.Accessors;
import java.io.Serializable;
import java.util.List;
@ -12,8 +11,8 @@ import java.util.List;
*
* @author zendwang
*/
@Builder
@Data
@Accessors(chain = true)
public class WebSocketMessageDto implements Serializable {
private static final long serialVersionUID = 1L;
@ -27,8 +26,4 @@ public class WebSocketMessageDto implements Serializable {
* 需要发送的消息
*/
private String message;
public static WebSocketMessageDto build(List<Long> sessionKeys, String message) {
return new WebSocketMessageDto().setMessage(message).setSessionKeys(sessionKeys);
}
}

3
ruoyi-common/ruoyi-common-websocket/src/main/java/com/ruoyi/common/websocket/handler/PlusWebSocketHandler.java
View File

@ -44,7 +44,8 @@ public class PlusWebSocketHandler extends AbstractWebSocketHandler {
LoginUser loginUser = (LoginUser) session.getAttributes().get(LOGIN_USER_KEY);
log.info("PlusWebSocketHandler, 连接：" + session.getId() + "，已收到消息:" + message.getPayload());
List<Long> userIds = List.of(loginUser.getUserId());
WebSocketMessageDto webSocketMessageDto = WebSocketMessageDto.build(userIds, message.getPayload());
WebSocketMessageDto webSocketMessageDto = WebSocketMessageDto.builder()
.sessionKeys(userIds).message(message.getPayload()).build();
WebSocketUtils.publishMessage(webSocketMessageDto);
}

3
ruoyi-common/ruoyi-common-websocket/src/main/java/com/ruoyi/common/websocket/utils/WebSocketUtils.java
View File

@ -68,7 +68,8 @@ public class WebSocketUtils {
}
// 不在当前服务内session,发布订阅消息
if (CollUtil.isNotEmpty(unsentSessionKeys)) {
WebSocketMessageDto broadcastMessage = WebSocketMessageDto.build(unsentSessionKeys, webSocketMessage.getMessage());
WebSocketMessageDto broadcastMessage = WebSocketMessageDto.builder()
.message(webSocketMessage.getMessage()).sessionKeys(unsentSessionKeys).build();
RedisUtils.publish(WEB_SOCKET_TOPIC, broadcastMessage, consumer -> {
log.info(" WebSocket发送主题订阅消息topic:{} session keys:{} message:{}",
WEB_SOCKET_TOPIC, unsentSessionKeys, webSocketMessage.getMessage());