Skip to content

Commit f1b60a3

Browse files
committed
Add modifyCredentials to invalidateClientRegistration
1 parent 0812a38 commit f1b60a3

File tree

1 file changed

+18
-3
lines changed

1 file changed

+18
-3
lines changed

plugins/core/jetbrains-community/src/software/aws/toolkits/jetbrains/core/credentials/sso/SsoAccessTokenProvider.kt

Lines changed: 18 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ import software.aws.toolkits.jetbrains.utils.sleepWithCancellation
2828
import software.aws.toolkits.resources.AwsCoreBundle
2929
import software.aws.toolkits.telemetry.AuthType
3030
import software.aws.toolkits.telemetry.AwsTelemetry
31+
import software.aws.toolkits.telemetry.CredentialModification
3132
import software.aws.toolkits.telemetry.CredentialSourceId
3233
import software.aws.toolkits.telemetry.Result
3334
import java.time.Clock
@@ -505,10 +506,24 @@ class SsoAccessTokenProvider(
505506
)
506507
}
507508

508-
509509
private fun invalidateClientRegistration() {
510-
cache.invalidateClientRegistration(dagClientRegistrationCacheKey)
511-
cache.invalidateClientRegistration(pkceClientRegistrationCacheKey)
510+
try {
511+
cache.invalidateClientRegistration(dagClientRegistrationCacheKey)
512+
cache.invalidateClientRegistration(pkceClientRegistrationCacheKey)
513+
} catch (e: Exception) {
514+
AwsTelemetry.modifyCredentials(
515+
credentialModification = CredentialModification.Delete,
516+
result = Result.Failed,
517+
reason = "Failed to invalidate client registration",
518+
reasonDesc = e.message,
519+
source = "SsoAccessTokenProvider.invalidateClientRegistration"
520+
)
521+
}
522+
AwsTelemetry.modifyCredentials(
523+
credentialModification = CredentialModification.Delete,
524+
result = Result.Succeeded,
525+
source = "SsoAccessTokenProvider.invalidateClientRegistration"
526+
)
512527
}
513528

514529
private fun saveAccessToken(token: AccessToken) {

0 commit comments

Comments
 (0)