Skip to content

Commit 766b2ac

Browse files
committed
Alter cache key for AI Service tokens
1 parent 29cf5b7 commit 766b2ac

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

components/apimgt/org.wso2.carbon.apimgt.impl/src/main/java/org/wso2/carbon/apimgt/impl/utils/AccessTokenGenerator.java

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ public AccessTokenGenerator(String tokenEndpoint, String authKey) {
5252
}
5353

5454
public String getAccessToken() {
55-
AccessTokenInfo accessTokenInfo = accessTokenInfoMap.get(tokenEndpoint);
55+
AccessTokenInfo accessTokenInfo = accessTokenInfoMap.get(authKey);
5656
if (accessTokenInfo != null) {
5757
long expiryTime = accessTokenInfo.getIssuedTime() + accessTokenInfo.getValidityPeriod();
5858
long buffer = 20000;
@@ -61,9 +61,9 @@ public String getAccessToken() {
6161
if (log.isDebugEnabled()) {
6262
log.debug("Access token expired. New token requested");
6363
}
64-
accessTokenInfoMap.remove(tokenEndpoint);
64+
accessTokenInfoMap.remove(authKey);
6565
accessTokenInfo = generateNewAccessToken();
66-
accessTokenInfoMap.put(tokenEndpoint, accessTokenInfo);
66+
accessTokenInfoMap.put(authKey, accessTokenInfo);
6767
assert accessTokenInfo != null;
6868
return accessTokenInfo.getAccessToken();
6969
} else {
@@ -75,7 +75,7 @@ public String getAccessToken() {
7575
} else {
7676
accessTokenInfo = generateNewAccessToken();
7777
if (accessTokenInfo != null) {
78-
accessTokenInfoMap.put(tokenEndpoint, accessTokenInfo);
78+
accessTokenInfoMap.put(authKey, accessTokenInfo);
7979
return accessTokenInfo.getAccessToken();
8080
}
8181
}

0 commit comments

Comments
 (0)