From 678383915a666889cb7200656b68c98ba03511bb Mon Sep 17 00:00:00 2001
From: hkjpotato <hkjpotato@gmail.com>
Date: Thu, 1 Jul 2021 10:41:09 -0400
Subject: [PATCH] rewrite retry as promise based

---
 .../amazon-cognito-identity-js/src/Client.js   | 18 +++++++-----------
 1 file changed, 7 insertions(+), 11 deletions(-)

diff --git a/packages/amazon-cognito-identity-js/src/Client.js b/packages/amazon-cognito-identity-js/src/Client.js
index 4f832ef69f7..35714526e7b 100644
--- a/packages/amazon-cognito-identity-js/src/Client.js
+++ b/packages/amazon-cognito-identity-js/src/Client.js
@@ -174,19 +174,16 @@ const isNonRetryableError = (obj) => {
 	return obj && obj[key];
 };
 
-async function retry(functionToRetry, args, delayFn, attempt = 1) {
+function retry(functionToRetry, args, delayFn, attempt = 1) {
 	if (typeof functionToRetry !== 'function') {
 		throw Error('functionToRetry must be a function');
 	}
 
 	logger.debug(`${functionToRetry.name} attempt #${attempt} with args: ${JSON.stringify(args)}`);
 
-	try {
-		return await functionToRetry(...args);
-	}
-	catch (err) {
+	return functionToRetry(...args).catch((err) => {
 		logger.debug(`error on ${functionToRetry.name}`, err);
-
+		
 		if (isNonRetryableError(err)) {
 			logger.debug(`${functionToRetry.name} non retryable error`, err);
 			throw err;
@@ -197,13 +194,12 @@ async function retry(functionToRetry, args, delayFn, attempt = 1) {
 		logger.debug(`${functionToRetry.name} retrying in ${retryIn} ms`);
 
 		if (retryIn !== false) {
-			await new Promise(res => setTimeout(res, retryIn));
-			return await retry(functionToRetry, args, delayFn, attempt + 1);
-		}
-		else {
+			return new Promise(res => setTimeout(res, retryIn))
+					.then(() => retry(functionToRetry, args, delayFn, attempt + 1))
+		} else {
 			throw err;
 		}
-	}
+	})
 }
 
 function jitteredBackoff(maxDelayMs) {