Cwikius-Spring/spring-security
1
0
Fork 0
mirror of https://github.com/spring-projects/spring-security.git synced 2025-06-13 23:52:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make InMemory*ClientRegistrationRepository Consistent

Browse Source 
The previous builders with the list argument were inconsistent with their 
respective builders of var args.
This commit is contained in:
dperezcabrera 2018-11-16 22:18:09 +01:00 committed by Rob Winch
parent a742c0c3f2
commit 8014114225
4 changed files with 32 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
oauth2/oauth2-client/src/main/java/org/springframework/security/oauth2/client/registration/InMemoryClientRegistrationRepository.java
View File

@ -18,16 +18,13 @@ package org.springframework.security.oauth2.client.registration;
import org.springframework.util.Assert; import org.springframework.util.Assert;
import java.util.Arrays; import java.util.Arrays;
import java.util.Collection;
import java.util.Collections; import java.util.Collections;
import java.util.Iterator; import java.util.Iterator;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.concurrent.ConcurrentMap;
import java.util.function.Function; import java.util.function.Function;
import java.util.stream.Collector; import java.util.stream.Collectors;
import static java.util.stream.Collectors.collectingAndThen;
import static java.util.stream.Collectors.toConcurrentMap;
/** /**
* A {@link ClientRegistrationRepository} that stores {@link ClientRegistration}(s) in-memory. * A {@link ClientRegistrationRepository} that stores {@link ClientRegistration}(s) in-memory.
@ -39,6 +36,7 @@ import static java.util.stream.Collectors.toConcurrentMap;
* @see ClientRegistration * @see ClientRegistration
*/ */
public final class InMemoryClientRegistrationRepository implements ClientRegistrationRepository, Iterable<ClientRegistration> { public final class InMemoryClientRegistrationRepository implements ClientRegistrationRepository, Iterable<ClientRegistration> {
private final Map<String, ClientRegistration> registrations; private final Map<String, ClientRegistration> registrations;
/** /**
@ -47,7 +45,8 @@ public final class InMemoryClientRegistrationRepository implements ClientRegistr
* @param registrations the client registration(s) * @param registrations the client registration(s)
*/ */
public InMemoryClientRegistrationRepository(ClientRegistration... registrations) { public InMemoryClientRegistrationRepository(ClientRegistration... registrations) {
this(Arrays.asList(registrations)); Assert.notEmpty(registrations, "registrations cannot be empty");
this.registrations = createClientRegistrationIdToClientRegistration(Arrays.asList(registrations));
} }
/** /**
@ -57,10 +56,13 @@ public final class InMemoryClientRegistrationRepository implements ClientRegistr
*/ */
public InMemoryClientRegistrationRepository(List<ClientRegistration> registrations) { public InMemoryClientRegistrationRepository(List<ClientRegistration> registrations) {
Assert.notEmpty(registrations, "registrations cannot be empty"); Assert.notEmpty(registrations, "registrations cannot be empty");
Collector<ClientRegistration, ?, ConcurrentMap<String, ClientRegistration>> collector = this.registrations = createClientRegistrationIdToClientRegistration(registrations);
toConcurrentMap(ClientRegistration::getRegistrationId, Function.identity()); }
this.registrations = registrations.stream()
.collect(collectingAndThen(collector, Collections::unmodifiableMap)); private static Map<String, ClientRegistration> createClientRegistrationIdToClientRegistration(Collection<ClientRegistration> registrations) {
return Collections.unmodifiableMap(registrations.stream()
.peek(registration -> Assert.notNull(registration, "registrations cannot contain null values"))
.collect(Collectors.toMap(ClientRegistration::getRegistrationId, Function.identity())));
} }
@Override @Override

24
oauth2/oauth2-client/src/main/java/org/springframework/security/oauth2/client/registration/InMemoryReactiveClientRegistrationRepository.java
View File

@ -17,12 +17,6 @@ package org.springframework.security.oauth2.client.registration;
import java.util.Iterator; import java.util.Iterator;
import java.util.List; import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;
import org.springframework.util.Assert;
import org.springframework.util.ConcurrentReferenceHashMap;
import reactor.core.publisher.Mono; import reactor.core.publisher.Mono;
@ -37,7 +31,7 @@ import reactor.core.publisher.Mono;
public final class InMemoryReactiveClientRegistrationRepository public final class InMemoryReactiveClientRegistrationRepository
implements ReactiveClientRegistrationRepository, Iterable<ClientRegistration> { implements ReactiveClientRegistrationRepository, Iterable<ClientRegistration> {
private final Map<String, ClientRegistration> clientIdToClientRegistration; private final InMemoryClientRegistrationRepository delegate;
/** /**
* Constructs an {@code InMemoryReactiveClientRegistrationRepository} using the provided parameters. * Constructs an {@code InMemoryReactiveClientRegistrationRepository} using the provided parameters.
@ -45,12 +39,7 @@ public final class InMemoryReactiveClientRegistrationRepository
* @param registrations the client registration(s) * @param registrations the client registration(s)
*/ */
public InMemoryReactiveClientRegistrationRepository(ClientRegistration... registrations) { public InMemoryReactiveClientRegistrationRepository(ClientRegistration... registrations) {
Assert.notEmpty(registrations, "registrations cannot be empty"); this.delegate = new InMemoryClientRegistrationRepository(registrations);
this.clientIdToClientRegistration = new ConcurrentReferenceHashMap<>();
for (ClientRegistration registration : registrations) {
Assert.notNull(registration, "registrations cannot contain null values");
this.clientIdToClientRegistration.put(registration.getRegistrationId(), registration);
}
} }
/** /**
@ -59,15 +48,12 @@ public final class InMemoryReactiveClientRegistrationRepository
* @param registrations the client registration(s) * @param registrations the client registration(s)
*/ */
public InMemoryReactiveClientRegistrationRepository(List<ClientRegistration> registrations) { public InMemoryReactiveClientRegistrationRepository(List<ClientRegistration> registrations) {
Assert.notEmpty(registrations, "registrations cannot be null or empty"); this.delegate = new InMemoryClientRegistrationRepository(registrations);
this.clientIdToClientRegistration = registrations.stream()
.collect(Collectors.toConcurrentMap(ClientRegistration::getRegistrationId, Function.identity()));
} }
@Override @Override
public Mono<ClientRegistration> findByRegistrationId(String registrationId) { public Mono<ClientRegistration> findByRegistrationId(String registrationId) {
return Mono.justOrEmpty(this.clientIdToClientRegistration.get(registrationId)); return Mono.justOrEmpty(this.delegate.findByRegistrationId(registrationId));
} }
/** /**
@ -77,6 +63,6 @@ public final class InMemoryReactiveClientRegistrationRepository
*/ */
@Override @Override
public Iterator<ClientRegistration> iterator() { public Iterator<ClientRegistration> iterator() {
return this.clientIdToClientRegistration.values().iterator(); return delegate.iterator();
} }
} }

6
oauth2/oauth2-client/src/test/java/org/springframework/security/oauth2/client/registration/InMemoryClientRegistrationRepositoryTests.java
View File

@ -35,6 +35,12 @@ public class InMemoryClientRegistrationRepositoryTests {
private InMemoryClientRegistrationRepository clients = new InMemoryClientRegistrationRepository(this.registration); private InMemoryClientRegistrationRepository clients = new InMemoryClientRegistrationRepository(this.registration);
@Test(expected = IllegalArgumentException.class)
public void constructorVarArgsListClientRegistrationWhenNullThenIllegalArgumentException() {
ClientRegistration nullRegistration = null;
new InMemoryClientRegistrationRepository(nullRegistration);
}
@Test(expected = IllegalArgumentException.class) @Test(expected = IllegalArgumentException.class)
public void constructorListClientRegistrationWhenNullThenIllegalArgumentException() { public void constructorListClientRegistrationWhenNullThenIllegalArgumentException() {
List<ClientRegistration> registrations = null; List<ClientRegistration> registrations = null;

11
oauth2/oauth2-client/src/test/java/org/springframework/security/oauth2/client/registration/InMemoryReactiveClientRegistrationRepositoryTests.java
View File

@ -19,6 +19,7 @@ package org.springframework.security.oauth2.client.registration;
import static org.assertj.core.api.Assertions.assertThat; import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatThrownBy; import static org.assertj.core.api.Assertions.assertThatThrownBy;
import java.util.Arrays;
import java.util.List; import java.util.List;
import org.junit.Before; import org.junit.Before;
@ -61,10 +62,16 @@ public class InMemoryReactiveClientRegistrationRepositoryTests {
.isInstanceOf(IllegalArgumentException.class); .isInstanceOf(IllegalArgumentException.class);
} }
@Test
public void constructorWhenClientRegistrationListHasNullThenIllegalArgumentException() {
List<ClientRegistration> registrations = Arrays.asList(null, registration);
assertThatThrownBy(() -> new InMemoryReactiveClientRegistrationRepository(registrations))
.isInstanceOf(IllegalArgumentException.class);
}
@Test @Test
public void constructorWhenClientRegistrationIsNullThenIllegalArgumentException() { public void constructorWhenClientRegistrationIsNullThenIllegalArgumentException() {
ClientRegistration registration = null; assertThatThrownBy(() -> new InMemoryReactiveClientRegistrationRepository(registration, null))
assertThatThrownBy(() -> new InMemoryReactiveClientRegistrationRepository(registration))
.isInstanceOf(IllegalArgumentException.class); .isInstanceOf(IllegalArgumentException.class);
} }