feat(webui): support ws reconnect

koishijs · Apr 12, 2021 · b0df6d8 · b0df6d8
1 parent 2b6cc3b
commit b0df6d8
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 14 deletions.
diff --git a/packages/plugin-eval/src/worker/index.ts b/packages/plugin-eval/src/worker/index.ts
@@ -79,7 +79,7 @@ export function formatError(error: Error) {
     .join('\n')
 }
 
-const main = wrap<MainHandle>(parentPort)
+export const main = wrap<MainHandle>(parentPort)
 
 export interface SessionData {
   id: string

diff --git a/packages/plugin-webui/client/main.ts b/packages/plugin-webui/client/main.ts
@@ -95,23 +95,32 @@ receive('expire', () => {
   router.push('/login')
 })
 
-const endpoint = new URL(KOISHI_CONFIG.endpoint, location.origin).toString()
-const socket = client.socket.value = new WebSocket(endpoint.replace(/^http/, 'ws'))
-
-socket.onmessage = (ev) => {
-  const data = JSON.parse(ev.data)
-  console.debug(data)
-  if (data.type in client.listeners) {
-    client.listeners[data.type](data.body)
+function connect() {
+  const endpoint = new URL(KOISHI_CONFIG.endpoint, location.origin).toString()
+  const socket = client.socket.value = new WebSocket(endpoint.replace(/^http/, 'ws'))
+
+  socket.onmessage = (ev) => {
+    const data = JSON.parse(ev.data)
+    console.debug(data)
+    if (data.type in client.listeners) {
+      client.listeners[data.type](data.body)
+    }
+  }
+
+  socket.onopen = () => {
+    if (!client.user.value) return
+    const { id, token } = client.user.value
+    client.send('validate', { id, token })
   }
-}
 
-socket.onopen = () => {
-  if (!client.user.value) return
-  const { id, token } = client.user.value
-  client.send('validate', { id, token })
+  socket.onclose = () => {
+    console.log('[koishi] websocket disconnected, will retry in 1s...')
+    setTimeout(connect, 1000)
+  }
 }
 
+connect()
+
 const loadingExtensions = Promise.all(KOISHI_CONFIG.extensions.map(path => {
   return import(/* @vite-ignore */ path)
 }))