feat: support multiple JITO endpoints with round-robin retry

Co-Authored-By: Ali Behjati <[email protected]>

Author: devin-ai-integration[bot]

apps/price_pusher/src/solana/command.ts

@@ -50,7 +50,7 @@ export default {
       default: 50000,
     } as Options,
     "jito-endpoint": {
-      description: "Jito endpoint",
+      description: "Jito endpoint(s) - comma-separated list of endpoints",
       type: "string",
       optional: true,
     } as Options,
@@ -209,7 +209,13 @@ export default {
         Uint8Array.from(JSON.parse(fs.readFileSync(jitoKeypairFile, "ascii"))),
       );
 
-      const jitoClient = searcherClient(jitoEndpoint, jitoKeypair);
+      const jitoEndpoints = jitoEndpoint
+        .split(",")
+        .map((endpoint: string) => endpoint.trim());
+      const jitoClients: SearcherClient[] = jitoEndpoints.map(
+        (endpoint: string) => searcherClient(endpoint, jitoKeypair),
+      );
+
       solanaPricePusher = new SolanaPricePusherJito(
         pythSolanaReceiver,
         hermesClient,
@@ -218,13 +224,16 @@ export default {
         jitoTipLamports,
         dynamicJitoTips,
         maxJitoTipLamports,
-        jitoClient,
+        jitoClients,
         jitoBundleSize,
         updatesPerJitoBundle,
+        60000, // Default max retry time of 60 seconds
         lookupTableAccount,
       );
 
-      onBundleResult(jitoClient, logger.child({ module: "JitoClient" }));
+      jitoClients.forEach((client, index) => {
+        onBundleResult(client, logger.child({ module: `JitoClient-${index}` }));
+      });
     } else {
       solanaPricePusher = new SolanaPricePusher(
         pythSolanaReceiver,

apps/price_pusher/src/solana/solana.ts

@@ -165,9 +165,10 @@ export class SolanaPricePusherJito implements IPricePusher {
     private defaultJitoTipLamports: number,
     private dynamicJitoTips: boolean,
     private maxJitoTipLamports: number,
-    private searcherClient: SearcherClient,
+    private searcherClients: SearcherClient[],
     private jitoBundleSize: number,
     private updatesPerJitoBundle: number,
+    private maxRetryTimeMs: number = 60000, // Default to 60 seconds max retry time
     private addressLookupTableAccount?: AddressLookupTableAccount,
   ) {}
 
@@ -242,27 +243,34 @@ export class SolanaPricePusherJito implements IPricePusher {
         jitoBundleSize: this.jitoBundleSize,
       });
 
-      let retries = 60;
-      while (retries > 0) {
-        try {
-          await sendTransactionsJito(
-            transactions,
-            this.searcherClient,
-            this.pythSolanaReceiver.wallet,
-          );
-          break;
-        } catch (err: any) {
-          if (err.code === 8 && err.details?.includes("Rate limit exceeded")) {
-            this.logger.warn("Rate limit hit, waiting before retry...");
-            await this.sleep(1100); // Wait slightly more than 1 second
-            retries--;
-            if (retries === 0) {
-              this.logger.error("Max retries reached for rate limit");
-              throw err;
-            }
-          } else {
-            throw err;
+      try {
+        await sendTransactionsJito(
+          transactions,
+          this.searcherClients,
+          this.pythSolanaReceiver.wallet,
+          { maxRetryTimeMs: this.maxRetryTimeMs },
+        );
+      } catch (err: any) {
+        if (err.code === 8 && err.details?.includes("Rate limit exceeded")) {
+          this.logger.warn("Rate limit hit, waiting before retry...");
+          await this.sleep(1100); // Wait slightly more than 1 second
+          try {
+            await sendTransactionsJito(
+              transactions,
+              this.searcherClients,
+              this.pythSolanaReceiver.wallet,
+              { maxRetryTimeMs: this.maxRetryTimeMs },
+            );
+          } catch (retryErr: any) {
+            this.logger.error("Failed after rate limit retry");
+            throw retryErr;
           }
+        } else {
+          this.logger.error(
+            { err },
+            "Failed to send transactions via all JITO endpoints",
+          );
+          throw err;
         }
       }
 

target_chains/solana/sdk/js/solana_utils/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pythnetwork/solana-utils",
-  "version": "0.4.4",
+  "version": "0.4.5",
   "description": "Utility functions for Solana",
   "homepage": "https://pyth.network",
   "main": "lib/index.js",

target_chains/solana/sdk/js/solana_utils/src/jito.ts

@@ -42,9 +42,23 @@ export async function sendTransactionsJito(
     tx: VersionedTransaction;
     signers?: Signer[] | undefined;
   }[],
-  searcherClient: SearcherClient,
+  searcherClients: SearcherClient | SearcherClient[],
   wallet: Wallet,
+  options: {
+    maxRetryTimeMs?: number;
+  } = {},
 ): Promise<string> {
+  const clients = Array.isArray(searcherClients)
+    ? searcherClients
+    : [searcherClients];
+
+  if (clients.length === 0) {
+    throw new Error("No searcher clients provided");
+  }
+
+  const maxRetryTimeMs = options.maxRetryTimeMs || 60000; // Default to 60 seconds
+  const startTime = Date.now();
+
   const signedTransactions = [];
 
   for (const transaction of transactions) {
@@ -64,7 +78,24 @@ export async function sendTransactionsJito(
   );
 
   const bundle = new Bundle(signedTransactions, 2);
-  await searcherClient.sendBundle(bundle);
 
-  return firstTransactionSignature;
+  let lastError: Error | null = null;
+  let clientIndex = 0;
+
+  while (Date.now() - startTime < maxRetryTimeMs) {
+    const currentClient = clients[clientIndex];
+    try {
+      await currentClient.sendBundle(bundle);
+      return firstTransactionSignature;
+    } catch (err: any) {
+      lastError = err;
+      clientIndex = (clientIndex + 1) % clients.length;
+      await new Promise((resolve) => setTimeout(resolve, 500));
+    }
+  }
+
+  throw (
+    lastError ||
+    new Error("Failed to send transactions via JITO after maximum retry time")
+  );
 }