batch headerId in transaction query

nautls · Jan 30, 2024 · ab78b1e · ab78b1e
1 parent 5545ee5
commit ab78b1e
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/src/context/transactions-repository.ts b/src/context/transactions-repository.ts
@@ -12,6 +12,7 @@ type TransactionFindOptions = FindManyParams<TransactionEntity> & {
   address?: string;
   addresses?: string[];
   transactionIds?: string[];
+  headerIds?: string[];
 };
 
 export class TransactionRepository extends BaseRepository<TransactionEntity> {
@@ -23,7 +24,7 @@ export class TransactionRepository extends BaseRepository<TransactionEntity> {
   }
 
   public override async find(options: TransactionFindOptions): Promise<TransactionEntity[]> {
-    const { minHeight, maxHeight, address, addresses, transactionIds } = options;
+    const { minHeight, maxHeight, address, addresses, transactionIds, headerIds } = options;
     const ergoTrees = addresses
       ? addresses.map((address) => ErgoAddress.fromBase58(address).ergoTree)
       : [];
@@ -65,13 +66,23 @@ export class TransactionRepository extends BaseRepository<TransactionEntity> {
           filterQuery = filterQuery.andWhere(`${this.alias}.transactionId IN (:...transactionIds)`);
         }
 
+        if (options.where?.headerId && headerIds) {
+          headerIds.push(options.where.headerId);
+          delete options.where.headerId;
+        }
+
+        if (headerIds && headerIds.length > 0) {
+          filterQuery = filterQuery.andWhere(`${this.alias}.headerId IN (:...headerIds)`);
+        } 
+
         filterQuery = filterQuery.setParameters(
           removeUndefined({
             height: maxHeight,
             maxHeight,
             minHeight,
             ergoTrees,
-            transactionIds
+            transactionIds,
+            headerIds
           })
         );
 

diff --git a/src/graphql/resolvers/transaction-resolver.ts b/src/graphql/resolvers/transaction-resolver.ts
@@ -16,9 +16,13 @@ class TransactionArguments {
   @ArrayMaxSize(20)
   transactionIds?: string[];
 
+  /** @deprecated */
   @Field(() => String, { nullable: true })
   headerId?: string;
 
+  @Field(() => [String], { nullable: true })
+  headerIds?: string[];
+
   @Field(() => Int, { nullable: true })
   inclusionHeight?: number;