From 68ee3fb913c7ad95c31c447824125a002ea271bc Mon Sep 17 00:00:00 2001 From: Yi Liu Date: Thu, 12 Nov 2020 15:33:10 +0800 Subject: [PATCH] store authorzied clients into http session --- .../AzureOAuth2AuthorizedClientRepository.java | 14 ++++---------- 1 file changed, 4 insertions(+), 10 deletions(-) diff --git a/sdk/spring/azure-spring-boot/src/main/java/com/azure/spring/aad/implementation/AzureOAuth2AuthorizedClientRepository.java b/sdk/spring/azure-spring-boot/src/main/java/com/azure/spring/aad/implementation/AzureOAuth2AuthorizedClientRepository.java index 8312c7605b71..22aadf5ce133 100644 --- a/sdk/spring/azure-spring-boot/src/main/java/com/azure/spring/aad/implementation/AzureOAuth2AuthorizedClientRepository.java +++ b/sdk/spring/azure-spring-boot/src/main/java/com/azure/spring/aad/implementation/AzureOAuth2AuthorizedClientRepository.java @@ -4,10 +4,8 @@ package com.azure.spring.aad.implementation; import org.springframework.security.core.Authentication; -import org.springframework.security.oauth2.client.InMemoryOAuth2AuthorizedClientService; import org.springframework.security.oauth2.client.OAuth2AuthorizedClient; -import org.springframework.security.oauth2.client.registration.ClientRegistrationRepository; -import org.springframework.security.oauth2.client.web.AuthenticatedPrincipalOAuth2AuthorizedClientRepository; +import org.springframework.security.oauth2.client.web.HttpSessionOAuth2AuthorizedClientRepository; import org.springframework.security.oauth2.client.web.OAuth2AuthorizedClientRepository; import org.springframework.security.oauth2.core.OAuth2AccessToken; import org.springframework.security.oauth2.core.OAuth2RefreshToken; @@ -23,16 +21,12 @@ public class AzureOAuth2AuthorizedClientRepository implements OAuth2AuthorizedCl private final AzureClientRegistrationRepository azureClientRegistrationRepository; private final OAuth2AuthorizedClientRepository delegatedOAuth2AuthorizedClientRepository; - private static OAuth2AuthorizedClientRepository createDefaultDelegate( - ClientRegistrationRepository clientRegistrationRepository - ) { - return new AuthenticatedPrincipalOAuth2AuthorizedClientRepository( - new InMemoryOAuth2AuthorizedClientService(clientRegistrationRepository) - ); + private static OAuth2AuthorizedClientRepository createDefaultDelegate() { + return new HttpSessionOAuth2AuthorizedClientRepository(); } public AzureOAuth2AuthorizedClientRepository(AzureClientRegistrationRepository azureClientRegistrationRepository) { - this(azureClientRegistrationRepository, createDefaultDelegate(azureClientRegistrationRepository)); + this(azureClientRegistrationRepository, createDefaultDelegate()); } public AzureOAuth2AuthorizedClientRepository(