package com.ruoyi.auth.controller;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.http.HttpHeaders;
|
import org.springframework.security.oauth2.common.OAuth2AccessToken;
|
import org.springframework.security.oauth2.common.OAuth2RefreshToken;
|
import org.springframework.security.oauth2.provider.token.TokenStore;
|
import org.springframework.web.bind.annotation.DeleteMapping;
|
import org.springframework.web.bind.annotation.RequestHeader;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RestController;
|
import com.ruoyi.common.core.domain.R;
|
import com.ruoyi.common.core.utils.StringUtils;
|
|
/**
|
* token 控制
|
*
|
* @author ruoyi
|
*/
|
@RestController
|
@RequestMapping("/token")
|
public class TokenController
|
{
|
@Autowired
|
private TokenStore tokenStore;
|
|
@DeleteMapping("/logout")
|
public R<?> logout(@RequestHeader(value = HttpHeaders.AUTHORIZATION, required = false) String authHeader)
|
{
|
if (StringUtils.isEmpty(authHeader))
|
{
|
return R.ok();
|
}
|
|
String tokenValue = authHeader.replace(OAuth2AccessToken.BEARER_TYPE, StringUtils.EMPTY).trim();
|
OAuth2AccessToken accessToken = tokenStore.readAccessToken(tokenValue);
|
if (accessToken == null || StringUtils.isEmpty(accessToken.getValue()))
|
{
|
return R.ok();
|
}
|
|
// 清空 access token
|
tokenStore.removeAccessToken(accessToken);
|
|
// 清空 refresh token
|
OAuth2RefreshToken refreshToken = accessToken.getRefreshToken();
|
tokenStore.removeRefreshToken(refreshToken);
|
return R.ok();
|
}
|
}
|
