From 8f9cdf80238df36337ca1f6469d7b542d2e1c2e2 Mon Sep 17 00:00:00 2001
From: "rawbin.zhang" <rawbin.zhang@lunarax.io>
Date: Tue, 11 Apr 2023 21:00:02 +0800
Subject: [PATCH] fix: issue #12664 avoid dev-server crash when ws proxy error

---
 packages/vite/src/node/server/middlewares/proxy.ts | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/packages/vite/src/node/server/middlewares/proxy.ts b/packages/vite/src/node/server/middlewares/proxy.ts
index f07b93e0746223..66f99f94f0609d 100644
--- a/packages/vite/src/node/server/middlewares/proxy.ts
+++ b/packages/vite/src/node/server/middlewares/proxy.ts
@@ -79,6 +79,19 @@ export function proxyMiddleware(
         res.end()
       }
     })
+
+    proxy.on('proxyReqWs', (proxyReq, req, socket, options, head) => {
+      socket.on('error', (err) => {
+        config.logger.error(
+          `${colors.red(`ws proxy socket error:`)}\n${err.stack}`,
+          {
+            timestamp: true,
+            error: err,
+          },
+        )
+      })
+    })
+
     // clone before saving because http-proxy mutates the options
     proxies[context] = [proxy, { ...opts }]
   })