不使用HttpSecurity,使用Spring Boot访问OAuth2受保护资源的方法是什么?
创始人
2024-12-28 20:01:04
0

使用Spring Boot访问OAuth2受保护资源的方法是通过使用OAuth2RestTemplate来发送OAuth2认证请求和访问受保护资源。

下面是一个使用OAuth2RestTemplate的代码示例:

import org.springframework.boot.autoconfigure.security.oauth2.client.OAuth2ClientProperties;
import org.springframework.boot.autoconfigure.security.oauth2.client.OAuth2ClientProperties.Provider;
import org.springframework.boot.autoconfigure.security.oauth2.client.OAuth2ClientProperties.Registration;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.RequestEntity;
import org.springframework.http.ResponseEntity;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.security.oauth2.client.OAuth2RestTemplate;
import org.springframework.security.oauth2.common.OAuth2AccessToken;
import org.springframework.security.oauth2.provider.authentication.OAuth2AuthenticationDetails;
import org.springframework.security.oauth2.provider.authentication.OAuth2AuthenticationToken;
import org.springframework.security.oauth2.provider.token.ResourceServerTokenServices;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;
import org.springframework.web.client.RestClientException;

import java.net.URI;
import java.util.Base64;

public class OAuth2ResourceClient {

    private final OAuth2RestTemplate restTemplate;
    private final ResourceServerTokenServices tokenServices;
    private final OAuth2ClientProperties clientProperties;

    public OAuth2ResourceClient(OAuth2RestTemplate restTemplate, ResourceServerTokenServices tokenServices, OAuth2ClientProperties clientProperties) {
        this.restTemplate = restTemplate;
        this.tokenServices = tokenServices;
        this.clientProperties = clientProperties;
    }

    public ResponseEntity getResource(String url) {
        HttpHeaders headers = new HttpHeaders();
        headers.add("Authorization", "Bearer " + getAccessToken());

        RequestEntity requestEntity = new RequestEntity<>(headers, HttpMethod.GET, URI.create(url));

        try {
            return restTemplate.exchange(requestEntity, String.class);
        } catch (RestClientException e) {
            throw new RuntimeException("Error accessing resource", e);
        }
    }

    private String getAccessToken() {
        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
        if (authentication instanceof OAuth2AuthenticationToken) {
            OAuth2AuthenticationToken oauthToken = (OAuth2AuthenticationToken) authentication;
            OAuth2AccessToken token = tokenServices.readAccessToken(oauthToken.getAccessToken().getTokenValue());
            if (token.isExpired()) {
                token = tokenServices.refreshAccessToken(token.getRefreshToken().getValue());
            }
            return token.getValue();
        }
        throw new IllegalStateException("No OAuth2 authentication found");
    }

    public static OAuth2RestTemplate createOAuth2RestTemplate(OAuth2ClientProperties clientProperties, ResourceServerTokenServices tokenServices) {
        Registration registration = clientProperties.getRegistration().values().iterator().next();
        Provider provider = clientProperties.getProvider().get(registration.getProvider());

        String clientId = registration.getClientId();
        String clientSecret = registration.getClientSecret();
        String accessTokenUri = provider.getTokenUri();

        OAuth2RestTemplate restTemplate = new OAuth2RestTemplate(clientId, clientSecret);
        restTemplate.setAccessTokenProvider(new CustomAccessTokenProvider(accessTokenUri));
        restTemplate.setAccessTokenConverter(new CustomAccessTokenConverter());
        restTemplate.setAuthenticator(new CustomOAuth2RequestAuthenticator());

        return restTemplate;
    }

    private static class CustomAccessTokenProvider extends OAuth2AccessTokenProviderChain {
        public CustomAccessTokenProvider(String accessTokenUri) {
            super(Arrays.asList(new CustomAuthorizationCodeAccessTokenProvider(accessTokenUri),
                    new CustomImplicitAccessTokenProvider(accessTokenUri),
                    new CustomClientCredentialsAccessTokenProvider(accessTokenUri)));
        }
    }

    private static class CustomAuthorizationCodeAccessTokenProvider extends AuthorizationCodeAccessTokenProvider {
        public CustomAuthorizationCodeAccessTokenProvider(String accessTokenUri) {
            this.setTokenRequestEnhancer(new CustomTokenRequestEnhancer(accessTokenUri));
        }
    }

    private static class CustomImplicitAccessTokenProvider extends ImplicitAccessTokenProvider {
        public CustomImplicitAccessTokenProvider(String accessTokenUri) {
            this.setTokenRequestEnhancer(new CustomTokenRequestEnhancer(accessTokenUri));
        }
    }

    private static class CustomClientCredentialsAccessTokenProvider extends ClientCredentialsAccessTokenProvider {
        public CustomClientCredentialsAccessTokenProvider(String accessTokenUri) {
            this.setTokenRequestEnhancer(new CustomTokenRequestEnhancer(accessTokenUri));
        }
    }

    private static class CustomTokenRequestEnhancer implements RequestEnhancer {

        private final String accessTokenUri;

        public CustomTokenRequestEnhancer(String accessTokenUri) {
            this.accessTokenUri = accessTokenUri;
        }

        @Override
        public void enhance(AccessTokenRequest request, OAuth2ProtectedResourceDetails resource,
                            MultiValueMap form, HttpHeaders headers) {
            headers.setBasicAuth(resource.getClientId(), resource.getClientSecret());
            form.set("grant_type", "client_credentials");
            form.set("scope", "read");
        }

相关内容

热门资讯

前端-session、jwt 目录:   (1)session (2&#x...
linux入门---制作进度条 了解缓冲区 我们首先来看看下面的操作: 我们首先创建了一个文件并在这个文件里面添加了...
关于测试,我发现了哪些新大陆 关于测试 平常也只是听说过一些关于测试的术语,但并没有使用过测试工具。偶然看到编程老师...
前缀和与对数器与二分法 1. 前缀和 假设有一个数组,我们想大量频繁的去访问L到R这个区间的和,...
nodejs:本地安装nvm实... 一、背景-使用不同版本node的原因 vue3+ts、nuxt3版本,node...
JAVA集合知识整理 Java集合知识整理 HashMap相关 HashMap的底层数据结构:jdk1.8之...
无刷直流电机介绍及单片机控制实... 无刷直流电机介绍及单片机控制实例前言基本概念优势与劣势使用寿命基本结构使用单片机控制实例电子调速器&...
fwdiary(2) dp2 1.传纸条  AcWing 275. 传纸条 - AcWing 走两条路,走一条最大的...
常用的DOS命令 常用的DOS命令 DOS(Disk Operating System,磁...
<C++> 类和对象(下) 1.const成员函数将const修饰的“成员函数”称之为const成员函数,cons...