Fix support for default values in banner placeholders

See gh-34764
pull/34975/head
Krzysztof Krason 2 years ago committed by Andy Wilkinson
parent 17a4d303f8
commit 5c01aa719c

@ -1,5 +1,5 @@
/*
* Copyright 2012-2019 the original author or authors.
* 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.
@ -19,7 +19,6 @@ package org.springframework.boot;
import java.io.PrintStream;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
@ -33,6 +32,7 @@ import org.springframework.core.env.Environment;
import org.springframework.core.env.MapPropertySource;
import org.springframework.core.env.MutablePropertySources;
import org.springframework.core.env.PropertyResolver;
import org.springframework.core.env.PropertySource;
import org.springframework.core.env.PropertySourcesPropertyResolver;
import org.springframework.core.io.Resource;
import org.springframework.core.log.LogMessage;
@ -45,13 +45,14 @@ import org.springframework.util.StreamUtils;
* @author Phillip Webb
* @author Vedran Pavic
* @author Toshiaki Maki
* @author Krzysztof Krason
* @since 1.2.0
*/
public class ResourceBanner implements Banner {
private static final Log logger = LogFactory.getLog(ResourceBanner.class);
private Resource resource;
private final Resource resource;
public ResourceBanner(Resource resource) {
Assert.notNull(resource, "Resource must not be null");
@ -77,18 +78,25 @@ public class ResourceBanner implements Banner {
}
protected List<PropertyResolver> getPropertyResolvers(Environment environment, Class<?> sourceClass) {
List<PropertyResolver> resolvers = new ArrayList<>();
resolvers.add(environment);
resolvers.add(getVersionResolver(sourceClass));
resolvers.add(getAnsiResolver());
resolvers.add(getTitleResolver(sourceClass));
return resolvers;
MutablePropertySources propertySources = new MutablePropertySources();
propertySources.addLast(getEnvironmentSource(environment));
propertySources.addLast(getTitleSource(sourceClass));
propertySources.addLast(getAnsiSource());
propertySources.addLast(getVersionSource(sourceClass));
return Collections.singletonList(new PropertySourcesPropertyResolver(propertySources));
}
private PropertyResolver getVersionResolver(Class<?> sourceClass) {
MutablePropertySources propertySources = new MutablePropertySources();
propertySources.addLast(new MapPropertySource("version", getVersionsMap(sourceClass)));
return new PropertySourcesPropertyResolver(propertySources);
private PropertySource<Environment> getEnvironmentSource(Environment environment) {
return new PropertySource<Environment>("environment", environment) {
@Override
public Object getProperty(String name) {
return environment.getProperty(name);
}
};
}
private MapPropertySource getVersionSource(Class<?> sourceClass) {
return new MapPropertySource("version", getVersionsMap(sourceClass));
}
private Map<String, Object> getVersionsMap(Class<?> sourceClass) {
@ -118,19 +126,15 @@ public class ResourceBanner implements Banner {
return format ? " (v" + version + ")" : version;
}
private PropertyResolver getAnsiResolver() {
MutablePropertySources sources = new MutablePropertySources();
sources.addFirst(new AnsiPropertySource("ansi", true));
return new PropertySourcesPropertyResolver(sources);
private AnsiPropertySource getAnsiSource() {
return new AnsiPropertySource("ansi", true);
}
private PropertyResolver getTitleResolver(Class<?> sourceClass) {
MutablePropertySources sources = new MutablePropertySources();
private MapPropertySource getTitleSource(Class<?> sourceClass) {
String applicationTitle = getApplicationTitle(sourceClass);
Map<String, Object> titleMap = Collections.singletonMap("application.title",
(applicationTitle != null) ? applicationTitle : "");
sources.addFirst(new MapPropertySource("title", titleMap));
return new PropertySourcesPropertyResolver(sources);
return new MapPropertySource("title", titleMap);
}
protected String getApplicationTitle(Class<?> sourceClass) {

@ -1,5 +1,5 @@
/*
* Copyright 2012-2019 the original author or authors.
* 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.
@ -40,6 +40,7 @@ import static org.assertj.core.api.Assertions.assertThat;
* @author Phillip Webb
* @author Vedran Pavic
* @author Toshiaki Maki
* @author Krzysztof Krason
*/
class ResourceBannerTests {
@ -48,6 +49,21 @@ class ResourceBannerTests {
AnsiOutput.setEnabled(Enabled.DETECT);
}
@Test
void doNotUseDefaultsIfValueExists() {
Resource resource = new ByteArrayResource(
"banner ${a:def} ${spring-boot.version:def} ${application.version:def}".getBytes());
String banner = printBanner(resource, "10.2", "1.0", null);
assertThat(banner).startsWith("banner 1 10.2 1.0");
}
@Test
void useDefaults() {
Resource resource = new ByteArrayResource("banner ${b:def1} ${c:def2} ${d:def3}".getBytes());
String banner = printBanner(resource, null, null, null);
assertThat(banner).startsWith("banner def1 def2 def3");
}
@Test
void renderVersions() {
Resource resource = new ByteArrayResource(

Loading…
Cancel
Save