Skip to content

Commit 48154a3

Browse files
nosansnicoll
authored andcommitted
Add support for optional Log4J2 configuration
This commit adds support for the standard 'optional:' prefix in Log4j2 override file locations, ensuring missing files are ignored without throwing exceptions. See gh-44488 Signed-off-by: Dmytro Nosan <[email protected]>
1 parent 93dc0fb commit 48154a3

File tree

3 files changed

+89
-14
lines changed

3 files changed

+89
-14
lines changed

Diff for: spring-boot-project/spring-boot-docs/src/docs/antora/modules/how-to/pages/logging.adoc

+2
Original file line numberDiff line numberDiff line change
@@ -201,3 +201,5 @@ To configure Log4j 2 to use an alternative configuration file format, add the ap
201201
Log4j 2 has support for combining multiple configuration files into a single composite configuration.
202202
To use this support in Spring Boot, configure configprop:logging.log4j2.config.override[] with the locations of one or more secondary configuration files.
203203
The secondary configuration files will be merged with the primary configuration, whether the primary's source is Spring Boot's defaults, a standard location such as `log4j.xml`, or the location configured by the configprop:logging.config[] property.
204+
205+
NOTE: Log4j2 override configuration file locations can be prefixed with `optional:`, for example, `optional:classpath:log4j2-override.xml`, to indicate that the location is optional and should only be loaded if the resource exists.

Diff for: spring-boot-project/spring-boot/src/main/java/org/springframework/boot/logging/log4j2/Log4J2LoggingSystem.java

+44-14
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616

1717
package org.springframework.boot.logging.log4j2;
1818

