Merge pull request #1 from Instadapp/batch-retry-provider

src/providers/retry-provider.ts

@@ -1,4 +1,4 @@
-import { JsonRpcProvider, StaticJsonRpcProvider } from '@ethersproject/providers'
+import { JsonRpcProvider, StaticJsonRpcProvider, JsonRpcBatchProvider } from '@ethersproject/providers'
 import { retry } from '../promises'
 
 export class JsonRpcRetryProvider extends JsonRpcProvider {
@@ -26,6 +26,31 @@ export class JsonRpcRetryProvider extends JsonRpcProvider {
   }
 }
 
+export class JsonRpcRetryBatchProvider extends JsonRpcBatchProvider {
+  timeouts: number[] = [5_000, 10_000, 15_000]
+  delay: number = 300
+
+  constructor (url: string, options?: { timeouts?: number[], delay?: number }) {
+    super(url)
+
+    if (options && options.timeouts) {
+      this.timeouts = options.timeouts
+    }
+
+    if (options && options.delay) {
+      this.delay = options.delay
+    }
+
+    Object.setPrototypeOf(this, JsonRpcRetryProvider.prototype)
+  }
+
+  public perform (method: string, params: any): Promise<any> {
+    const operation = () => super.perform(method, params)
+
+    return retry(operation, { timeouts: this.timeouts, delay: this.delay })
+  }
+}
+
 export class StaticJsonRpcRetryProvider extends StaticJsonRpcProvider {
   timeouts: number[] = [5_000, 10_000, 15_000]
   delay: number = 300