Add resolver to fetch all flows without pages

src/domain-services/flows/graphql/resolver.ts

@@ -6,6 +6,7 @@ import { SearchFlowsArgs, SearchFlowsArgsNonPaginated } from './args';
 import {
   FlowPaged,
   FlowSearchResult,
+  FlowSearchResultNonPaginated,
   FlowSearchTotalAmountResult,
 } from './types';
 
@@ -31,4 +32,50 @@ export default class FlowResolver {
   ): Promise<FlowSearchTotalAmountResult> {
     return await this.flowSearchService.searchTotalAmount(context.models, args);
   }
+
+  @Query(() => FlowSearchResultNonPaginated)
+  async searchFlowsBatches(
+    @Ctx() context: Context,
+    @Args(() => SearchFlowsArgs, { validate: false })
+    args: SearchFlowsArgs
+  ): Promise<FlowSearchResultNonPaginated> {
+    // Set default batch size to 1000
+    args.limit = args.limit > 0 ? args.limit : 1000;
+
+    const flowSearchResponse = await this.flowSearchService.search(
+      context.models,
+      args
+    );
+
+    const batchesMissing =
+      Math.round(flowSearchResponse.total / args.limit) - 1;
+    const flows: FlowPaged[] = flowSearchResponse.flows;
+
+    let hasNextPage = flowSearchResponse.hasNextPage;
+    let batchCount = 1;
+
+    let cursor = flowSearchResponse.endCursor;
+    let nextArgs: SearchFlowsArgs = { ...args, afterCursor: cursor };
+
+    let nextFlowSearchResponse: FlowSearchResult;
+    while (hasNextPage) {
+      batchCount++;
+
+      nextFlowSearchResponse = await this.flowSearchService.search(
+        context.models,
+        nextArgs
+      );
+
+      flows.push(...nextFlowSearchResponse.flows);
+
+      hasNextPage =
+        nextFlowSearchResponse.hasNextPage && batchCount <= batchesMissing;
+
+      cursor = nextFlowSearchResponse.endCursor;
+      // Update the cursor for the next iteration
+      nextArgs = { ...args, afterCursor: cursor };
+    }
+
+    return { flows, flowsCount: flows.length };
+  }
 }