From e1699c4851f6ca397cd0ad1ff63f32c737654836 Mon Sep 17 00:00:00 2001
From: 罗广辉 <guanghui.luo@foxmail.com>
Date: Wed, 02 Apr 2025 19:14:54 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'
---
src/websocket/index.js | 74 +++++++++++++++++++++++++++++++++++++
1 files changed, 74 insertions(+), 0 deletions(-)
diff --git a/src/websocket/index.js b/src/websocket/index.js
new file mode 100644
index 0000000..da0a2e0
--- /dev/null
+++ b/src/websocket/index.js
@@ -0,0 +1,74 @@
+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
--
Gitblit v1.9.3