From 355dc574f8c8d6191de2285de80cd312076d9eac Mon Sep 17 00:00:00 2001
From: Sergii KLiuchnyk <redexp@ukr.net>
Date: Fri, 30 Sep 2022 12:26:08 +0300
Subject: [PATCH] build

---
 dist/src/instance.js                       | 11 ++++++-----
 dist/src/services/webSocketServer/index.js |  2 +-
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/dist/src/instance.js b/dist/src/instance.js
index c528f02..0ee2bf9 100644
--- a/dist/src/instance.js
+++ b/dist/src/instance.js
@@ -27,12 +27,13 @@ const createInstance = ({ app, server, options }) => {
     app.use(options.path, api);
     //use mountpath for WS server
     const customConfig = Object.assign(Object.assign({}, config), { path: path_1.default.posix.join(app.path(), options.path, '/') });
-    const wss2 = new webSocketServer_1.WebSocketServer({
+    const wss = new webSocketServer_1.WebSocketServer({
         server,
         realm,
         config: customConfig
     });
-    wss2.on("connection", (client) => {
+    app.set('peerWs', wss);
+    wss.on("connection", (client) => {
         const messageQueue = realm.getMessageQueueById(client.getId());
         if (messageQueue) {
             let message;
@@ -43,14 +44,14 @@ const createInstance = ({ app, server, options }) => {
         }
         app.emit("connection", client);
     });
-    wss2.on("message", (client, message) => {
+    wss.on("message", (client, message) => {
         app.emit("message", client, message);
         messageHandler.handle(client, message);
     });
-    wss2.on("close", (client) => {
+    wss.on("close", (client) => {
         app.emit("disconnect", client);
     });
-    wss2.on("error", (error) => {
+    wss.on("error", (error) => {
         app.emit("error", error);
     });
     messagesExpire.startMessagesExpiration();
diff --git a/dist/src/services/webSocketServer/index.js b/dist/src/services/webSocketServer/index.js
index f3cb5f3..b36a749 100644
--- a/dist/src/services/webSocketServer/index.js
+++ b/dist/src/services/webSocketServer/index.js
@@ -18,7 +18,7 @@ class WebSocketServer extends events_1.default {
         this.config = config;
         const path = this.config.path;
         this.path = `${path}${path.endsWith('/') ? "" : "/"}${WS_PATH}`;
-        this.socketServer = new ws_1.default.Server({ path: this.path, server });
+        this.socketServer = new ws_1.default.Server(Object.assign({ path: this.path, server }, this.config.ws));
         this.socketServer.on("connection", (socket, req) => this._onSocketConnection(socket, req));
         this.socketServer.on("error", (error) => this._onSocketError(error));
     }