From 28d65ec2776bcbc364ab00dbe83026e8b8b6c038 Mon Sep 17 00:00:00 2001
From: Andrey Platov <andrey@hardcoreeng.com>
Date: Thu, 19 Aug 2021 19:34:58 +0200
Subject: [PATCH] buffer messages while session creates

Signed-off-by: Andrey Platov <andrey@hardcoreeng.com>
---
 server/ws/src/server.ts | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/server/ws/src/server.ts b/server/ws/src/server.ts
index d500da09ec..8585fef2d4 100644
--- a/server/ws/src/server.ts
+++ b/server/ws/src/server.ts
@@ -128,10 +128,17 @@ export function start (storageFactory: (workspace: string) => Promise<ServerStor
   const wss = new Server({ noServer: true })
   // eslint-disable-next-line @typescript-eslint/no-misused-promises
   wss.on('connection', async (ws: WebSocket, request: any, token: _Token) => {
+    const buffer: string[] = []
+
+    ws.on('message', (msg: string) => { buffer.push(msg) })
     const session = await sessions.addSession(ws, token, storageFactory)
     // eslint-disable-next-line @typescript-eslint/no-misused-promises
     ws.on('message', async (msg: string) => await handleRequest(session, ws, msg))
     ws.on('close', (code: number, reason: string) => sessions.close(ws, token, code, reason))
+
+    for (const msg of buffer) {
+      await handleRequest(session, ws, msg)
+    }
   })
 
   const server = createServer()