add maxAttempts

pull/7076/head
baseballyama 5 years ago
parent 6ef128e295
commit b1b11f09e2

@ -280,13 +280,13 @@ export function prettyPrintPuppeteerAssertionError(message) {
} }
} }
export async function retryAsync<T>(fn: () => Promise<T>): Promise<T> { export async function retryAsync<T>(fn: () => Promise<T>, maxAttempts: number = 3): Promise<T> {
let attempts = 0; let attempts = 0;
while (attempts <= 3) { while (attempts <= maxAttempts) {
try { try {
return await fn(); return await fn();
} catch (err) { } catch (err) {
if (++attempts >= 3) throw err; if (++attempts >= maxAttempts) throw err;
} }
} }
} }

Loading…
Cancel
Save