🐛websocket自动适配SSL

ballcat-projects · Mar 5, 2024 · dd2ea04 · dd2ea04
1 parent 33185e1
commit dd2ea04
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/hooks/websocket.ts b/src/hooks/websocket.ts
@@ -12,8 +12,8 @@ const useAdminWebSocket = () => {
 
   // ws地址
   const baseUri = import.meta.env.VITE_API_URL
-  const host = window.location.host
-  const wsUri = `ws://${host}${baseUri}/ws?access_token=${accessToken}`
+  const protocol = window.location.protocol || ''
+  const wsUri = `${protocol.startsWith('https') ? 'wss' : 'ws'}://${host}${baseUri}/ws?access_token=${accessToken}`
 
   useWebSocketReturn = useWebSocket(wsUri, {
     autoReconnect: {