19+
import java.io.FileNotFoundException;
1920
import java.io.IOException;
2021
import java.io.InputStream;
2122
import java.net.URL;
@@ -68,6 +69,7 @@
6869
import org.springframework.core.annotation.Order;
6970
import org.springframework.core.env.Environment;
7071
import org.springframework.core.io.Resource;
72+
import org.springframework.core.io.ResourceLoader;
7173
import org.springframework.util.Assert;
7274
import org.springframework.util.ClassUtils;
7375
import org.springframework.util.CollectionUtils;
@@ -85,6 +87,8 @@
8587
*/
8688
public class Log4J2LoggingSystem extends AbstractLoggingSystem {
8789

90+
private static final String OPTIONAL_PREFIX = "optional:";
91+
8892
private static final String LOG4J_BRIDGE_HANDLER = "org.apache.logging.log4j.jul.Log4jBridgeHandler";
8993

9094
private static final String LOG4J_LOG_MANAGER = "org.apache.logging.log4j.jul.LogManager";
@@ -269,21 +273,42 @@ protected void loadConfiguration(String location, LogFile logFile, List<String>
269273
try {
270274
List<Configuration> configurations = new ArrayList<>();
271275
LoggerContext context = getLoggerContext();
272-
configurations.add(load(location, context));
276+
ResourceLoader resourceLoader = ApplicationResourceLoader.get();
277+
configurations.add(loadConfiguration(resourceLoader, location, context));
273278
for (String override : overrides) {
274-
configurations.add(load(override, context));
279+
Configuration overrideConfiguration = loadOptionalConfiguration(resourceLoader, override, context);
280+
if (overrideConfiguration != null) {
281+
configurations.add(overrideConfiguration);
282+
}
275283
}
276-
Configuration configuration = (configurations.size() > 1) ? createComposite(configurations)
277-
: configurations.iterator().next();
278-
context.start(configuration);
284+
context.start(createCompositeConfigurationIfNecessary(configurations));
279285
}
280286
catch (Exception ex) {
281287
throw new IllegalStateException("Could not initialize Log4J2 logging from " + location, ex);
282288
}
283289
}
284290

285-
private Configuration load(String location, LoggerContext context) throws IOException {
286-
Resource resource = ApplicationResourceLoader.get().getResource(location);
291+
private Configuration loadOptionalConfiguration(ResourceLoader resourceLoader, String location,
292+
LoggerContext context) throws IOException {
293+
if (location.startsWith(OPTIONAL_PREFIX)) {
294+
Resource resource = resourceLoader.getResource(location.substring(OPTIONAL_PREFIX.length()));
295+
try {
296+
return (resource.exists()) ? loadConfiguration(resource, context) : null;
297+
}
298+
catch (FileNotFoundException ex) {
299+
return null;
300+
}
301+
}
302+
return loadConfiguration(resourceLoader, location, context);
303+
}
304+
305+
private Configuration loadConfiguration(ResourceLoader resourceLoader, String location, LoggerContext context)
306+
throws IOException {
307+
Resource resource = resourceLoader.getResource(location);
308+
return loadConfiguration(resource, context);
309+
}
310+
311+
private Configuration loadConfiguration(Resource resource, LoggerContext context) throws IOException {
287312
ConfigurationFactory factory = ConfigurationFactory.getInstance();
288313
if (resource.isFile()) {
289314
try (InputStream inputStream = resource.getInputStream()) {
@@ -303,7 +328,10 @@ private Configuration load(String location, LoggerContext context) throws IOExce
303328
}
304329
}
305330

306-
private CompositeConfiguration createComposite(List<Configuration> configurations) {
331+
private Configuration createCompositeConfigurationIfNecessary(List<Configuration> configurations) {
332+
if (configurations.size() == 1) {
333+
return configurations.iterator().next();
334+
}
307335
return new CompositeConfiguration(configurations.stream().map(AbstractConfiguration.class::cast).toList());
308336
}
309337

@@ -321,19 +349,21 @@ protected void reinitialize(LoggingInitializationContext initializationContext)
321349

322350
private void reinitializeWithOverrides(List<String> overrides) {
323351
LoggerContext context = getLoggerContext();
324-
Configuration base = context.getConfiguration();
325-
List<AbstractConfiguration> configurations = new ArrayList<>();
326-
configurations.add((AbstractConfiguration) base);
352+
List<Configuration> configurations = new ArrayList<>();
353+
configurations.add(context.getConfiguration());
354+
ResourceLoader resourceLoader = ApplicationResourceLoader.get();
327355
for (String override : overrides) {
328356
try {
329-
configurations.add((AbstractConfiguration) load(override, context));
357+
Configuration overrideConfiguration = loadOptionalConfiguration(resourceLoader, override, context);
358+
if (overrideConfiguration != null) {
359+
configurations.add(overrideConfiguration);
360+
}
330361
}
331362
catch (IOException ex) {
332363
throw new RuntimeException("Failed to load overriding configuration from '" + override + "'", ex);
333364
}
334365
}
335-
CompositeConfiguration composite = new CompositeConfiguration(configurations);
336-
context.reconfigure(composite);
366+
context.reconfigure(createCompositeConfigurationIfNecessary(configurations));
337367
}
338368

339369
@Override

Diff for: spring-boot-project/spring-boot/src/test/java/org/springframework/boot/logging/log4j2/Log4J2LoggingSystemTests.java

+43
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@
4343
import org.apache.logging.log4j.core.config.Reconfigurable;
4444
import org.apache.logging.log4j.core.config.composite.CompositeConfiguration;
4545
import org.apache.logging.log4j.core.config.plugins.util.PluginRegistry;
46+
import org.apache.logging.log4j.core.config.xml.XmlConfiguration;
4647
import org.apache.logging.log4j.core.util.ShutdownCallbackRegistry;
4748
import org.apache.logging.log4j.jul.Log4jBridgeHandler;
4849
import org.apache.logging.log4j.status.StatusListener;
@@ -453,6 +454,48 @@ void shutdownHookIsDisabled() {
453454
.isFalse();
454455
}
455456

457+
@Test
458+
@WithNonDefaultXmlResource
459+
void loadOptionalOverrideConfigurationWhenDoesNotExist() {
460+
this.environment.setProperty("logging.log4j2.config.override", "optional:classpath:override.xml");
461+
this.loggingSystem.initialize(this.initializationContext, "classpath:nondefault.xml", null);
462+
assertThat(this.loggingSystem.getConfiguration()).isInstanceOf(XmlConfiguration.class);
463+
}
464+
465+
@Test
466+
void loadOptionalOverrideConfigurationWhenDoesNotExistUponReinitialization() {
467+
this.environment.setProperty("logging.log4j2.config.override", "optional:classpath:override.xml");
468+
this.loggingSystem.beforeInitialize();
469+
this.loggingSystem.initialize(this.initializationContext, null, null);
470+
assertThat(this.loggingSystem.getConfiguration()).isInstanceOf(XmlConfiguration.class);
471+
this.loggingSystem.cleanUp();
472+
this.loggingSystem.beforeInitialize();
473+
this.loggingSystem.initialize(this.initializationContext, null, null);
474+
assertThat(this.loggingSystem.getConfiguration()).isInstanceOf(XmlConfiguration.class);
475+
}
476+
477+
@Test
478+
@WithNonDefaultXmlResource
479+
@WithOverrideXmlResource
480+
void loadOptionalOverrideConfiguration() {
481+
this.environment.setProperty("logging.log4j2.config.override", "optional:classpath:override.xml");
482+
this.loggingSystem.initialize(this.initializationContext, "classpath:nondefault.xml", null);
483+
assertThat(this.loggingSystem.getConfiguration()).isInstanceOf(CompositeConfiguration.class);
484+
}
485+
486+
@Test
487+
@WithOverrideXmlResource
488+
void loadOptionalOverrideConfigurationUponReinitialization() {
489+
this.environment.setProperty("logging.log4j2.config.override", "optional:classpath:override.xml");
490+
this.loggingSystem.beforeInitialize();
491+
this.loggingSystem.initialize(this.initializationContext, null, null);
492+
assertThat(this.loggingSystem.getConfiguration()).isInstanceOf(CompositeConfiguration.class);
493+
this.loggingSystem.cleanUp();
494+
this.loggingSystem.beforeInitialize();
495+
this.loggingSystem.initialize(this.initializationContext, null, null);
496+
assertThat(this.loggingSystem.getConfiguration()).isInstanceOf(CompositeConfiguration.class);
497+
}
498+
456499
@Test
457500
@WithNonDefaultXmlResource
458501
@WithOverrideXmlResource

0 commit comments

Comments
 (0)