admin/MaxKey
1
0
Fork 0
mirror of https://gitee.com/dromara/MaxKey.git synced 2025-12-07 17:38:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

!9 修复redis刷新token报错

Browse Source 
Merge pull request !9 from when/master
This commit is contained in:
MaxKeyTop 2022-03-04 00:59:13 +00:00 committed by Gitee
commit 85c0c2dbc1
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 18 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
maxkey-core/src/main/java/org/maxkey/persistence/redis/RedisConnection.java
View File

@ -147,5 +147,10 @@ public class RedisConnection {
public void setConn(Jedis conn) {
this.conn = conn;
}
public Pipeline getPipeline() {
return pipeline;
}
}

22
maxkey-protocols/maxkey-protocol-oauth-2.0/src/main/java/org/maxkey/authz/oauth2/provider/token/store/RedisTokenStore.java
View File

@ -35,6 +35,7 @@ import org.maxkey.util.ObjectTransformer;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.Date;
/**
@ -216,14 +217,15 @@ public class RedisTokenStore implements TokenStore {
RedisConnection conn = getConnection();
try {
conn.openPipeline();
conn.delete(accessKey);
conn.delete(accessToRefreshKey);
// Don't remove the refresh token - it's up to the caller to do that
conn.delete(authKey);
conn.getPipeline().get(accessKey);
conn.getPipeline().get(authKey);
conn.getPipeline().del(accessKey);
conn.getPipeline().del(accessToRefreshKey);
//Don't remove the refresh token - it's up to the caller to do that
conn.getPipeline().del(authKey);
List<Object> results = conn.closePipeline();
String access = (String) results.get(0);
String auth = (String) results.get(1);
OAuth2Authentication authentication = ObjectTransformer.deserialize(auth);
if (authentication != null) {
String key = authenticationKeyGenerator.extractKey(authentication);
@ -251,6 +253,7 @@ public class RedisTokenStore implements TokenStore {
conn.openPipeline();
conn.setObject(refreshKey, refreshToken);
conn.setObject(refreshAuthKey, authentication);
if (refreshToken instanceof ExpiringOAuth2RefreshToken) {
ExpiringOAuth2RefreshToken expiringRefreshToken = (ExpiringOAuth2RefreshToken) refreshToken;
Date expiration = expiringRefreshToken.getExpiration();
@ -314,8 +317,8 @@ public class RedisTokenStore implements TokenStore {
RedisConnection conn = getConnection();
try {
conn.openPipeline();
conn.get(key);
conn.delete(key);
conn.getPipeline().get(key);
conn.getPipeline().del(key);
results = conn.closePipeline();
} finally {
conn.close();
@ -323,11 +326,12 @@ public class RedisTokenStore implements TokenStore {
if (results == null) {
return;
}
String bytes = (String) results.get(0);
String accessToken = ObjectTransformer.deserialize(bytes);
String accessToken = (String) results.get(0);
//String accessToken = ObjectTransformer.deserialize(bytes);
if (accessToken != null) {
removeAccessToken(accessToken);
}
}
@Override