103 lines
2.5 KiB
TypeScript
103 lines
2.5 KiB
TypeScript
/*
|
||
* @Date: 2024-04-15 17:48:09
|
||
* @LastEditors: jiangchao
|
||
* @LastEditTime: 2024-04-15 17:48:13
|
||
* @Description: 文件注释
|
||
* @Author: jiangchao
|
||
* @FilePath: \mars3d-vue-project\src\api\websocket\index.ts
|
||
*/
|
||
import { message } from 'ant-design-vue'
|
||
import ReconnectingWebSocket from 'reconnecting-websocket'
|
||
import EventBus from '../im/eventBus.js'
|
||
|
||
|
||
interface WebSocketOptions {
|
||
data: any
|
||
cache?: boolean | string
|
||
destroyCache?: string
|
||
}
|
||
|
||
export interface MessageHandler {
|
||
(data : {[key: string]: any}): void
|
||
}
|
||
|
||
/**
|
||
* ConnectWebSocket 类
|
||
* TODO: 优化messageHandler: EventEmitter。暂时传入回调函数
|
||
*/
|
||
class ConnectWebSocket {
|
||
_url: string
|
||
_socket: ReconnectingWebSocket | null
|
||
_hasInit: boolean
|
||
_messageHandler:
|
||
| null
|
||
|
||
constructor (url: string) {
|
||
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('连接成功')
|
||
// message.success("ws链接成功")
|
||
}
|
||
|
||
_onClose () {
|
||
// message.error("连接已断开")
|
||
localStorage.removeItem('index-otherEquipmentList')
|
||
console.log('连接已断开')
|
||
}
|
||
|
||
_onError (err:any) {
|
||
console.info("err",err)
|
||
// message.error("ws连接错误，请检查网络是否通畅")
|
||
}
|
||
|
||
registerMessageHandler (messageHandler: MessageHandler) {
|
||
this._messageHandler = messageHandler
|
||
}
|
||
|
||
_onMessage (msg: MessageEvent) {
|
||
|
||
const data = JSON.parse(msg.data)
|
||
|
||
EventBus.emit('WebScoket-Message',data)
|
||
this._messageHandler && this._messageHandler(data)
|
||
}
|
||
|
||
sendMessage = (message: WebSocketOptions): void => {
|
||
this._socket?.send(JSON.stringify(message.data))
|
||
}
|
||
|
||
close () {
|
||
this._socket?.close()
|
||
}
|
||
}
|
||
|
||
export default ConnectWebSocket
|