Merge pull request #395 from proto-kit/feature/mempool-sorting

Mempool Sorting

Author: rpanic

CHANGELOG.md

@@ -8,6 +8,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/)
 
 ### Added
 
+- Added Mempool sorting [#395](https://github.com/proto-kit/framework/pull/395)
 - Introduced dynamic block building and JIT transaction fetching [#394](https://github.com/proto-kit/framework/pull/394)
 - Introduced block explorer [#381](https://github.com/proto-kit/framework/pull/381)
 - Added CircuitAnalysisModule for easy analysis of protocol circuits [#379](https://github.com/proto-kit/framework/pull/379)

packages/indexer/src/tasks/IndexPendingTxTask.ts

@@ -30,7 +30,7 @@ export class IndexPendingTxTask
 
   public async compute(input: PendingTransaction): Promise<string | void> {
     try {
-      await this.transactionStorage.pushUserTransaction(input);
+      await this.transactionStorage.pushUserTransaction(input, 0);
       return "";
     } catch (err) {
       log.error("Failed to process pending tx task", err);

packages/persistance/prisma/migrations/20260117134313_transaction_priority/migration.sql

@@ -0,0 +1,10 @@
+-- CreateTable
+CREATE TABLE "TransactionPriority" (
+    "transactionHash" TEXT NOT NULL,
+    "priority" BIGINT NOT NULL,
+
+    CONSTRAINT "TransactionPriority_pkey" PRIMARY KEY ("transactionHash")
+);
+
+-- AddForeignKey
+ALTER TABLE "TransactionPriority" ADD CONSTRAINT "TransactionPriority_transactionHash_fkey" FOREIGN KEY ("transactionHash") REFERENCES "Transaction"("hash") ON DELETE RESTRICT ON UPDATE CASCADE;

packages/persistance/prisma/schema.prisma

@@ -54,6 +54,18 @@ model Transaction {
   executionResult TransactionExecutionResult?
 
   IncomingMessageBatchTransaction IncomingMessageBatchTransaction[]
+
+  priority TransactionPriority?
+}
+
+model TransactionPriority {
+  transactionHash String
+
+  priority BigInt
+
+  Transaction Transaction @relation(fields: [transactionHash], references: [hash])
+
+  @@id([transactionHash])
 }
 
 model TransactionExecutionResult {

packages/persistance/src/services/prisma/PrismaTransactionStorage.ts

@@ -34,6 +34,11 @@ export class PrismaTransactionStorage implements TransactionStorage {
           equals: false,
         },
       },
+      orderBy: {
+        priority: {
+          priority: "desc",
+        },
+      },
       skip: offset,
       take: limit,
     });
@@ -56,13 +61,27 @@ export class PrismaTransactionStorage implements TransactionStorage {
     }
   }
 
-  public async pushUserTransaction(tx: PendingTransaction): Promise<boolean> {
+  public async pushUserTransaction(
+    tx: PendingTransaction,
+    priority: number
+  ): Promise<boolean> {
     const { prismaClient } = this.connection;
 
-    const result = await prismaClient.transaction.createMany({
-      data: [this.transactionMapper.mapOut(tx)],
-      skipDuplicates: true,
-    });
+    const transactionData = this.transactionMapper.mapOut(tx);
+
+    const [result] = await prismaClient.$transaction([
+      prismaClient.transaction.createMany({
+        data: [transactionData],
+        skipDuplicates: true,
+      }),
+
+      prismaClient.transactionPriority.create({
+        data: {
+          priority,
+          transactionHash: transactionData.hash,
+        },
+      }),
+    ]);
 
     return result.count === 1;
   }

packages/sdk/test/fees-multi-zkprograms.test.ts

@@ -185,9 +185,7 @@ describe("check fee analyzer", () => {
         },
       },
       Sequencer: {
-        Mempool: {
-          validationEnabled: true,
-        },
+        Mempool: {},
       },
     });
 

packages/sequencer/src/index.ts

@@ -4,6 +4,8 @@ export * from "./mempool/Mempool";
 export * from "./mempool/PendingTransaction";
 export * from "./mempool/CompressedSignature";
 export * from "./mempool/private/PrivateMempool";
+export * from "./mempool/sorting/MempoolSorting";
+export * from "./mempool/sorting/DefaultMempoolSorting";
 export * from "./sequencer/executor/Sequencer";
 export * from "./sequencer/executor/Sequenceable";
 export * from "./sequencer/SequencerIdProvider";

packages/sequencer/src/mempool/private/PrivateMempool.ts

@@ -12,20 +12,38 @@ import { TransactionValidator } from "../verification/TransactionValidator";
 import { Tracer } from "../../logging/Tracer";
 import { trace } from "../../logging/trace";
 import { IncomingMessagesService } from "../../settlement/messages/IncomingMessagesService";
+import { MempoolSorting } from "../sorting/MempoolSorting";
+import { DefaultMempoolSorting } from "../sorting/DefaultMempoolSorting";
+
+type PrivateMempoolConfig = {
+  type?: "hybrid" | "private" | "based";
+};
 
 @sequencerModule()
-export class PrivateMempool extends SequencerModule implements Mempool {
+export class PrivateMempool
+  extends SequencerModule<PrivateMempoolConfig>
+  implements Mempool
+{
   public readonly events = new EventEmitter<MempoolEvents>();
 
+  private readonly mempoolSorting: MempoolSorting;
+
   public constructor(
     private readonly transactionValidator: TransactionValidator,
     @inject("TransactionStorage")
     private readonly transactionStorage: TransactionStorage,
     @inject("IncomingMessagesService", { isOptional: true })
     private readonly messageService: IncomingMessagesService | undefined,
-    @inject("Tracer") public readonly tracer: Tracer
+    @inject("Tracer") public readonly tracer: Tracer,
+    @inject("MempoolSorting", { isOptional: true })
+    mempoolSorting: MempoolSorting | undefined
   ) {
     super();
+    this.mempoolSorting = mempoolSorting ?? new DefaultMempoolSorting();
+  }
+
+  private type() {
+    return this.config.type ?? "hybrid";
   }
 
   public async length(): Promise<number> {
@@ -36,7 +54,12 @@ export class PrivateMempool extends SequencerModule implements Mempool {
   public async add(tx: PendingTransaction): Promise<boolean> {
     const [txValid, error] = this.transactionValidator.validateTx(tx);
     if (txValid) {
-      const success = await this.transactionStorage.pushUserTransaction(tx);
+      const sortingValue = this.mempoolSorting!.presortingPriority(tx);
+
+      const success = await this.transactionStorage.pushUserTransaction(
+        tx,
+        sortingValue
+      );
       if (success) {
         this.events.emit("mempool-transaction-added", tx);
         log.trace(`Transaction added to mempool: ${tx.hash().toString()}`);
@@ -69,14 +92,27 @@ export class PrivateMempool extends SequencerModule implements Mempool {
     offset?: number,
     limit?: number
   ): Promise<PendingTransaction[]> {
-    return await this.transactionStorage.getPendingUserTransactions(
+    if (this.type() === "based") {
+      return [];
+    }
+
+    let txs = await this.transactionStorage.getPendingUserTransactions(
       offset ?? 0,
       limit
     );
+
+    if (this.mempoolSorting.enablePostSorting()) {
+      txs = this.mempoolSorting.postSorting(txs);
+    }
+
+    return txs;
   }
 
   @trace("mempool.get_mandatory_txs")
   public async getMandatoryTxs(): Promise<PendingTransaction[]> {
+    if (this.type() === "private") {
+      return [];
+    }
     return (await this.messageService?.getPendingMessages()) ?? [];
   }
 

packages/sequencer/src/mempool/sorting/DefaultMempoolSorting.ts

@@ -0,0 +1,32 @@
+import { noop } from "@proto-kit/common";
+
+import { PendingTransaction } from "../PendingTransaction";
+import {
+  SequencerModule,
+  sequencerModule,
+} from "../../sequencer/builder/SequencerModule";
+
+import { MempoolSorting } from "./MempoolSorting";
+
+@sequencerModule()
+export class DefaultMempoolSorting
+  extends SequencerModule
+  implements MempoolSorting
+{
+  public async start() {
+    noop();
+  }
+
+  public enablePostSorting(): boolean {
+    return false;
+  }
+
+  public postSorting(transactions: PendingTransaction[]): PendingTransaction[] {
+    return transactions;
+  }
+
+  public presortingPriority(tx: PendingTransaction): number {
+    // This means we order by first in, first out in the db
+    return Date.UTC(2500, 0) - Date.now();
+  }
+}

packages/sequencer/src/mempool/sorting/MempoolSorting.ts

@@ -0,0 +1,26 @@
+import { PendingTransaction } from "../PendingTransaction";
+
+export interface MempoolSorting {
+  /**
+   * Presorting happens on the backend (i.e. the DB), before the data travels to the sequencer.
+   * It's very fast, but limited to only integer sorting.
+   * The value returned here has to be static per transaction, since it will be sorted and
+   * compared on the DB-side.
+   *
+   * @param tx
+   * @returns Priority of the transaction - larger is better (therefore will be
+   * put in the block first)
+   */
+  presortingPriority(tx: PendingTransaction): number;
+
+  /**
+   * Indicate whether to do pre-sorting (as it's expensive depending on your block size)
+   */
+  enablePostSorting(): boolean;
+
+  /**
+   * Postsorting happens on the sequencer-side. It's less fast but can take in any two
+   * transactions and directly compare them based on arbitrary logic
+   */
+  postSorting(transactions: PendingTransaction[]): PendingTransaction[];
+}