package com.ruoyi.common.websocket.service;
|
|
import com.ruoyi.common.utils.SecurityUtils;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.stereotype.Component;
|
import org.springframework.stereotype.Service;
|
|
import javax.websocket.*;
|
import javax.websocket.server.PathParam;
|
import javax.websocket.server.ServerEndpoint;
|
import java.io.IOException;
|
|
import static com.ruoyi.common.websocket.util.WebSocketUtils.*;
|
|
|
/**
|
* @ClassName ChatServerEndpoint
|
* @Description:
|
* @Author 刘苏义
|
* @Date 2023/1/27 7:42
|
* @Version 1.0
|
*/
|
|
@Component
|
@Slf4j(topic = "websocket")
|
@ServerEndpoint("/websocket/{userId}")
|
public class ChatServerEndpoint {
|
|
@OnOpen
|
public void openSession( @PathParam("userId")String userId, Session session) {
|
|
ONLINE_USER_SESSIONS.put(userId, session);
|
String message = "用户[" + userId + "] 成功连接！";
|
log.info("用户登录："+message);
|
sendMessage(session,message);
|
}
|
|
@OnMessage
|
public void onMessage(@PathParam("userId")String userId, String message) {
|
log.info("收到消息："+message);
|
Session session = ONLINE_USER_SESSIONS.get(userId);
|
sendMessage(session,message);
|
// sendMessageAll("用户[" + userid + "] : " + message);
|
}
|
|
@OnClose
|
public void onClose( @PathParam("userId")String userId,Session session) {
|
//当前的Session 移除
|
ONLINE_USER_SESSIONS.remove(userId);
|
//并且通知其他人当前用户已经离开聊天室了
|
String message="用户[" + userId + "] 断开连接！";
|
sendMessage(session,message);
|
log.info("发送消息："+message);
|
try {
|
session.close();
|
} catch (IOException e) {
|
log.error("onClose error",e);
|
}
|
}
|
|
@OnError
|
public void onError(Session session, Throwable throwable) {
|
try {
|
session.close();
|
} catch (IOException e) {
|
log.error("onError excepiton",e);
|
}
|
log.info("Throwable msg "+throwable.getMessage());
|
}
|
}
|
