Add an option to log banner rather than printing it to standard out

This commit adds the option to output the banner using the logger
instead of standard out. Rather than taking a boolean
spring.main.show-banner is now configured using an enum. Three values
are supported:

 - LOG: the banner is logged
 - CONSOLE: the banner is printed to standard out (previously true)
 - OFF: the banner is switched off (previously false)

The default behavior remains unchanged; the banner will be printed to
standard out.

Closes gh-4022
See gh-4001
pull/4118/merge
“Jeremy 9 years ago committed by Andy Wilkinson
parent 031a8d5afa
commit 3090659971

@ -123,7 +123,7 @@ might have.
[source,properties,indent=0,subs="verbatim,quotes,attributes"]
----
spring.main.web_environment=false
spring.main.show_banner=false
spring.main.show_banner=OFF
----
and then the Spring Boot banner will not be printed on startup, and the application will
@ -139,7 +139,7 @@ consider this application
[source,java,indent=0]
----
new SpringApplicationBuilder()
.showBanner(false)
.showBanner(Banner.Mode.OFF)
.sources(demo.MyApp.class)
.run(args);
----
@ -149,7 +149,7 @@ used with the following configuration:
[source,properties,indent=0,subs="verbatim,quotes,attributes"]
----
spring.main.sources=com.acme.Config,com.acme.ExtraConfig
spring.main.show-banner=true
spring.main.show-banner=CONSOLE
----
The actual application will _now_ show the banner (as overridden by configuration) and use

@ -124,7 +124,7 @@ For example:
[source,java,indent=0]
----
new SpringApplicationBuilder()
.showBanner(false)
.showBanner(Banner.Mode.OFF)
.sources(Parent.class)
.child(Application.class)
.run(args);

