refactor: websocket servers instantiation

dherault · Sep 22, 2022 · 4788e04 · 4788e04
1 parent 1a6c2fe
commit 4788e04
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 3 deletions.
diff --git a/src/ServerlessOffline.js b/src/ServerlessOffline.js
@@ -206,6 +206,8 @@ export default class ServerlessOffline {
       this.#lambda,
     )
 
+    await this.#webSocket.createServer()
+
     this.#webSocket.create(events)
 
     await this.#webSocket.start()

diff --git a/src/events/websocket/WebSocket.js b/src/events/websocket/WebSocket.js
@@ -6,16 +6,32 @@ import WebSocketServer from './WebSocketServer.js'
 export default class WebSocket {
   #httpServer = null
 
+  #lambda = null
+
+  #options = null
+
+  #serverless = null
+
   #webSocketServer = null
 
   constructor(serverless, options, lambda) {
-    const webSocketClients = new WebSocketClients(serverless, options, lambda)
+    this.#lambda = lambda
+    this.#options = options
+    this.#serverless = serverless
+  }
+
+  async createServer() {
+    const webSocketClients = new WebSocketClients(
+      this.#serverless,
+      this.#options,
+      this.#lambda,
+    )
 
-    this.#httpServer = new HttpServer(options, webSocketClients)
+    this.#httpServer = new HttpServer(this.#options, webSocketClients)
 
     // share server
     this.#webSocketServer = new WebSocketServer(
-      options,
+      this.#options,
       webSocketClients,
       this.#httpServer.server,
     )