From 87e1f5838263d53d8ccf856926ff0a627d763b4c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=B0=8F=E6=B7=BC=E7=84=B1=5Frawbin?= <fcrshjx44@gmail.com>
Date: Mon, 8 May 2023 22:12:42 +0800
Subject: [PATCH] fix: avoid dev-server crash when ws proxy error (#12829)

---
 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 }]
   })