fix: buffer full reload messages

Author: sapphi-red

packages/vite/src/node/server/ws.ts

@@ -9,7 +9,11 @@ import colors from 'picocolors'
 import type { WebSocket as WebSocketRaw } from 'ws'
 import { WebSocketServer as WebSocketServerRaw_ } from 'ws'
 import type { WebSocket as WebSocketTypes } from 'dep-types/ws'
-import type { ErrorPayload, HotPayload } from 'types/hmrPayload'
+import type {
+  ErrorPayload,
+  FullReloadPayload,
+  HotPayload,
+} from 'types/hmrPayload'
 import type { InferCustomEventPayload } from 'types/customEvent'
 import type { ResolvedConfig } from '..'
 import { isObject } from '../utils'
@@ -231,9 +235,9 @@ export function createWebSocketServer(
       })
     })
     socket.send(JSON.stringify({ type: 'connected' }))
-    if (bufferedError) {
-      socket.send(JSON.stringify(bufferedError))
-      bufferedError = null
+    if (bufferedMessage) {
+      socket.send(JSON.stringify(bufferedMessage))
+      bufferedMessage = null
     }
   })
 
@@ -279,13 +283,18 @@ export function createWebSocketServer(
   // sends the error payload before the client connection is established.
   // If we have no open clients, buffer the error and send it to the next
   // connected client.
-  let bufferedError: ErrorPayload | null = null
+  // The same thing may happen when the optimizer runs fast enough to
+  // finish the bundling before the client connects.
+  let bufferedMessage: ErrorPayload | FullReloadPayload | null = null
 
   const normalizedHotChannel = normalizeHotChannel(
     {
       send(payload) {
-        if (payload.type === 'error' && !wss.clients.size) {
-          bufferedError = payload
+        if (
+          (payload.type === 'error' || payload.type === 'full-reload') &&
+          !wss.clients.size
+        ) {
+          bufferedMessage = payload
           return
         }