103 lines
2.3 KiB
TypeScript
103 lines
2.3 KiB
TypeScript
// WebSocket连接配置
|
||
type WebSocketOptions = {
|
||
fullUrl: string;
|
||
onMessage?: (data: string) => void;
|
||
onOpen?: () => void;
|
||
onError?: (error: Event) => void;
|
||
onClose?: () => void;
|
||
reconnectInterval?: number;
|
||
};
|
||
|
||
let websocket: WebSocket | null = null;
|
||
let reconnectTimer: any = null;
|
||
let isManualClose = false;
|
||
|
||
// 初始化WebSocket连接
|
||
export function initWebSocket(options: WebSocketOptions) {
|
||
const {
|
||
fullUrl,
|
||
onMessage,
|
||
onOpen,
|
||
onError,
|
||
onClose,
|
||
reconnectInterval = 3000
|
||
} = options;
|
||
|
||
// 关闭旧连接
|
||
if (websocket) {
|
||
websocket.close();
|
||
}
|
||
|
||
// 检查浏览器支持性
|
||
if (!('WebSocket' in window)) {
|
||
console.error('当前浏览器不支持WebSocket');
|
||
return;
|
||
}
|
||
|
||
// 创建WebSocket连接
|
||
try {
|
||
websocket = new WebSocket(fullUrl);
|
||
|
||
// 连接成功回调
|
||
websocket.onopen = () => {
|
||
clearTimeout(reconnectTimer!);
|
||
isManualClose = false;
|
||
onOpen?.();
|
||
};
|
||
|
||
// 接收消息回调
|
||
websocket.onmessage = (event) => {
|
||
onMessage?.(event.data);
|
||
};
|
||
|
||
// 错误回调
|
||
websocket.onerror = (error) => {
|
||
onError?.(error);
|
||
startReconnect(() => initWebSocket(options), reconnectInterval);
|
||
};
|
||
|
||
// 关闭回调
|
||
websocket.onclose = () => {
|
||
onClose?.();
|
||
if (!isManualClose) {
|
||
startReconnect(() => initWebSocket(options), reconnectInterval);
|
||
}
|
||
};
|
||
} catch (error) {
|
||
console.error('WebSocket连接创建失败:', error);
|
||
startReconnect(() => initWebSocket(options), reconnectInterval);
|
||
}
|
||
}
|
||
|
||
// 发送消息
|
||
export function sendWebSocketMessage(message: any): boolean {
|
||
if (websocket && websocket.readyState === WebSocket.OPEN) {
|
||
websocket.send(JSON.stringify(message));
|
||
return true;
|
||
}
|
||
console.error('WebSocket未连接，无法发送消息');
|
||
return false;
|
||
}
|
||
|
||
// 手动关闭WebSocket
|
||
export function closeWebSocket(): void {
|
||
isManualClose = true;
|
||
if (reconnectTimer) {
|
||
clearTimeout(reconnectTimer);
|
||
}
|
||
websocket?.close();
|
||
websocket = null;
|
||
}
|
||
|
||
// 启动重连
|
||
function startReconnect(connectCallback: () => void, interval: number): void {
|
||
if (reconnectTimer) {
|
||
clearTimeout(reconnectTimer);
|
||
}
|
||
reconnectTimer = setTimeout(connectCallback, interval);
|
||
}
|
||
|
||
// 获取当前连接状态
|
||
export function getWebSocketState(): number | null {
|
||
return websocket?.readyState || null;
|
||
} |