82 lines
1.7 KiB
JavaScript
82 lines
1.7 KiB
JavaScript
import { WS_URL } from "./config";
|
|
|
|
let ws = null;
|
|
|
|
// 建立WebSocket连接
|
|
export function connectWebSocket(token, onMessage, onOpen, onClose, onError) {
|
|
// 如果已经连接，就不重复连接
|
|
if (ws && ws.readyState === WebSocket.OPEN) {
|
|
console.log("WebSocket已经连接");
|
|
return;
|
|
}
|
|
|
|
// 携带JWT token
|
|
ws = new WebSocket(`${WS_URL}?token=${token}`);
|
|
|
|
// 连接成功
|
|
ws.onopen = () => {
|
|
console.log("WebSocket连接成功");
|
|
|
|
if (onOpen) {
|
|
onOpen();
|
|
}
|
|
};
|
|
|
|
// 接收到服务器消息
|
|
ws.onmessage = (event) => {
|
|
try {
|
|
const data = JSON.parse(event.data);
|
|
|
|
console.log("收到WebSocket消息：", data);
|
|
|
|
if (onMessage) {
|
|
onMessage(data);
|
|
}
|
|
} catch (error) {
|
|
console.error("WebSocket消息解析失败：", error);
|
|
}
|
|
};
|
|
|
|
// 连接关闭
|
|
ws.onclose = () => {
|
|
console.log("WebSocket连接关闭");
|
|
|
|
if (onClose) {
|
|
onClose();
|
|
}
|
|
|
|
ws = null;
|
|
};
|
|
|
|
// 连接错误
|
|
ws.onerror = (error) => {
|
|
console.error("WebSocket连接错误：", error);
|
|
|
|
if (onError) {
|
|
onError(error);
|
|
}
|
|
};
|
|
}
|
|
|
|
// 发送消息
|
|
export function sendWebSocketMessage(data) {
|
|
if (!ws || ws.readyState !== WebSocket.OPEN) {
|
|
console.error("WebSocket没有连接");
|
|
return;
|
|
}
|
|
|
|
ws.send(JSON.stringify(data));
|
|
}
|
|
|
|
// 关闭WebSocket
|
|
export function closeWebSocket() {
|
|
if (ws) {
|
|
ws.close();
|
|
ws = null;
|
|
}
|
|
}
|
|
|
|
// 获取当前WebSocket
|
|
export function getWebSocket() {
|
|
return ws;
|
|
} |