amplify-android: CredentialStoreClient$loadCredentials$2$1.invoke java.lang.IllegalStateException - Already resumed

Before opening, please confirm:

• I have searched for duplicate or closed issues and discussions.

Language and Async Model

Kotlin

Amplify Categories

Authentication

Gradle script dependencies

Environment information

Welcome to Gradle 7.5!

Here are the highlights of this release:
 - Support for Java 18
 - Support for building with Groovy 4
 - Much more responsive continuous builds
 - Improved diagnostics for dependency resolution

For more details see https://docs.gradle.org/7.5/release-notes.html


------------------------------------------------------------
Gradle 7.5
------------------------------------------------------------

Build time:   2022-07-14 12:48:15 UTC
Revision:     c7db7b958189ad2b0c1472b6fe663e6d654a5103

Kotlin:       1.6.21
Groovy:       3.0.10
Ant:          Apache Ant(TM) version 1.10.11 compiled on July 10 2021
JVM:          17 (Oracle Corporation 17+35-2724)
OS:           Mac OS X 13.2.1 x86_64

Please include any relevant guides or documentation you’re referencing

No response

Describe the bug

I am using Amplify Authentication library. I am trying to call this function multiple times in the background when it return false,

private suspend fun getAuthSessionAndSetAccessTokenWhenRefreshed(): AWSCognitoAuthSession? {
        return try {
            val authSession = Amplify.Auth.fetchAuthSession() as AWSCognitoAuthSession
            val idResult = authSession.identityIdResult
            if (idResult.type == AuthSessionResult.Type.SUCCESS) {
                authSession
            } else {
                if (idResult.error is SessionExpiredException) {
                    if (idResult.error?.cause is NotAuthorizedException) {
                        logout()
                        authSession
                    } else {
                      AuthFetchSessionOptions.builder().forceRefresh(true).build().let { option ->
                            (Amplify.Auth.fetchAuthSession(option) as AWSCognitoAuthSession).also { session ->
                                val newIdentityIdResult = session.identityIdResult
                                if (newIdentityIdResult.type == AuthSessionResult.Type.FAILURE && newIdentityIdResult.error?.cause is NotAuthorizedException) {
                                       logout()
                                } else {
                                    val tokenResult = session.userPoolTokensResult
                                    if (tokenResult.type == AuthSessionResult.Type.SUCCESS) {
                                        setAccessToken(tokenResult.value?.accessToken)
                                    }
                                }
                            }
                        }
                    }
                } else {
                    authSession
                }
            }
        } catch (exception: AuthException) {
            null
        }
    }

But sometimes I found there are some crash happening in the Amplify library. Here is the crash output,

Fatal Exception: java.lang.IllegalStateException: Already resumed
       at kotlin.coroutines.SafeContinuation.resumeWith(SafeContinuationJvm.kt:44)
       at com.amplifyframework.auth.cognito.CredentialStoreClient$loadCredentials$2$1.invoke(CredentialStoreClient.kt:97)
       at com.amplifyframework.auth.cognito.CredentialStoreClient$loadCredentials$2$1.invoke(CredentialStoreClient.kt:95)
       at com.amplifyframework.auth.cognito.CredentialStoreClient$listenForResult$1.invoke(CredentialStoreClient.kt:77)
       at com.amplifyframework.auth.cognito.CredentialStoreClient$listenForResult$1.invoke(CredentialStoreClient.kt:60)
       at com.amplifyframework.statemachine.StateMachine.notifySubscribers(StateMachine.kt:176)
       at com.amplifyframework.statemachine.StateMachine.process(StateMachine.kt:191)
       at com.amplifyframework.statemachine.StateMachine.access$process(StateMachine.kt:49)
       at com.amplifyframework.statemachine.StateMachine$send$1.invokeSuspend(StateMachine.kt:160)
       at kotlin.coroutines.jvm.internal.BaseContinuationImpl.resumeWith(ContinuationImpl.kt:33)
       at kotlinx.coroutines.DispatchedTask.run(DispatchedTask.kt:106)
       at java.util.concurrent.Executors$RunnableAdapter.call(Executors.java:463)
       at java.util.concurrent.FutureTask.run(FutureTask.java:264)
       at java.util.concurrent.ScheduledThreadPoolExecutor$ScheduledFutureTask.run(ScheduledThreadPoolExecutor.java:307)
       at java.util.concurrent.ThreadPoolExecutor.runWorker(ThreadPoolExecutor.java:1137)
       at java.util.concurrent.ThreadPoolExecutor$Worker.run(ThreadPoolExecutor.java:637)
       at java.lang.Thread.run(Thread.java:1012)

I went to check the library code to see where this crash happening. I found it is happening in below code where continuation gets called multiple time,

    override suspend fun loadCredentials(credentialType: CredentialType): AmplifyCredential {
        return suspendCoroutine { continuation ->
            listenForResult(
                CredentialStoreEvent(CredentialStoreEvent.EventType.LoadCredentialStore(credentialType)),
                { continuation.resumeWith(it) },
                { continuation.resumeWithException(it) }
            )
        }
    }

Could you tell me what to do to prevent this error happening as it crashing my app.

Reproduction steps (if applicable)

No response

Code Snippet

// Put your code below this line.

Log output

// Put your logs below this line

amplifyconfiguration.json

No response

GraphQL Schema

// Put your schema below this line

Additional information and screenshots

No response