Support @Order on [CommandLine|Application]Runner @Bean definitions

Prior to this commit, `@Order` annotation on `@Bean` method was not
considered for `CommandLineRunner` and `ApplicationRunner`.

This commit introduces a `Runner` marker interface and uses it to
retrieve the runner beans. As a result, it enables the use of `@Order`
annotations on `@Bean` methods for both `CommandLineRunner` and
`ApplicationRunner`.

Signed-off-by: Tadaya Tsuyukubo <tadaya@ttddyy.net>

See gh-37905
This commit is contained in:
Tadaya Tsuyukubo 2023-10-16 21:10:47 -07:00 committed by Andy Wilkinson
parent 27f06ea1e0
commit 8ac597c245
5 changed files with 69 additions and 8 deletions

View File

@ -30,7 +30,7 @@ import org.springframework.core.annotation.Order;
* @see CommandLineRunner * @see CommandLineRunner
*/ */
@FunctionalInterface @FunctionalInterface
public interface ApplicationRunner { public interface ApplicationRunner extends Runner {
/** /**
* Callback used to run the bean. * Callback used to run the bean.

View File

@ -33,7 +33,7 @@ import org.springframework.core.annotation.Order;
* @see ApplicationRunner * @see ApplicationRunner
*/ */
@FunctionalInterface @FunctionalInterface
public interface CommandLineRunner { public interface CommandLineRunner extends Runner {
/** /**
* Callback used to run the bean. * Callback used to run the bean.

View File

@ -0,0 +1,28 @@
/*
* Copyright 2012-2023 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.springframework.boot;
/**
* Marker interface for runners.
*
* @author Tadaya Tsuyukubo
* @see ApplicationRunner
* @see CommandLineRunner
*/
interface Runner {
}

View File

@ -156,6 +156,7 @@ import org.springframework.util.StringUtils;
* @author Brian Clozel * @author Brian Clozel
* @author Ethan Rubinson * @author Ethan Rubinson
* @author Chris Bono * @author Chris Bono
* @author Tadaya Tsuyukubo
* @since 1.0.0 * @since 1.0.0
* @see #run(Class, String[]) * @see #run(Class, String[])
* @see #run(Class[], String[]) * @see #run(Class[], String[])
@ -740,18 +741,14 @@ public class SpringApplication {
} }
private void callRunners(ApplicationContext context, ApplicationArguments args) { private void callRunners(ApplicationContext context, ApplicationArguments args) {
List<Object> runners = new ArrayList<>(); context.getBeanProvider(Runner.class).orderedStream().forEach((runner) -> {
runners.addAll(context.getBeansOfType(ApplicationRunner.class).values());
runners.addAll(context.getBeansOfType(CommandLineRunner.class).values());
AnnotationAwareOrderComparator.sort(runners);
for (Object runner : new LinkedHashSet<>(runners)) {
if (runner instanceof ApplicationRunner) { if (runner instanceof ApplicationRunner) {
callRunner((ApplicationRunner) runner, args); callRunner((ApplicationRunner) runner, args);
} }
if (runner instanceof CommandLineRunner) { if (runner instanceof CommandLineRunner) {
callRunner((CommandLineRunner) runner, args); callRunner((CommandLineRunner) runner, args);
} }
} });
} }
private void callRunner(ApplicationRunner runner, ApplicationArguments args) { private void callRunner(ApplicationRunner runner, ApplicationArguments args) {

View File

@ -95,6 +95,7 @@ import org.springframework.context.event.SmartApplicationListener;
import org.springframework.context.support.AbstractApplicationContext; import org.springframework.context.support.AbstractApplicationContext;
import org.springframework.context.support.StaticApplicationContext; import org.springframework.context.support.StaticApplicationContext;
import org.springframework.core.Ordered; import org.springframework.core.Ordered;
import org.springframework.core.annotation.Order;
import org.springframework.core.env.CommandLinePropertySource; import org.springframework.core.env.CommandLinePropertySource;
import org.springframework.core.env.CompositePropertySource; import org.springframework.core.env.CompositePropertySource;
import org.springframework.core.env.ConfigurableEnvironment; import org.springframework.core.env.ConfigurableEnvironment;
@ -154,6 +155,7 @@ import static org.mockito.Mockito.spy;
* @author Nguyen Bao Sach * @author Nguyen Bao Sach
* @author Chris Bono * @author Chris Bono
* @author Brian Clozel * @author Brian Clozel
* @author Tadaya Tsuyukubo
*/ */
@ExtendWith(OutputCaptureExtension.class) @ExtendWith(OutputCaptureExtension.class)
class SpringApplicationTests { class SpringApplicationTests {
@ -667,6 +669,15 @@ class SpringApplicationTests {
assertThat(this.context).has(runTestRunnerBean("runnerC")); assertThat(this.context).has(runTestRunnerBean("runnerC"));
} }
@Test
void runFunctionalCommandLineRunnersAndApplicationRunners() {
SpringApplication application = new SpringApplication(FunctionalRunnerConfig.class);
application.setWebApplicationType(WebApplicationType.NONE);
this.context = application.run("arg");
FunctionalRunnerConfig config = this.context.getBean(FunctionalRunnerConfig.class);
assertThat(config.runners).containsExactly("runnerA", "runnerB", "runnerC");
}
@Test @Test
@SuppressWarnings("unchecked") @SuppressWarnings("unchecked")
void runnersAreCalledAfterStartedIsLoggedAndBeforeApplicationReadyEventIsPublished(CapturedOutput output) void runnersAreCalledAfterStartedIsLoggedAndBeforeApplicationReadyEventIsPublished(CapturedOutput output)
@ -1573,6 +1584,31 @@ class SpringApplicationTests {
} }
@Configuration(proxyBeanMethods = false)
static class FunctionalRunnerConfig {
List<String> runners = new ArrayList<>();
@Bean
@Order // default is LOWEST_PRECEDENCE
CommandLineRunner runnerC() {
return (args) -> this.runners.add("runnerC");
}
@Bean
@Order(Ordered.LOWEST_PRECEDENCE - 1)
ApplicationRunner runnerB() {
return (args) -> this.runners.add("runnerB");
}
@Bean
@Order(Ordered.HIGHEST_PRECEDENCE)
CommandLineRunner runnerA() {
return (args) -> this.runners.add("runnerA");
}
}
@Configuration(proxyBeanMethods = false) @Configuration(proxyBeanMethods = false)
static class ExitCodeCommandLineRunConfig { static class ExitCodeCommandLineRunConfig {