spring-security-oauth2自定义token返回格式

最新推荐文章于 2024-04-10 13:16:27 发布

javachen__

最新推荐文章于 2024-04-10 13:16:27 发布

阅读量1.3k

点赞数

文章标签： oauth2

本文链接：https://blog.csdn.net/u013008898/article/details/125466622

版权

该博客主要展示了如何在Spring Security OAuth2环境中自定义`AuthController`，重写`/oauth/token`和`/oauth/check_token`接口，以统一返回数据格式。通过`@Autowired`注入`TokenEndpoint`和`CheckTokenEndpoint`，实现了获取和校验OAuth2访问令牌的功能。

摘要由CSDN通过智能技术生成

AuthController:

package com.sdkj.security.oauth.controller;

import com.sdkj.security.oauth.entity.AjaxResult;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.oauth2.common.OAuth2AccessToken;
import org.springframework.security.oauth2.provider.endpoint.CheckTokenEndpoint;
import org.springframework.security.oauth2.provider.endpoint.TokenEndpoint;
import org.springframework.web.HttpRequestMethodNotSupportedException;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import java.security.Principal;
import java.util.Map;

@RestController
@RequestMapping("/oauth")
public class AuthController implements InitializingBean {

    //令牌请求的端点
    @Autowired
    private TokenEndpoint tokenEndpoint;

    @Autowired
    private CheckTokenEndpoint checkTokenEndpoint;

    /**
     * 重写/oauth/token这个默认接口，返回的数据格式统一
     */
    @PostMapping(value = "/token")
    public AjaxResult postAccessToken(Principal principal, @RequestParam
            Map<String, String> parameters) throws HttpRequestMethodNotSupportedException {
        OAuth2AccessToken accessToken = tokenEndpoint.postAccessToken(principal, parameters).getBody();
        return AjaxResult.success(accessToken);
    }

    /**
     * 重写/oauth/check_token这个默认接口，用于校验令牌，返回的数据格式统一
     */
    @PostMapping(value = "/check_token")
    public AjaxResult checkToken(@RequestParam("token") String value)  {
        Map<String, ?> map = checkTokenEndpoint.checkToken(value);
        return AjaxResult.success(map);
    }

    @Override
    public void afterPropertiesSet() throws Exception {

    }
}