fix: catch and handle websocket error (#11991) (#12007)

Co-authored-by: Samuel Elie <selie@anduril.com> Co-authored-by: Bjorn Lu <bjornlu.dev@gmail.com>
vitejs · Feb 18, 2023 · 4b5cc9f · 4b5cc9f
1 parent a0702a1
commit 4b5cc9f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/packages/vite/src/node/server/ws.ts b/packages/vite/src/node/server/ws.ts
@@ -152,6 +152,12 @@ export function createWebSocketServer(
       const client = getSocketClient(socket)
       listeners.forEach((listener) => listener(parsed.data, client))
     })
+    socket.on('error', (err) => {
+      config.logger.error(`${colors.red(`ws error:`)}\n${err.stack}`, {
+        timestamp: true,
+        error: err,
+      })
+    })
     socket.send(JSON.stringify({ type: 'connected' }))
     if (bufferedError) {
       socket.send(JSON.stringify(bufferedError))