Skip to content

Commit 813d180

Browse files
kse-musicrwinch
authored andcommitted
Support add nested security configurers during builder initialization
Closes spring-projectsgh-17011 Signed-off-by: DingHao <[email protected]>
1 parent 4fa2d32 commit 813d180

File tree

2 files changed

+39
-2
lines changed

2 files changed

+39
-2
lines changed

config/src/main/java/org/springframework/security/config/annotation/AbstractConfiguredSecurityBuilder.java

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@
5050
* @param <O> The object that this builder returns
5151
* @param <B> The type of this builder (that is returned by the base class)
5252
* @author Rob Winch
53+
* @author DingHao
5354
* @see WebSecurity
5455
*/
5556
public abstract class AbstractConfiguredSecurityBuilder<O, B extends SecurityBuilder<O>>
@@ -369,8 +370,10 @@ private void init() throws Exception {
369370
for (SecurityConfigurer<O, B> configurer : configurers) {
370371
configurer.init((B) this);
371372
}
372-
for (SecurityConfigurer<O, B> configurer : this.configurersAddedInInitializing) {
373-
configurer.init((B) this);
373+
while (!this.configurersAddedInInitializing.isEmpty()) {
374+
for (SecurityConfigurer<O, B> configurer : getConfigurersInInitializing()) {
375+
configurer.init((B) this);
376+
}
374377
}
375378
}
376379

@@ -390,6 +393,12 @@ private Collection<SecurityConfigurer<O, B>> getConfigurers() {
390393
return result;
391394
}
392395

396+
private List<SecurityConfigurer<O, B>> getConfigurersInInitializing() {
397+
List<SecurityConfigurer<O, B>> result = new ArrayList<>(this.configurersAddedInInitializing);
398+
this.configurersAddedInInitializing.clear();
399+
return result;
400+
}
401+
393402
/**
394403
* Determines if the object is unbuilt.
395404
* @return true, if unbuilt else false

config/src/test/java/org/springframework/security/config/annotation/web/AbstractConfiguredSecurityBuilderTests.java

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -163,6 +163,34 @@ public void withWhenDuplicateConfigurerAddedThenDuplicateConfigurerRemoved() thr
163163
assertThat(this.builder.getConfigurers(TestSecurityConfigurer.class)).hasSize(1);
164164
}
165165

166+
@Test
167+
public void withWhenConfigurerAddInitializing() throws Exception {
168+
this.builder.with(new FooConfigurer(), Customizer.withDefaults());
169+
assertThat(this.builder.build()).isEqualTo("success");
170+
}
171+
172+
private static class FooConfigurer extends SecurityConfigurerAdapter<Object, TestConfiguredSecurityBuilder> {
173+
174+
@Override
175+
public void init(TestConfiguredSecurityBuilder builder) throws Exception {
176+
builder.with(new BarConfigurer(), Customizer.withDefaults());
177+
}
178+
179+
}
180+
181+
private static class BarConfigurer extends SecurityConfigurerAdapter<Object, TestConfiguredSecurityBuilder> {
182+
183+
@Override
184+
public void init(TestConfiguredSecurityBuilder http) throws Exception {
185+
http.with(new CooConfigurer(), Customizer.withDefaults());
186+
}
187+
188+
}
189+
190+
private static class CooConfigurer extends SecurityConfigurerAdapter<Object, TestConfiguredSecurityBuilder> {
191+
192+
}
193+
166194
private static class ApplyAndRemoveSecurityConfigurer
167195
extends SecurityConfigurerAdapter<Object, TestConfiguredSecurityBuilder> {
168196

0 commit comments

Comments
 (0)