使用TokenEnhancer类怎样实现授权码模式

在Spring Security中，TokenEnhancer是一个接口，它允许你在每次认证成功后对JWT（JSON Web Token）进行自定义增强。实现授权码模式通常涉及到OAuth 2.0流程，其中服务器首先返回一个授权码给用户，用户随后使用这个授权码换取访问令牌。在使用TokenEnhancer时，可以按以下步骤操作：

1. 首先，创建一个实现了`TokenEnhancer`接口的类。例如，你可以创建一个名为`MyCustomTokenEnhancer`的类：

import org.springframework.security.core.Authentication;
import org.springframework.security.oauth2.common.OAuth2AccessToken;
import org.springframework.security.oauth2.provider.token.DefaultTokenServices;

public class MyCustomTokenEnhancer implements TokenEnhancer<OAuth2AccessToken> {
    // 添加必要的字段和方法，比如处理额外信息，如用户角色、有效期等
}

2. 在类中实现必要的逻辑，比如添加额外的声明（claims），设置访问令牌的有效期等。例如：

@Override
protected OAuth2AccessToken enhance(OAuth2AccessToken accessToken, Authentication authentication) {
    DefaultTokenServices tokenServices = new DefaultTokenServices();
    // 设置访问令牌的额外属性
    accessToken.setAdditionalInformation("customClaim", "someValue");
    
    // 如果需要，更新令牌过期时间
    Date expiration = tokenServices.getTokenExpiration(accessToken);
    accessToken.setExpiresAt(expiration);
    
    return accessToken;
}

3. 将`TokenEnhancer`应用到你的配置中。在`SecurityConfiguration`或`WebSecurityConfigurerAdapter`中注册你的自定义`TokenEnhancer`：

@Autowired
private MyCustomTokenEnhancer myCustomTokenEnhancer;

@Bean
public TokenEnhancerChain customTokenEnhancer() {
    List<TokenEnhancer<?>> enhancers = new ArrayList<>();
    enhancers.add(myCustomTokenEnhancer); // 把你的TokenEnhancer添加到列表中
    TokenEnhancerChain chain = new TokenEnhancerChain("enhancer");
    chain.setTokenEnhancers(enhancers);
    return chain;
}