From 4dccbb68399da2d46943b8d613585afb49aa5455 Mon Sep 17 00:00:00 2001
From: sean.zhou <sean.zhou@dji.com>
Date: Fri, 22 Jul 2022 20:48:54 +0800
Subject: [PATCH] V1.1.0 for dock
---
src/api/websocket.ts | 21 ++++++++++++---------
1 files changed, 12 insertions(+), 9 deletions(-)
diff --git a/src/api/websocket.ts b/src/api/websocket.ts
index 4613457..0d5a08f 100644
--- a/src/api/websocket.ts
+++ b/src/api/websocket.ts
@@ -1,14 +1,14 @@
import ReconnectingWebSocket from 'reconnecting-websocket'
+import { ELocalStorageKey } from '../types/enums'
import { CURRENT_CONFIG as config } from '/@/api/http/config'
-let socket = {}
+let socket: ReconnectingWebSocket
export default {
- init (getMsgFunc) {
- const token = localStorage.getItem('x-auth-token')
- const wspath =
- config.websocketURL + '?x-auth-token=' + escape(token)
- socket = new ReconnectingWebSocket(wspath)
+ init (getMsgFunc: any) {
+ const token: string = localStorage.getItem(ELocalStorageKey.Token)!
+ const wspath = config.websocketURL + '?x-auth-token=' + encodeURI(token)
+ socket = new ReconnectingWebSocket(wspath, '', { maxRetries: 5 })
socket.onopen = this.onOpen
socket.onerror = this.onError
socket.onmessage = getMsgFunc
@@ -18,13 +18,16 @@
onOpen () {
console.log('ws opened')
},
- onError (err) {
+ onError (err: any) {
console.error(err)
},
onClose () {
console.log('ws closed')
},
- sendMsg (data) {
- this.socket.send(data)
+ sendMsg (data: any) {
+ socket.send(data)
+ },
+ close () {
+ socket.close()
}
}
--
Gitblit v1.9.3