fix(batch): renable doBatchRead when sortKey and blockHeight are not …

src/api/warp.ts

@@ -24,7 +24,7 @@
 } from 'warp-contracts';
 import {
   DEFAULT_EVALUATION_OPTIONS,
-  // DEFAULT_PAGES_PER_BATCH,
+  DEFAULT_PAGES_PER_BATCH,
   allowedContractTypes,
 } from '../constants';
 import { ContractType, EvaluatedContractState } from '../types';
@@ -107,7 +107,7 @@
   constructor(
     public readonly contractTxId: string,
     public readonly functionName: string,
     public readonly input: any,
     public readonly warp: Warp,
     public readonly evaluationOptions: Partial<EvaluationOptions>,
     public readonly logger?: winston.Logger,
@@ -218,21 +218,22 @@
     ),
   });
 
-  // only use batch read if no block height provided (it does not currently support block heights)
-  const doBatchRead = !!providedSortKey || providedBlockHeight === undefined;
+  // only use batch read if no block height or sort key provided
+  const doBatchRead = !providedSortKey && !providedBlockHeight;
   logger?.debug('Evaluating contract state...', {
     contractTxId,
     cacheKey: cacheKey.toString(),
     doBatchRead, // there is a bug in warp where readStateBatch is not properly using sortKeys
     evaluationOptions,
   });
 
-  // TODO: re-enable readStateBatch after sortKey is fixed
-  const readStatePromise = contract.readState(
-    providedSortKey || providedBlockHeight,
-    undefined,
-    signal,
-  );
+  const readStatePromise = doBatchRead
+    ? contract.readStateBatch(DEFAULT_PAGES_PER_BATCH, undefined, signal)
+    : contract.readState(
+        providedSortKey || providedBlockHeight,
+        undefined,
+        signal,
+      );
 
   // set cached value for multiple requests during initial promise
   stateRequestMap.set(cacheId, readStatePromise);
@@ -308,8 +309,8 @@
     return new NotFoundError(error.message);
   } else if (
     (error instanceof Error &&
       (error as any).type &&
       ['TX_NOT_FOUND', 'TX_INVALID'].includes((error as any).type)) ||
     (typeof error === 'string' && (error as string).includes('TX_INVALID'))
   ) {
     return new NotFoundError(`Contract not found. ${error}`);
@@ -481,7 +482,7 @@
   functionName: string;
   input: ParsedUrlQuery;
 }): Promise<{
   result: any;
   evaluationOptions: Partial<EvaluationOptions>;
 }> {
   try {