温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

Spring Security OAuth2实现登录互踢的方法

发布时间:2020-07-29 11:28:39 来源:亿速云 阅读:329 作者:小猪 栏目:编程语言

这篇文章主要讲解了Spring Security OAuth2实现登录互踢的方法,内容清晰明了,对此有兴趣的小伙伴可以学习一下,相信大家阅读完之后会有帮助。

Spring Security OAuth2实现登录互踢的方法

背景说明

一个账号只能一处登录,类似的业务需求在现有后管类系统是非常常见的。 但在原有的 spring security oauth3 令牌方法流程(所谓的登录)无法满足类似的需求。

我们先来看 TokenEndpoint 的方法流程

客户端 带参访问 /oauth/token 接口,最后去调用 TokenGranter

Spring Security OAuth2实现登录互踢的方法

TokenGranter 根据不同的授权类型,获取用户认证信息 并去调用TokenServices 生成令牌

Spring Security OAuth2实现登录互踢的方法

重新 TokenService

重写发放逻辑createAccessToken,当用户管理的令牌存在时则删除重新创建,这样会导致之前登陆获取的token 失效,顺理成章的被挤掉。

@Transactional
  public OAuth3AccessToken createAccessToken(OAuth3Authentication authentication) throws AuthenticationException {

    OAuth3AccessToken existingAccessToken = tokenStore.getAccessToken(authentication);
    OAuth3RefreshToken refreshToken = null;
    // 重写此处,当用户关联的token 存在时,删除原有令牌
    if (existingAccessToken != null) {
      tokenStore.removeAccessToken(existingAccessToken);
    }
    else if (refreshToken instanceof ExpiringOAuth3RefreshToken) {
      ExpiringOAuth3RefreshToken expiring = (ExpiringOAuth3RefreshToken) refreshToken;
      if (System.currentTimeMillis() > expiring.getExpiration().getTime()) {
        refreshToken = createRefreshToken(authentication);
      }
    }

    OAuth3AccessToken accessToken = createAccessToken(authentication, refreshToken);
    tokenStore.storeAccessToken(accessToken, authentication);
    // In case it was modified
    refreshToken = accessToken.getRefreshToken();
    if (refreshToken != null) {
      tokenStore.storeRefreshToken(refreshToken, authentication);
    }
    return accessToken;
  }

重写 Token key 生成逻辑

如上代码,我们实现用户单一终端的唯一性登录,什么是单一终端 我们可以类比 QQ 登录 移动端和 PC 端可以同时登录,但 移动端 和移动端不能同时在线。

如何能够实现 在不同客户端也能够唯一性登录呢?

先来看上文源码 `OAuth3AccessToken existingAccessToken=tokenStore.getAccessToken(authentication);

是如何根据用户信息判断 token 存在的呢?

public OAuth3AccessToken getAccessToken(OAuth3Authentication authentication) {
    String key = authenticationKeyGenerator.extractKey(authentication);
     // redis 查询逻辑,根据 key
    return accessToken;
  
}

AuthenticationKeyGenerator key值生成器 默认情况下根据 username/clientId/scope 参数组合生成唯一token

public String extractKey(OAuth3Authentication authentication) {
  Map<String, String> values = new LinkedHashMap<String, String>();
  OAuth3Request authorizationRequest = authentication.getOAuth3Request();
  if (!authentication.isClientOnly()) {
    values.put(USERNAME, authentication.getName());
  }
  values.put(CLIENT_ID, authorizationRequest.getClientId());
  if (authorizationRequest.getScope() != null) {
    values.put(SCOPE, OAuth3Utils.formatParameterList(new TreeSet<String>(authorizationRequest.getScope())));
  }
  return generateKey(values);
}

若想实现,多终端的唯一性登录,只需要使得同一个用户在多个终端生成的 token 一致,加上上文提到的 createToken 修改逻辑,既去掉extractKey 的 clientId 条件,不区分终端即可

public String extractKey(OAuth3Authentication authentication) {
  Map<String, String> values = new LinkedHashMap<String, String>();
  OAuth3Request authorizationRequest = authentication.getOAuth3Request();
  if (!authentication.isClientOnly()) {
    values.put(USERNAME, authentication.getName());
  }
  if (authorizationRequest.getScope() != null) {
    values.put(SCOPE, OAuth3Utils.formatParameterList(new TreeSet<String>(authorizationRequest.getScope())));
  }
  return generateKey(values);
}

最后在 authserver 中注入新的 TokenService 即可

看完上述内容,是不是对Spring Security OAuth2实现登录互踢的方法有进一步的了解,如果还想学习更多内容,欢迎关注亿速云行业资讯频道。

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI