设为首页 收藏本站
查看: 3066|回复: 0

[经验分享] 05.oauth2中RedisTokenStore使用redis cluster的问题

[复制链接]

尚未签到

发表于 2018-11-3 08:38:14 | 显示全部楼层 |阅读模式
private static final String ACCESS = "access:";  private static final String AUTH_TO_ACCESS = "auth_to_access:";
  private static final String AUTH = "auth:";
  private static final String REFRESH_AUTH = "refresh_auth:";
  private static final String ACCESS_TO_REFRESH = "access_to_refresh:";
  private static final String REFRESH = "refresh:";
  private static final String REFRESH_TO_ACCESS = "refresh_to_access:";
  private static final String CLIENT_ID_TO_ACCESS = "client_id_to_access:";
  private static final String UNAME_TO_ACCESS = "uname_to_access:";
  private RedisTemplate redisTemplate ;
  public RedisTemplate getRedisTemplate() {
  return redisTemplate;
  }
  public void setRedisTemplate(RedisTemplate redisTemplate) {
  this.redisTemplate = redisTemplate;
  }
  private AuthenticationKeyGenerator authenticationKeyGenerator = new DefaultAuthenticationKeyGenerator();
  public void setAuthenticationKeyGenerator(AuthenticationKeyGenerator authenticationKeyGenerator) {
  this.authenticationKeyGenerator = authenticationKeyGenerator;
  }
  public OAuth2AccessToken getAccessToken(OAuth2Authentication authentication) {
  String key = authenticationKeyGenerator.extractKey(authentication);
  OAuth2AccessToken accessToken = (OAuth2AccessToken) redisTemplate.opsForValue().get(AUTH_TO_ACCESS+key);
  if (accessToken != null
  && !key.equals(authenticationKeyGenerator.extractKey(readAuthentication(accessToken.getValue())))) {
  // Keep the stores consistent (maybe the same user is represented by this authentication but the details
  // have changed)
  storeAccessToken(accessToken, authentication);
  }
  return accessToken;
  }
  public OAuth2Authentication readAuthentication(OAuth2AccessToken token) {
  return readAuthentication(token.getValue());
  }
  public OAuth2Authentication readAuthentication(String token) {
  return (OAuth2Authentication) this.redisTemplate.opsForValue().get(AUTH +  token);
  }
  public OAuth2Authentication readAuthenticationForRefreshToken(OAuth2RefreshToken token) {
  return readAuthenticationForRefreshToken(token.getValue());
  }
  public OAuth2Authentication readAuthenticationForRefreshToken(String token) {
  return (OAuth2Authentication) this.redisTemplate.opsForValue().get( REFRESH_AUTH+token);
  }
  public void storeAccessToken(OAuth2AccessToken token, OAuth2Authentication authentication) {
  this.redisTemplate.opsForValue().set(ACCESS+ token.getValue(), token);
  this.redisTemplate.opsForValue().set(AUTH +token.getValue(), authentication);
  this.redisTemplate.opsForValue().set(AUTH_TO_ACCESS+authenticationKeyGenerator.extractKey(authentication), token);
  if (!authentication.isClientOnly()) {
  redisTemplate.opsForList().rightPush(UNAME_TO_ACCESS+getApprovalKey(authentication), token) ;
  }
  redisTemplate.opsForList().rightPush(CLIENT_ID_TO_ACCESS+authentication.getOAuth2Request().getClientId(), token) ;
  if (token.getExpiration() != null) {
  int seconds = token.getExpiresIn();
  redisTemplate.expire(ACCESS+ token.getValue(), seconds, TimeUnit.SECONDS) ;
  redisTemplate.expire(AUTH+ token.getValue(), seconds, TimeUnit.SECONDS) ;
  redisTemplate.expire(AUTH_TO_ACCESS+ authenticationKeyGenerator.extractKey(authentication), seconds, TimeUnit.SECONDS) ;
  redisTemplate.expire(CLIENT_ID_TO_ACCESS+authentication.getOAuth2Request().getClientId(), seconds, TimeUnit.SECONDS) ;
  redisTemplate.expire(UNAME_TO_ACCESS+ getApprovalKey(authentication), seconds, TimeUnit.SECONDS) ;
  }
  if (token.getRefreshToken() != null && token.getRefreshToken().getValue() != null) {
  this.redisTemplate.opsForValue().set( REFRESH_TO_ACCESS+   token.getRefreshToken().getValue(), token.getValue());
  this.redisTemplate.opsForValue().set(ACCESS_TO_REFRESH+token.getValue(), token.getRefreshToken().getValue());
  }
  }
  private String getApprovalKey(OAuth2Authentication authentication) {
  String userName = authentication.getUserAuthentication() == null ? "" : authentication.getUserAuthentication()
  .getName();
  return getApprovalKey(authentication.getOAuth2Request().getClientId(), userName);
  }
  private String getApprovalKey(String clientId, String userName) {
  return clientId + (userName==null ? "" : ":" + userName);
  }
  public void removeAccessToken(OAuth2AccessToken accessToken) {
  removeAccessToken(accessToken.getValue());
  }
  public OAuth2AccessToken readAccessToken(String tokenValue) {
  return (OAuth2AccessToken) this.redisTemplate.opsForValue().get(ACCESS+tokenValue);
  }
  public void removeAccessToken(String tokenValue) {
  OAuth2AccessToken removed = (OAuth2AccessToken) redisTemplate.opsForValue().get(ACCESS+tokenValue);
  // Don't remove the refresh token - it's up to the caller to do that
  OAuth2Authentication authentication = (OAuth2Authentication) this.redisTemplate.opsForValue().get(AUTH+tokenValue);
  this.redisTemplate.delete(AUTH+tokenValue);
  redisTemplate.delete(ACCESS+tokenValue);
  this.redisTemplate.delete(ACCESS_TO_REFRESH +tokenValue);
  if (authentication != null) {
  this.redisTemplate.delete(AUTH_TO_ACCESS+authenticationKeyGenerator.extractKey(authentication));
  String clientId = authentication.getOAuth2Request().getClientId();


运维网声明 1、欢迎大家加入本站运维交流群:群②:261659950 群⑤:202807635 群⑦870801961 群⑧679858003
2、本站所有主题由该帖子作者发表,该帖子作者与运维网享有帖子相关版权
3、所有作品的著作权均归原作者享有,请您和我们一样尊重他人的著作权等合法权益。如果您对作品感到满意,请购买正版
4、禁止制作、复制、发布和传播具有反动、淫秽、色情、暴力、凶杀等内容的信息,一经发现立即删除。若您因此触犯法律,一切后果自负,我们对此不承担任何责任
5、所有资源均系网友上传或者通过网络收集,我们仅提供一个展示、介绍、观摩学习的平台,我们不对其内容的准确性、可靠性、正当性、安全性、合法性等负责,亦不承担任何法律责任
6、所有作品仅供您个人学习、研究或欣赏,不得用于商业或者其他用途,否则,一切后果均由您自己承担,我们对此不承担任何法律责任
7、如涉及侵犯版权等问题,请您及时通知我们,我们将立即采取措施予以解决
8、联系人Email:admin@iyunv.com 网址:www.yunweiku.com

所有资源均系网友上传或者通过网络收集,我们仅提供一个展示、介绍、观摩学习的平台,我们不对其承担任何法律责任,如涉及侵犯版权等问题,请您及时通知我们,我们将立即处理,联系人Email:kefu@iyunv.com,QQ:1061981298 本贴地址:https://www.yunweiku.com/thread-630059-1-1.html 上篇帖子: Redis 3.2.8集群搭建 下篇帖子: Django框架设置session存储在redis数据库中
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

扫码加入运维网微信交流群X

扫码加入运维网微信交流群

扫描二维码加入运维网微信交流群,最新一手资源尽在官方微信交流群!快快加入我们吧...

扫描微信二维码查看详情

客服E-mail:kefu@iyunv.com 客服QQ:1061981298


QQ群⑦:运维网交流群⑦ QQ群⑧:运维网交流群⑧ k8s群:运维网kubernetes交流群


提醒:禁止发布任何违反国家法律、法规的言论与图片等内容;本站内容均来自个人观点与网络等信息,非本站认同之观点.


本站大部分资源是网友从网上搜集分享而来,其版权均归原作者及其网站所有,我们尊重他人的合法权益,如有内容侵犯您的合法权益,请及时与我们联系进行核实删除!



合作伙伴: 青云cloud

快速回复 返回顶部 返回列表