feat: add support for custom HTTP server in WebSocket server options

Koatty · Nov 27, 2024 · fbe0a9a · fbe0a9a
1 parent 6298470
commit fbe0a9a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/src/server/serve.ts b/src/server/serve.ts
@@ -55,6 +55,7 @@ export function NewServe(app: KoattyApplication, opt?: ListeningOptions): Koatty
       key: "",
       cert: "",
       protoFile: "",
+      server: null, // used by websocket
     },
     ...opt
   };

diff --git a/src/server/ws.ts b/src/server/ws.ts
@@ -40,15 +40,21 @@ export class WsServer implements KoattyServer {
     this.options.wsOptions = { ...options.ext, ...{ noServer: true } }
 
     this.server = new WebSocketServer(this.options.wsOptions);
-    if (this.options.protocol == "wss") {
-      const opt: httpsServerOptions = {
-        key: this.options.ext.key,
-        cert: this.options.ext.cert,
-      }
-      this.httpServer = httpsCreateServer(opt);
+    // set http server
+    if (options.ext.server) {
+      this.httpServer = options.ext.server;
     } else {
-      this.httpServer = createServer();
+      if (this.options.protocol == "wss") {
+        const opt: httpsServerOptions = {
+          key: this.options.ext.key,
+          cert: this.options.ext.cert,
+        }
+        this.httpServer = httpsCreateServer(opt);
+      } else {
+        this.httpServer = createServer();
+      }
     }
+
     CreateTerminus(this);
   }