Give option to mongodb for decoded tx (#38)

metagame-xyz · Aug 19, 2022 · cfe7bae · cfe7bae
1 parent 6294587
commit cfe7bae
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/src/Translator.ts b/src/Translator.ts
@@ -437,12 +437,12 @@ class Translator {
     /******       DB ONLY CALLs            ********/
     /**********************************************/
 
-    async getManyDecodedTxFromDB(txHashArr: string[]): Promise<(DecodedTx | null)[]> {
+    async getManyDecodedTxFromDB(txHashArr: string[], options?: any): Promise<(DecodedTx | null)[]> {
         if (!(this.databaseInterface instanceof MongooseDatabaseInterface)) {
             throw new Error('This function only works with MongodB')
         }
         await this.initializeMongoose()
-        const txArr = await this.databaseInterface.getManyDecodedTxArr(txHashArr)
+        const txArr = await this.databaseInterface.getManyDecodedTxArr(txHashArr, 50, options)
         return txArr
     }
 

diff --git a/src/utils/mongoose/index.ts b/src/utils/mongoose/index.ts
@@ -253,7 +253,7 @@ export class MongooseDatabaseInterface extends DatabaseInterface {
         // return here instead of in the try, so that it still works if the db is down
         return decodedTxMap
     }
-    async getManyDecodedTxArr(txHashes: string[], maxAddresses = 50): Promise<DecodedTx[]> {
+    async getManyDecodedTxArr(txHashes: string[], maxAddresses = 50, options: any = {}): Promise<DecodedTx[]> {
         let decodedTxArr = []
 
         // hack for $in being n * log(m) where n = chunk.length and m = db size
@@ -265,7 +265,11 @@ export class MongooseDatabaseInterface extends DatabaseInterface {
             .map((chunk: any) => chunk.all())
 
         const promises = chunks.map((chunk) => {
-            return DecodedTxModel.find({ txHash: { $in: chunk }, [`allAddresses.${maxAddresses}`]: { $exists: false } })
+            return DecodedTxModel.find({
+                txHash: { $in: chunk },
+                [`allAddresses.${maxAddresses}`]: { $exists: false },
+                timestamp: { $gte: options.tsStart || 0 },
+            })
         })
 
         try {