From 2aef64d0a0b8c44818713448f65460bfb9aa668c Mon Sep 17 00:00:00 2001
From: Nicko Guyer <nicko.guyer@kaleido.io>
Date: Mon, 26 Feb 2024 14:37:11 -0500
Subject: [PATCH] Fix tracking expected acks per client

Signed-off-by: Nicko Guyer <nicko.guyer@kaleido.io>
---
 src/eventstream-proxy/eventstream-proxy.base.ts | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/eventstream-proxy/eventstream-proxy.base.ts b/src/eventstream-proxy/eventstream-proxy.base.ts
index fdb2a55..5d9696e 100644
--- a/src/eventstream-proxy/eventstream-proxy.base.ts
+++ b/src/eventstream-proxy/eventstream-proxy.base.ts
@@ -69,6 +69,11 @@ export abstract class EventStreamProxyBase extends WebSocketEventsBase {
 
   handleConnection(client: WebSocketEx) {
     super.handleConnection(client);
+
+    if (!this.awaitingAck.get(client.id)) {
+      this.awaitingAck.set(client.id, []);
+    }
+
     client.on('message', async (message: string) => {
       const action = JSON.parse(message) as WebSocketActionBase;
       switch (action.type) {