fix: fix ws server (#623)

asyncapi · Dec 4, 2023 · 994b957 · 994b957
1 parent d2d4264
commit 994b957
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/src/adapters/ws/server.ts b/src/adapters/ws/server.ts
@@ -68,7 +68,16 @@ class WebSocketsAdapter extends Adapter {
   private async _connect(): Promise<this> {
     this._validatePort()
     this._createServers()
-    this.wsHttpServer.on('upgrade', this._onUpgrade)
+
+    this.wsHttpServer.on('upgrade', (request: IncomingMessage, socket: Duplex, head: Buffer) => {
+      try {
+        this._handleRequest(request, socket, head)
+      } catch (e) {
+        const errorMessage = `Error handling request: ${e.message}`
+        this.emit('error', new Error(errorMessage))
+        if (!socket.writableEnded) this._endRequest(500, 'Internal Server Error', socket)
+      }
+    })
 
     if (!this.customHttpServer) {
       this.wsHttpServer.listen(this._getPort())
@@ -170,15 +179,6 @@ class WebSocketsAdapter extends Adapter {
     })
   }
 
-  private async _onUpgrade(request: IncomingMessage, socket: Duplex, head: Buffer) {
-    try {
-      this._handleRequest(request, socket, head)
-    } catch (e) {
-      const errorMessage = `Error handling request: ${e.message}`
-      this.emit('error', new Error(errorMessage))
-      if (!socket.writableEnded) this._endRequest(500, 'Internal Server Error', socket)
-    }
-  }
 
   private _extractPathname(req: IncomingMessage) {
     const serverUrl = new URL(this.serverUrlExpanded)