import ReconnectingWebSocket from 'reconnecting-websocket';
|
|
/**
|
* ConnectWebSocket 类
|
* TODO: 优化messageHandler: EventEmitter。暂时传入回调函数
|
*/
|
|
class ConnectWebSocket {
|
_url;
|
_socket;
|
_hasInit;
|
_messageHandler;
|
|
constructor(url) {
|
this._url = url;
|
this._socket = null;
|
this._hasInit = false;
|
this._messageHandler = null;
|
}
|
|
initSocket() {
|
if (this._hasInit) {
|
return;
|
}
|
if (!this._url) {
|
return;
|
}
|
// 会自动重连，无需处理重连逻辑
|
this._socket = new ReconnectingWebSocket(this._url, [], {
|
maxReconnectionDelay: 20000, // 断开后最大的重连时间： 20s，每多一次重连，会增加 1.3 倍，5 * 1.3 * 1.3 * 1.3...
|
minReconnectionDelay: 5000, // 断开后最短的重连时间： 5s
|
maxRetries: 5,
|
});
|
this._hasInit = true;
|
|
this._socket.addEventListener('open', this._onOpen.bind(this));
|
this._socket.addEventListener('close', this._onClose.bind(this));
|
this._socket.addEventListener('error', this._onError.bind(this));
|
this._socket.addEventListener('message', this._onMessage.bind(this));
|
}
|
|
_onOpen() {
|
console.log('连接成功');
|
}
|
|
_onClose() {
|
console.log('连接已断开');
|
}
|
|
_onError() {
|
console.log('连接 error');
|
}
|
|
registerMessageHandler(messageHandler) {
|
this._messageHandler = messageHandler;
|
}
|
|
_onMessage(msg) {
|
// const data = JSON.parse(msg.data);
|
this._messageHandler && this._messageHandler(msg.data);
|
}
|
|
sendMessage = (message) => {
|
// 目前只供AI聊天使用
|
this._socket?.send(message)
|
// this._socket?.send(JSON.stringify(message.data));
|
};
|
|
close() {
|
this._socket?.close();
|
}
|
}
|
|
export default ConnectWebSocket;
|
