Merge branch 'master' into fabien/copeland

snapshot-labs · Jan 31, 2025 · 9e45d49 · 9e45d49
2 parents 8d3c2ab + 5a15ac4
commit 9e45d49
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/src/utils/provider.ts b/src/utils/provider.ts
@@ -8,20 +8,25 @@ const batchedProviders = {};
 
 export type ProviderOptions = {
   broviderUrl?: string;
+  timeout?: number;
 };
 
 const DEFAULT_BROVIDER_URL = 'https://rpc.snapshot.org';
+const DEFAULT_TIMEOUT = 25000;
 
 export default function getProvider(
   network,
-  { broviderUrl = DEFAULT_BROVIDER_URL }: ProviderOptions = {}
+  {
+    broviderUrl = DEFAULT_BROVIDER_URL,
+    timeout = DEFAULT_TIMEOUT
+  }: ProviderOptions = {}
 ) {
   const url = `${broviderUrl}/${network}`;
   if (!providers[network])
     providers[network] = new StaticJsonRpcProvider(
       {
         url,
-        timeout: 25000,
+        timeout,
         allowGzip: true
       },
       Number(network)
@@ -31,13 +36,16 @@ export default function getProvider(
 
 export function getBatchedProvider(
   network,
-  { broviderUrl = DEFAULT_BROVIDER_URL }: ProviderOptions = {}
+  {
+    broviderUrl = DEFAULT_BROVIDER_URL,
+    timeout = DEFAULT_TIMEOUT
+  }: ProviderOptions = {}
 ) {
   const url = `${broviderUrl}/${network}`;
   if (!batchedProviders[network])
     batchedProviders[network] = new JsonRpcBatchProvider({
       url,
-      timeout: 25000,
+      timeout,
       allowGzip: true
     });
   return batchedProviders[network];