refactor: any type, not just websocket

Signed-off-by: Niall Shaw <[email protected]>
openwallet-foundation · Aug 29, 2023 · e741b6a · e741b6a
1 parent c460a86
commit e741b6a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/packages/core/src/agent/TransportService.ts b/packages/core/src/agent/TransportService.ts
@@ -13,8 +13,8 @@ export class TransportService {
   public transportSessionTable: TransportSessionTable = {}
 
   public saveSession(session: TransportSession) {
-    if (session.type === 'WebSocket' && session.connectionId) {
-      const oldSessions = this.getWebSocketSessionsForConnectionId(session.connectionId)
+    if (session.connectionId) {
+      const oldSessions = this.getExistingSessionsForConnectionIdAndType(session.connectionId, session.type)
       oldSessions.forEach((oldSession) => {
         if (oldSession) {
           this.removeSession(oldSession)
@@ -49,9 +49,9 @@ export class TransportService {
     delete this.transportSessionTable[session.id]
   }
 
-  private getWebSocketSessionsForConnectionId(connectionId: string) {
+  private getExistingSessionsForConnectionIdAndType(connectionId: string, type: string) {
     return Object.values(this.transportSessionTable).filter(
-      (session) => session?.connectionId === connectionId && session.type === 'WebSocket'
+      (session) => session?.connectionId === connectionId && session.type === type
     )
   }
 }