forgetest5
/
ifffff
mirror of https://gitee.com/maxjhandsome/pig
1
0
Fork 0
Code Issues Releases Wiki Activity

fix: 确保RedisConnection总是关闭

This commit is contained in:
cj 2020-08-03 18:38:40 +08:00
parent 2159e746cf
commit bc1ffd321c
1 changed files with 15 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pig-common/pig-common-security/src/main/java/com/pig4cloud/pig/common/security/component/PigRedisTokenStore.java
View File

@ -33,16 +33,15 @@ public class PigRedisTokenStore extends RedisTokenStore {
@Override
public void storeAccessToken(OAuth2AccessToken token, OAuth2Authentication authentication) {
super.storeAccessToken(token, authentication);
try (RedisConnection connection = connectionFactory.getConnection()) {
// KEY
byte[] key = StrUtil.bytes(CacheConstants.PROJECT_OAUTH_TOKEN + authentication.getName());
// value
byte[] tokenVal = StrUtil.bytes(token.getValue());
// 获取redis连接
RedisConnection connection = connectionFactory.getConnection();
RedisStringCommands stringCommand = connection.stringCommands();
stringCommand.set(key, tokenVal, Expiration.seconds(token.getExpiresIn()),
RedisStringCommands.SetOption.SET_IF_ABSENT);
connection.close();
}
}
/**
@ -52,13 +51,12 @@ public class PigRedisTokenStore extends RedisTokenStore {
@Override
public void removeAccessToken(OAuth2AccessToken accessToken) {
super.removeAccessToken(accessToken);
try (RedisConnection connection = connectionFactory.getConnection()) {
// KEY
OAuth2Authentication authentication = readAuthentication(accessToken);
byte[] key = StrUtil.bytes(CacheConstants.PROJECT_OAUTH_TOKEN + authentication.getName());
// 获取redis连接
RedisConnection connection = connectionFactory.getConnection();
connection.del(key);
connection.close();
}
}
}