feat(ref-imp): added timeout for fetaching operations of a DID

decentralized-identity · Apr 9, 2021 · b840682 · b840682
1 parent 4cc7511
commit b840682
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/lib/common/MongoDbStore.ts b/lib/common/MongoDbStore.ts
@@ -5,6 +5,9 @@ import Logger from '../common/Logger';
  * Base class that contains the common MongoDB collection setup.
  */
 export default class MongoDbStore {
+
+  public static readonly defaultQueryTimeoutInMilliseconds = 10000;
+
   /** MondoDB instance. */
   protected db: Db | undefined;
   /** MongoDB collection */

diff --git a/lib/core/MongoDbOperationStore.ts b/lib/core/MongoDbOperationStore.ts
@@ -57,7 +57,12 @@ export default class MongoDbOperationStore extends MongoDbStore implements IOper
    * Gets all operations of the given DID unique suffix in ascending chronological order.
    */
   public async get (didUniqueSuffix: string): Promise<AnchoredOperationModel[]> {
-    const mongoOperations = await this.collection!.find({ didSuffix: didUniqueSuffix }).sort({ txnNumber: 1, opIndex: 1 }).toArray();
+    const mongoOperations = await this.collection!
+      .find({ didSuffix: didUniqueSuffix })
+      .sort({ txnNumber: 1, opIndex: 1 })
+      .maxTimeMS(MongoDbStore.defaultQueryTimeoutInMilliseconds)
+      .toArray();
+
     return mongoOperations.map((operation) => { return MongoDbOperationStore.convertToAnchoredOperationModel(operation); });
   }