优化websocket监听消息
This commit is contained in:
+2
-7
@@ -24,27 +24,22 @@ function initWebSocket(server) {
|
||||
|
||||
if (!token) {
|
||||
console.log("没有提供token");
|
||||
|
||||
ws.close();
|
||||
return;
|
||||
}
|
||||
|
||||
// 验证JWT
|
||||
let user;
|
||||
|
||||
try {
|
||||
user = verifyToken(token);
|
||||
} catch (error) {
|
||||
console.log("JWT验证失败:", error.message);
|
||||
|
||||
ws.close();
|
||||
return;
|
||||
}
|
||||
|
||||
console.log("WebSocket用户:", user);
|
||||
|
||||
const username = user.username;
|
||||
|
||||
// 保存连接
|
||||
addConnection(username, ws);
|
||||
|
||||
@@ -52,7 +47,6 @@ function initWebSocket(server) {
|
||||
ws.on("message", (message) => {
|
||||
try {
|
||||
const data = JSON.parse(message.toString());
|
||||
|
||||
console.log("收到WebSocket消息:", data);
|
||||
|
||||
handleMessage(username, data);
|
||||
@@ -93,6 +87,7 @@ function handleMessage(username, data) {
|
||||
function handleSendMessage(senderName, data) {
|
||||
const receiverName = data.receiverName;
|
||||
const content = data.content;
|
||||
const createdAt = data.createdAt;
|
||||
|
||||
console.log(
|
||||
`${senderName} → ${receiverName}:${content}`
|
||||
@@ -100,7 +95,6 @@ function handleSendMessage(senderName, data) {
|
||||
|
||||
// 找到接收者的WebSocket
|
||||
const receiverWs = getConnection(receiverName);
|
||||
|
||||
if (!receiverWs) {
|
||||
console.log(`${receiverName} 当前不在线`);
|
||||
return;
|
||||
@@ -114,6 +108,7 @@ function handleSendMessage(senderName, data) {
|
||||
senderName,
|
||||
receiverName,
|
||||
content,
|
||||
createdAt
|
||||
},
|
||||
})
|
||||
);
|
||||
|
||||
Reference in New Issue
Block a user