fix(client): detect ws close correctly (#18548)

vitejs · Nov 1, 2024 · 637d31b · 637d31b
1 parent 1ed03c6
commit 637d31b
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/packages/vite/src/client/client.ts b/packages/vite/src/client/client.ts
@@ -91,9 +91,19 @@ function setupWebSocket(
     handleMessage(JSON.parse(data))
   })
 
+  let willUnload = false
+  window.addEventListener(
+    'beforeunload',
+    () => {
+      willUnload = true
+    },
+    { once: true },
+  )
+
   // ping server
-  socket.addEventListener('close', async ({ wasClean }) => {
-    if (wasClean) return
+  socket.addEventListener('close', async () => {
+    // ignore close caused by top-level navigation
+    if (willUnload) return
 
     if (!isOpened && onCloseWithoutOpen) {
       onCloseWithoutOpen()