From 4b5cc9f0b6d497ec9c7850afddb8000af27ad965 Mon Sep 17 00:00:00 2001
From: Sam Elie <samradelie@gmail.com>
Date: Sat, 18 Feb 2023 09:36:13 -0800
Subject: [PATCH] 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>
---
 packages/vite/src/node/server/ws.ts | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/packages/vite/src/node/server/ws.ts b/packages/vite/src/node/server/ws.ts
index 87170d11df8b39..0ccfbd8582233a 100644
--- 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))