fix: do not send package via outdated session (#1559)

Signed-off-by: Niall Shaw <niall.shaw@absa.africa>
openwallet-foundation · Aug 30, 2023 · de6a735 · de6a735
1 parent c0d9304
commit de6a735
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/packages/core/src/agent/TransportService.ts b/packages/core/src/agent/TransportService.ts
@@ -13,6 +13,14 @@ export class TransportService {
   public transportSessionTable: TransportSessionTable = {}
 
   public saveSession(session: TransportSession) {
+    if (session.connectionId) {
+      const oldSessions = this.getExistingSessionsForConnectionIdAndType(session.connectionId, session.type)
+      oldSessions.forEach((oldSession) => {
+        if (oldSession) {
+          this.removeSession(oldSession)
+        }
+      })
+    }
     this.transportSessionTable[session.id] = session
   }
 
@@ -40,6 +48,12 @@ export class TransportService {
   public removeSession(session: TransportSession) {
     delete this.transportSessionTable[session.id]
   }
+
+  private getExistingSessionsForConnectionIdAndType(connectionId: string, type: string) {
+    return Object.values(this.transportSessionTable).filter(
+      (session) => session?.connectionId === connectionId && session.type === type
+    )
+  }
 }
 
 interface TransportSessionTable {