Add in optional flag

proto-kit · Feb 4, 2025 · 3cf6f70 · 3cf6f70
1 parent 4d6b3b6
commit 3cf6f70
Showing 1 changed file with 17 additions and 9 deletions.
diff --git a/packages/sequencer/src/mempool/private/PrivateMempool.ts b/packages/sequencer/src/mempool/private/PrivateMempool.ts
@@ -32,8 +32,14 @@ type MempoolTransactionPaths = {
   transaction: PendingTransaction;
   paths: Field[];
 };
+interface PrivateMempoolConfig {
+  validationEnabled?: boolean;
+}
 @sequencerModule()
-export class PrivateMempool extends SequencerModule implements Mempool {
+export class PrivateMempool
+  extends SequencerModule<PrivateMempoolConfig>
+  implements Mempool
+{
   public readonly events = new EventEmitter<MempoolEvents>();
 
   private readonly accountStateHook: AccountStateHook;
@@ -100,14 +106,16 @@ export class PrivateMempool extends SequencerModule implements Mempool {
 
     const networkState =
       (await this.getStagedNetworkState()) ?? NetworkState.empty();
-
-    const sortedTxs = await this.checkTxValid(
-      txs,
-      baseCachedStateService,
-      this.protocol.stateServiceProvider,
-      networkState,
-      limit
-    );
+    const validationEnabled = this.config.validationEnabled ?? true;
+    const sortedTxs = validationEnabled
+      ? await this.checkTxValid(
+          txs,
+          baseCachedStateService,
+          this.protocol.stateServiceProvider,
+          networkState,
+          limit
+        )
+      : txs;
     this.protocol.stateServiceProvider.popCurrentStateService();
     return sortedTxs;
   }