oauth2 自定义 令牌增强 该怎么实现

要实现OAuth2自定义令牌增强，您可以遵循以下步骤：

1. 创建一个自定义的TokenEnhancer类，实现Spring Security的TokenEnhancer接口。这个类将负责生成或修改令牌中的附加信息。

2. 在您的授权服务器配置类中，通过重写configure(AuthorizationServerEndpointsConfigurer endpoints)方法，将自定义的TokenEnhancer添加到TokenEnhancerChain中。

   @Configuration
   @EnableAuthorizationServer
   public class AuthorizationServerConfig extends AuthorizationServerConfigurerAdapter {

       @Autowired
       private TokenEnhancer tokenEnhancer;

       @Override
       public void configure(AuthorizationServerEndpointsConfigurer endpoints) {
           TokenEnhancerChain enhancerChain = new TokenEnhancerChain();
           enhancerChain.setTokenEnhancers(Arrays.asList(tokenEnhancer));
           endpoints.tokenEnhancer(enhancerChain);
       }

       // 其他配置...
   }

3. 在自定义的TokenEnhancer类中，实现enhance(TokenRequest tokenRequest, OAuth2AccessToken accessToken)方法，根据需要修改或添加令牌的附加信息。

   @Component
   public class CustomTokenEnhancer implements TokenEnhancer {

       @Override
       public OAuth2AccessToken enhance(OAuth2AccessToken accessToken, TokenRequest tokenRequest) {
           // 获取当前用户信息或其他需要的信息
           // 修改或添加令牌的附加信息
           Map<String, Object> additionalInfo = new HashMap<>();
           additionalInfo.put("custom_key", "custom_value");
           ((DefaultOAuth2AccessToken) accessToken).setAdditionalInformation(additionalInfo);

           return accessToken;
       }
   }

4. 在授权服务器配置类中，通过重写configure(ClientDetailsServiceConfigurer clients)方法，设置您的客户端详情服务，包括客户端ID、密钥、授权类型等。

   @Configuration
   @EnableAuthorizationServer
   public class AuthorizationServerConfig extends AuthorizationServerConfigurerAdapter {

       // 其他配置...

       @Override
       public void configure(ClientDetailsServiceConfigurer clients) throws Exception {
           clients.inMemory()
                   .withClient("client_id")
                   .secret("client_secret")
                   .authorizedGrantTypes("authorization_code", "refresh_token")
                   .scopes("scope")
                   .accessTokenValiditySeconds(3600);
       }
   }

这样，当您的客户端通过授权服务器进行认证和授权时，自定义的TokenEnhancer将会被调用，从而实现自定义令牌增强。请根据您的具体需求，对自定义的TokenEnhancer进行进一步的实现。