|
@ -48,7 +48,12 @@ public class PlusWebSocketHandler extends AbstractWebSocketHandler { |
|
|
protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception { |
|
|
protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception { |
|
|
// 从WebSocket会话中获取登录用户信息
|
|
|
// 从WebSocket会话中获取登录用户信息
|
|
|
LoginUser loginUser = (LoginUser) session.getAttributes().get(LOGIN_USER_KEY); |
|
|
LoginUser loginUser = (LoginUser) session.getAttributes().get(LOGIN_USER_KEY); |
|
|
|
|
|
// 检查loginUser是否为空
|
|
|
|
|
|
if (ObjectUtil.isNull(loginUser)) { |
|
|
|
|
|
log.error("[handleTextMessage] loginUser is null. sessionId: {}", session.getId()); |
|
|
|
|
|
session.close(CloseStatus.BAD_DATA); |
|
|
|
|
|
return; |
|
|
|
|
|
} |
|
|
// 创建WebSocket消息DTO对象
|
|
|
// 创建WebSocket消息DTO对象
|
|
|
WebSocketMessageDto webSocketMessageDto = new WebSocketMessageDto(); |
|
|
WebSocketMessageDto webSocketMessageDto = new WebSocketMessageDto(); |
|
|
webSocketMessageDto.setSessionKeys(List.of(loginUser.getUserId())); |
|
|
webSocketMessageDto.setSessionKeys(List.of(loginUser.getUserId())); |
|
@ -89,6 +94,13 @@ public class PlusWebSocketHandler extends AbstractWebSocketHandler { |
|
|
*/ |
|
|
*/ |
|
|
@Override |
|
|
@Override |
|
|
public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception { |
|
|
public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception { |
|
|
|
|
|
LoginUser loginUser = (LoginUser) session.getAttributes().get(LOGIN_USER_KEY); |
|
|
|
|
|
// 检查loginUser是否为空
|
|
|
|
|
|
if (ObjectUtil.isNull(loginUser)) { |
|
|
|
|
|
log.error("[handleTextMessage] loginUser is null. sessionId: {}", session.getId()); |
|
|
|
|
|
session.close(CloseStatus.BAD_DATA); |
|
|
|
|
|
return; |
|
|
|
|
|
} |
|
|
log.error("[transport error] sessionId: {} , exception:{}", session.getId(), exception.getMessage()); |
|
|
log.error("[transport error] sessionId: {} , exception:{}", session.getId(), exception.getMessage()); |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|