Spring Boot下自定义OAuth2+JWT令牌响应的可行性问询

完全可以实现你需要的自定义响应格式，下面是具体的实现方案：

核心思路 #

通过重写OAuth2令牌端点（/oauth/token）的处理逻辑，将默认响应数据包装到data字段中，同时添加自定义的details字段存放额外信息。

步骤1：自定义令牌端点处理类 #

创建一个继承自TokenEndpoint的自定义类，重写postAccessToken方法来构建你需要的响应结构：

import org.springframework.security.oauth2.provider.endpoint.TokenEndpoint;
import org.springframework.security.oauth2.provider.token.DefaultTokenServices;
import org.springframework.http.ResponseEntity;
import java.security.Principal;
import java.util.HashMap;
import java.util.Map;

public class CustomTokenEndpoint extends TokenEndpoint {

    private DefaultTokenServices tokenServices;

    @Override
    public ResponseEntity<Map<String, Object>> postAccessToken(Principal principal, Map<String, String> parameters) throws Exception {
        // 调用父类方法获取默认的令牌响应
        ResponseEntity<Map<String, Object>> originalResponse = super.postAccessToken(principal, parameters);
        Map<String, Object> originalTokenData = originalResponse.getBody();
        
        // 构建自定义响应结构
        Map<String, Object> customResponse = new HashMap<>();
        // 将原有令牌数据放入data字段
        customResponse.put("data", originalTokenData);
        
        // 构建details字段，可根据实际需求填充额外信息
        Map<String, Object> details = new HashMap<>();
        // 示例：从Principal获取当前登录用户名
        details.put("username", principal.getName());
        // 可扩展更多字段，比如用户ID、角色、昵称等
        details.put("user_id", 1001);
        details.put("roles", new String[]{"admin", "user"});
        
        customResponse.put("details", details);
        
        // 返回自定义响应，保持原有响应状态码
        return ResponseEntity.status(originalResponse.getStatusCode()).body(customResponse);
    }

    @Override
    public void setTokenServices(DefaultTokenServices tokenServices) {
        this.tokenServices = tokenServices;
        super.setTokenServices(tokenServices);
    }
}

步骤2：配置Spring Security替换默认端点 #

在你的OAuth2配置类中，注册自定义的TokenEndpoint Bean，替换默认的端点实现：

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.security.oauth2.provider.endpoint.TokenEndpoint;
import org.springframework.security.oauth2.provider.token.DefaultTokenServices;

@Configuration
public class OAuth2Configuration {

    @Bean
    public TokenEndpoint tokenEndpoint(DefaultTokenServices tokenServices) {
        CustomTokenEndpoint customEndpoint = new CustomTokenEndpoint();
        customEndpoint.setTokenServices(tokenServices);
        // 若需要，可设置clientDetailsService等其他依赖
        return customEndpoint;
    }
}

注意事项 #

• details字段的内容可以根据业务需求灵活扩展，比如从数据库查询用户的详细信息、权限列表等；
• 如果使用的是Spring Security OAuth2.0（而非OAuth2.1），上述代码完全适用；
• 确保自定义端点的依赖注入正确，比如tokenServices和clientDetailsService等核心组件。

内容的提问来源于stack exchange，提问作者Naidu