chore: update dedup logic

packages/auth/src/providers/cognito/utils/refreshAuthTokens.ts

@@ -6,6 +6,7 @@ import { AuthConfig } from '@aws-amplify/core';
 import {
 	assertTokenProviderConfig,
 	decodeJWT,
+	deDupCallback,
 } from '@aws-amplify/core/internals/utils';
 import { initiateAuth } from '../utils/clients/CognitoIdentityProvider';
 import { getRegion } from '../utils/clients/CognitoIdentityProvider/utils';
@@ -40,7 +41,8 @@ export const refreshAuthTokens: TokenRefresher = async ({
 		userPoolClientId: authConfig.Cognito.userPoolClientId,
 	});
 
-	const { AuthenticationResult } = await initiateAuth(
+	const deDuplicatedInitiateAuth = deDupCallback(initiateAuth);
+	const { AuthenticationResult } = await deDuplicatedInitiateAuth(
 		{ region },
 		{
 			ClientId: authConfig?.Cognito?.userPoolClientId,

packages/core/src/libraryUtils.ts

@@ -17,6 +17,7 @@ export {
 	retry,
 	urlSafeDecode,
 	urlSafeEncode,
+	deDupCallback,
 } from './utils';
 export { parseAWSExports } from './parseAWSExports';
 export { LegacyConfig } from './singleton/types';

packages/core/src/singleton/apis/fetchAuthSession.ts

@@ -4,10 +4,7 @@
 import { fetchAuthSession as fetchAuthSessionInternal } from './internal/fetchAuthSession';
 import { Amplify } from '../Amplify';
 import { FetchAuthSessionOptions } from '../Auth/types';
-import { debounceCallback } from '../../utils/debounceCallback';
 
-export const fetchAuthSession = debounceCallback(
-	(options?: FetchAuthSessionOptions) => {
-		return fetchAuthSessionInternal(Amplify, options);
-	}
-);
+export const fetchAuthSession = (options?: FetchAuthSessionOptions) => {
+	return fetchAuthSessionInternal(Amplify, options);
+};

packages/core/src/utils/deDupCallback.ts

@@ -0,0 +1,34 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+/**
+ * returns inflight promise if there hasn't been resolved yet
+ *
+ * @param callback - callback to be deDup.
+ * @returns - the return type of the callback
+ */
+export const deDupCallback = <F extends (...args: any[]) => any>(
+	callback: F
+) => {
+	let inflightPromise: Promise<Awaited<ReturnType<F>>> | undefined;
+	return async (...args: Parameters<F>): Promise<Awaited<ReturnType<F>>> => {
+		if (inflightPromise) return inflightPromise;
+
+		if (!inflightPromise) {
+			inflightPromise = new Promise(async (resolve, reject) => {
+				try {
+					const result = await callback(...args);
+					resolve(result);
+				} catch (error) {
+					reject(error);
+				}
+			});
+		}
+
+		try {
+			return await inflightPromise;
+		} finally {
+			inflightPromise = undefined;
+		}
+	};
+};

packages/core/src/utils/index.ts

@@ -15,3 +15,4 @@ export {
 export { urlSafeDecode } from './urlSafeDecode';
 export { urlSafeEncode } from './urlSafeEncode';
 export { deepFreeze } from './deepFreeze';
+export { deDupCallback } from './deDupCallback';