@ -25,6 +25,7 @@ import org.springframework.core.env.Environment;
*
* @author Phillip Webb
* @author Michael Stummvoll
* @author Jeremy Rickard
* @since 1.2.0
*/
public interface Banner {
@ -37,4 +38,25 @@ public interface Banner {
*/
void printBanner(Environment environment, Class<?> sourceClass, PrintStream out);
/**
* An enumeration of possible values for configuring the Banner.
*/
enum Mode {
/**
* Disable printing of the banner.
*/
OFF,
/**
* Print the banner to System.out.
*/
CONSOLE,
/**
* Print the banner to the log file.
*/
LOG
}
}

@ -16,6 +16,8 @@
package org.springframework.boot;
import java.io.ByteArrayOutputStream;
import java.io.PrintStream;
import java.lang.reflect.Constructor;
import java.security.AccessControlException;
import java.util.ArrayList;
@ -134,6 +136,7 @@ import org.springframework.web.context.support.StandardServletEnvironment;
* @author Andy Wilkinson
* @author Christian Dupuis
* @author Stephane Nicoll
* @author Jeremy Rickard
* @see #run(Object, String[])
* @see #run(Object[], String[])
* @see #SpringApplication(Object...)
@ -179,7 +182,7 @@ public class SpringApplication {
private Class<?> mainApplicationClass;
private boolean showBanner = true;
private Banner.Mode showBanner = Banner.Mode.CONSOLE;
private boolean logStartupInfo = true;
@ -321,7 +324,7 @@ public class SpringApplication {
environment = convertToStandardEnvironment(environment);
}
if (this.showBanner) {
if (this.showBanner != Banner.Mode.OFF) {
printBanner(environment);
}
@ -521,27 +524,37 @@ public class SpringApplication {
* banner.location=classpath:banner.txt, banner.charset=UTF-8. If the banner file does
* not exist or cannot be printed, a simple default is created.
* @param environment the environment
* @see #setShowBanner(boolean)
* @see #setShowBanner(org.springframework.boot.Banner.Mode)
*/
protected void printBanner(Environment environment) {
Banner selectedBanner = selectBanner(environment);
if (this.showBanner == Banner.Mode.LOG) {
ByteArrayOutputStream baos = new ByteArrayOutputStream();
selectedBanner.printBanner(environment, this.mainApplicationClass,
new PrintStream(baos));
this.log.info(baos.toString());
}
else {
selectedBanner
.printBanner(environment, this.mainApplicationClass, System.out);
}
}
private Banner selectBanner(Environment environment) {
String location = environment.getProperty("banner.location", "banner.txt");
ResourceLoader resourceLoader = this.resourceLoader != null ? this.resourceLoader
: new DefaultResourceLoader(getClassLoader());
Resource resource = resourceLoader.getResource(location);
if (resource.exists()) {
new ResourceBanner(resource).printBanner(environment,
this.mainApplicationClass, System.out);
return;
return new ResourceBanner(resource);
}
if (this.banner != null) {
this.banner.printBanner(environment, this.mainApplicationClass, System.out);
return;
return this.banner;
}
printDefaultBanner();
}
private void printDefaultBanner() {
DEFAULT_BANNER.printBanner(null, this.mainApplicationClass, System.out);
return DEFAULT_BANNER;
}
/**
@ -840,11 +853,11 @@ public class SpringApplication {
/**
* Sets if the Spring banner should be displayed when the application runs. Defaults
* to {@code true}.
* @param showBanner if the banner should be shown
* to {@code org.springframework.boot.Banner.Mode.CONSOLE}.
* @param bannerMode if the banner should be shown in log or console, or turned off.
*/
public void setShowBanner(boolean showBanner) {
this.showBanner = showBanner;
public void setShowBanner(Banner.Mode bannerMode) {
this.showBanner = bannerMode;
}
/**

@ -172,7 +172,7 @@ public class SpringApplicationBuilder {
web(false);
// Probably not interested in multiple banners
showBanner(false);
showBanner(Banner.Mode.OFF);
// Make sure sources get copied over
this.application.setSources(this.sources);
@ -311,7 +311,7 @@ public class SpringApplicationBuilder {
* @param showBanner the flag to set. Default true.
* @return the current builder
*/
public SpringApplicationBuilder showBanner(boolean showBanner) {
public SpringApplicationBuilder showBanner(Banner.Mode showBanner) {
this.application.setShowBanner(showBanner);
return this;
}

@ -57,6 +57,14 @@ public class BannerTests {
assertThat(this.out.toString(), containsString(":: Spring Boot ::"));
}
@Test
public void testDefaultBannerInLog() throws Exception {
SpringApplication application = new SpringApplication(Config.class);
application.setWebEnvironment(false);
this.context = application.run();
assertThat(this.out.toString(), containsString(":: Spring Boot ::"));
}
@Test
public void testCustomBanner() throws Exception {
SpringApplication application = new SpringApplication(Config.class);

@ -76,7 +76,7 @@ public class SimpleMainTests {
private String[] getArgs(String... args) {
List<String> list = new ArrayList<String>(Arrays.asList(
"--spring.main.webEnvironment=false", "--spring.main.showBanner=false",
"--spring.main.webEnvironment=false", "--spring.main.showBanner=OFF",
"--spring.main.registerShutdownHook=false"));
if (args.length > 0) {
list.add("--spring.main.sources="

@ -82,6 +82,7 @@ import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertThat;
import static org.junit.Assert.assertTrue;
import static org.mockito.Matchers.anyObject;
import static org.mockito.Mockito.atLeastOnce;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.spy;
@ -95,6 +96,7 @@ import static org.mockito.Mockito.verify;
* @author Andy Wilkinson
* @author Christian Dupuis
* @author Stephane Nicoll
* @author Jeremy Rickard
*/
public class SpringApplicationTests {
@ -164,7 +166,7 @@ public class SpringApplicationTests {
public void disableBanner() throws Exception {
SpringApplication application = spy(new SpringApplication(ExampleConfig.class));
application.setWebEnvironment(false);
application.setShowBanner(false);
application.setShowBanner(Banner.Mode.OFF);
this.context = application.run();
verify(application, never()).printBanner((Environment) anyObject());
}
@ -173,7 +175,7 @@ public class SpringApplicationTests {
public void disableBannerViaProperty() throws Exception {
SpringApplication application = spy(new SpringApplication(ExampleConfig.class));
application.setWebEnvironment(false);
this.context = application.run("--spring.main.show_banner=false");
this.context = application.run("--spring.main.show_banner=OFF");
verify(application, never()).printBanner((Environment) anyObject());
}
@ -213,6 +215,24 @@ public class SpringApplicationTests {
containsString("The following profiles are active: myprofile"));
}
@Test
public void enableBannerInLogViaProperty() throws Exception {
SpringApplication application = spy(new SpringApplication(ExampleConfig.class));
application.setWebEnvironment(false);
this.context = application.run("--spring.main.show_banner=LOG");
verify(application, atLeastOnce()).setShowBanner(Banner.Mode.LOG);
}
@Test
public void verifyBannerOutputContainsLogInfo() throws Exception {
SpringApplication application = spy(new SpringApplication(ExampleConfig.class));
application.setWebEnvironment(false);
application.run("--spring.main.show_banner=LOG",
"--banner.location=classpath:test-banner.txt");
verify(application, atLeastOnce()).setShowBanner(Banner.Mode.LOG);
assertThat(this.output.toString(), containsString("o.s.boot.SpringApplication"));
}
@Test
public void customId() throws Exception {
SpringApplication application = new SpringApplication(ExampleConfig.class);
@ -545,8 +565,8 @@ public class SpringApplicationTests {
TestSpringApplication application = new TestSpringApplication(
ExampleConfig.class);
application.setWebEnvironment(false);
this.context = application.run("--spring.main.show_banner=false");
assertThat(application.getShowBanner(), is(false));
this.context = application.run("--spring.main.show_banner=OFF");
assertThat(application.getShowBanner(), is(Banner.Mode.OFF));
}
@Test
@ -713,7 +733,7 @@ public class SpringApplicationTests {
private boolean useMockLoader;
private boolean showBanner;
private Banner.Mode showBanner;
TestSpringApplication(Object... sources) {
super(sources);
@ -744,12 +764,12 @@ public class SpringApplicationTests {
}
@Override
public void setShowBanner(boolean showBanner) {
super.setShowBanner(showBanner);
this.showBanner = showBanner;
public void setShowBanner(Banner.Mode bannerMode) {
super.setShowBanner(bannerMode);
this.showBanner = bannerMode;
}
public boolean getShowBanner() {
public Banner.Mode getShowBanner() {
return this.showBanner;
}

@ -34,6 +34,7 @@ import org.junit.After;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.ExpectedException;
import org.springframework.boot.Banner;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.context.config.ConfigFileEnvironmentPostProcessor.ConfigurationPropertySources;
import org.springframework.boot.env.EnumerableCompositePropertySource;
@ -625,17 +626,17 @@ public class ConfigFileEnvironmentPostProcessorTests {
this.initializer.postProcessEnvironment(this.environment, this.application);
Field field = ReflectionUtils.findField(SpringApplication.class, "showBanner");
field.setAccessible(true);
assertThat((Boolean) field.get(this.application), equalTo(false));
assertThat((Banner.Mode) field.get(this.application), equalTo(Banner.Mode.OFF));
}
@Test
public void bindsSystemPropertyToSpringApplication() throws Exception {
// gh-951
System.setProperty("spring.main.showBanner", "false");
System.setProperty("spring.main.showBanner", "OFF");
this.initializer.postProcessEnvironment(this.environment, this.application);
Field field = ReflectionUtils.findField(SpringApplication.class, "showBanner");
field.setAccessible(true);
assertThat((Boolean) field.get(this.application), equalTo(false));
assertThat((Banner.Mode) field.get(this.application), equalTo(Banner.Mode.OFF));
}
private static Matcher<? super ConfigurableEnvironment> containsPropertySource(

@ -1 +1 @@
spring.main.show_banner=false
spring.main.show_banner=OFF

Loading…
Cancel
Save