mirrors/pig
1
0
Fork 0
mirror of https://gitee.com/log4j/pig.git synced 2024-12-23 05:00:23 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

🐛 fix #IWG40 用户退出没有删除refresh_token

Browse Source
This commit is contained in:
冷冷 2019-05-16 09:13:47 +08:00
parent 5d562e8555
commit 486816c88e
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pig-auth/src/main/java/com/pig4cloud/pig/auth/endpoint/PigTokenEndpoint.java
View File

@ -33,6 +33,7 @@ import org.springframework.http.HttpHeaders;
import org.springframework.security.authentication.UsernamePasswordAuthenticationToken; import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
import org.springframework.security.core.Authentication; import org.springframework.security.core.Authentication;
import org.springframework.security.oauth2.common.OAuth2AccessToken; import org.springframework.security.oauth2.common.OAuth2AccessToken;
import org.springframework.security.oauth2.common.OAuth2RefreshToken;
import org.springframework.security.oauth2.common.util.OAuth2Utils; import org.springframework.security.oauth2.common.util.OAuth2Utils;
import org.springframework.security.oauth2.provider.OAuth2Authentication; import org.springframework.security.oauth2.provider.OAuth2Authentication;
import org.springframework.security.oauth2.provider.token.TokenStore; import org.springframework.security.oauth2.provider.token.TokenStore;
@ -69,7 +70,7 @@ public class PigTokenEndpoint {
public R<Boolean> logout(@RequestHeader(value = HttpHeaders.AUTHORIZATION, required = false) String authHeader) { public R<Boolean> logout(@RequestHeader(value = HttpHeaders.AUTHORIZATION, required = false) String authHeader) {
if (StrUtil.isBlank(authHeader)) { if (StrUtil.isBlank(authHeader)) {
return R.<Boolean>builder() return R.<Boolean>builder()
.code(CommonConstants.FAIL) .code(CommonConstants.SUCCESS)
.data(Boolean.FALSE) .data(Boolean.FALSE)
.msg("退出失败，token 为空").build(); .msg("退出失败，token 为空").build();
} }
@ -78,11 +79,15 @@ public class PigTokenEndpoint {
OAuth2AccessToken accessToken = tokenStore.readAccessToken(tokenValue); OAuth2AccessToken accessToken = tokenStore.readAccessToken(tokenValue);
if (accessToken == null || StrUtil.isBlank(accessToken.getValue())) { if (accessToken == null || StrUtil.isBlank(accessToken.getValue())) {
return R.<Boolean>builder() return R.<Boolean>builder()
.code(CommonConstants.FAIL) .code(CommonConstants.SUCCESS)
.data(Boolean.FALSE) .data(Boolean.FALSE)
.msg("退出失败，token 无效").build(); .msg("退出失败，token 无效").build();
} }
tokenStore.removeAccessToken(accessToken); tokenStore.removeAccessToken(accessToken);
OAuth2RefreshToken refreshToken = tokenStore.readRefreshToken(tokenValue);
tokenStore.removeRefreshToken(refreshToken);
return R.<Boolean>builder() return R.<Boolean>builder()
.code(CommonConstants.SUCCESS) .code(CommonConstants.SUCCESS)
.data(Boolean.TRUE) .data(Boolean.TRUE)