feat(routing): support promise in message repo (#959)

Signed-off-by: Timo Glastra <[email protected]>
openwallet-foundation · Jul 24, 2022 · 79c5d8d · 79c5d8d
1 parent 3e415d8
commit 79c5d8d
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/packages/core/src/agent/MessageSender.ts b/packages/core/src/agent/MessageSender.ts
@@ -152,7 +152,7 @@ export class MessageSender {
     // If the other party shared a queue service endpoint in their did doc we queue the message
     if (queueService) {
       this.logger.debug(`Queue packed message for connection ${connection.id} (${connection.theirLabel})`)
-      this.messageRepository.add(connection.id, encryptedMessage)
+      await this.messageRepository.add(connection.id, encryptedMessage)
       return
     }
 
@@ -267,7 +267,7 @@ export class MessageSender {
       }
 
       const encryptedMessage = await this.envelopeService.packMessage(payload, keys)
-      this.messageRepository.add(connection.id, encryptedMessage)
+      await this.messageRepository.add(connection.id, encryptedMessage)
       return
     }
 

diff --git a/packages/core/src/modules/routing/services/MessagePickupService.ts b/packages/core/src/modules/routing/services/MessagePickupService.ts
@@ -21,7 +21,7 @@ export class MessagePickupService {
     const connection = messageContext.assertReadyConnection()
 
     const { message } = messageContext
-    const messages = this.messageRepository.takeFromQueue(connection.id, message.batchSize)
+    const messages = await this.messageRepository.takeFromQueue(connection.id, message.batchSize)
 
     // TODO: each message should be stored with an id. to be able to conform to the id property
     // of batch message
@@ -39,7 +39,7 @@ export class MessagePickupService {
     return createOutboundMessage(connection, batchMessage)
   }
 
-  public queueMessage(connectionId: string, message: EncryptedMessage) {
-    this.messageRepository.add(connectionId, message)
+  public async queueMessage(connectionId: string, message: EncryptedMessage) {
+    await this.messageRepository.add(connectionId, message)
   }
 }
diff --git a/packages/core/src/storage/MessageRepository.ts b/packages/core/src/storage/MessageRepository.ts
@@ -1,6 +1,6 @@
 import type { EncryptedMessage } from '../types'
 
 export interface MessageRepository {
-  takeFromQueue(connectionId: string, limit?: number): EncryptedMessage[]
-  add(connectionId: string, payload: EncryptedMessage): void
+  takeFromQueue(connectionId: string, limit?: number): EncryptedMessage[] | Promise<EncryptedMessage[]>
+  add(connectionId: string, payload: EncryptedMessage): void | Promise<void>
 }