feat: add an option to not start a websocket server

vitejs · Mar 20, 2024 · 2ddc0d1 · 2ddc0d1
1 parent d443428
commit 2ddc0d1
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/packages/vite/src/node/server/index.ts b/packages/vite/src/node/server/index.ts
@@ -97,6 +97,10 @@ export interface ServerOptions extends CommonServerOptions {
  * Configure HMR-specific options (port, host, path & protocol)
  */
  hmr?: HmrOptions | boolean
+ /**
+ * Do not start the websocket connection.
+ */
+ ws?: false
  /**
  * Warm-up files to transform and cache the results in advance. This improves the
  * initial page load during server starts and prevents transform waterfalls.

diff --git a/packages/vite/src/node/server/ws.ts b/packages/vite/src/node/server/ws.ts
@@ -85,11 +85,32 @@ const wsServerEvents = [
  'message',
 ]
 
+function noop() {
+ // noop
+}
+
 export function createWebSocketServer(
  server: HttpServer | null,
  config: ResolvedConfig,
  httpsOptions?: HttpsServerOptions,
 ): WebSocketServer {
+ if (config.server.ws === false) {
+ const clients = new Set<WebSocketClient>()
+ return {
+ name: 'ws',
+ get clients() {
+ return clients
+ },
+ async close() {
+ // noop
+ },
+ on: noop as any as WebSocketServer['on'],
+ off: noop as any as WebSocketServer['off'],
+ listen: noop,
+ send: noop,
+ }
+ }
+
  let wss: WebSocketServerRaw_
  let wsHttpServer: Server | undefined = undefined