소스 검색

修复token错误

zhouhao 7 년 전
부모
커밋
3260a34bdd

+ 6 - 6
hsweb-authorization/hsweb-authorization-oauth2/hsweb-authorization-oauth2-client/src/main/java/org/hswebframework/web/authorization/oauth2/client/simple/SimpleOAuth2SessionBuilder.java

@@ -70,14 +70,14 @@ public class SimpleOAuth2SessionBuilder implements OAuth2SessionBuilder {
         return token -> {
             AccessTokenInfo tokenInfo = tokenGetter.get();
             if (tokenInfo != null) {
+                token.setId(tokenInfo.getId());
                 tokenInfo.setUpdateTime(System.currentTimeMillis());
-                oAuth2UserTokenRepository.update(tokenInfo.getId(), tokenInfo);
+                oAuth2UserTokenRepository.update(tokenInfo.getId(), token);
             } else {
-                tokenInfo = oAuth2UserTokenRepository.createToken();
-                tokenInfo.setGrantType(grantType);
-                tokenInfo.setCreateTime(System.currentTimeMillis());
-                tokenInfo.setServerId(serverConfig.getId());
-                oAuth2UserTokenRepository.insert(tokenInfo);
+                token.setGrantType(grantType);
+                token.setCreateTime(System.currentTimeMillis());
+                token.setServerId(serverConfig.getId());
+                oAuth2UserTokenRepository.insert(token);
             }
         };
     }