fix: socket connection

Signed-off-by: Innei <[email protected]>
mx-space · Feb 2, 2024 · cbb459b · cbb459b
1 parent 78515b2
commit cbb459b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/router/guard.ts b/src/router/guard.ts
@@ -53,6 +53,10 @@ router.beforeEach(async (to) => {
     if (!ok) {
       return `/login?from=${encodeURI(to.fullPath)}`
     } else {
+      import('socket').then((mo) => {
+        mo.socket.initIO()
+      })
+
       const sessionWithLogin = sessionStorage.getItem(SESSION_WITH_LOGIN)
       if (sessionWithLogin) return
       // login with token only

diff --git a/src/socket/socket-client.tsx b/src/socket/socket-client.tsx
@@ -6,9 +6,9 @@ import io from 'socket.io-client'
 import { getToken } from 'utils/auth'
 import { bus } from 'utils/event-bus'
 import { BrowserNotification } from 'utils/notification'
+import type { NotificationTypes } from './types'
 
 import { configs } from '../configs'
-import type { NotificationTypes } from './types'
 import { EventTypes } from './types'
 
 const Notification = {
@@ -45,6 +45,7 @@ export class SocketClient {
     }
     this.destory()
     const token = getToken()
+
     if (!token) {
       return
     }