View Javadoc
1   /**
2    * Portions of this software was developed by employees of the National Institute
3    * of Standards and Technology (NIST), an agency of the Federal Government and is
4    * being made available as a public service. Pursuant to title 17 United States
5    * Code Section 105, works of NIST employees are not subject to copyright
6    * protection in the United States. This software may be subject to foreign
7    * copyright. Permission in the United States and in foreign countries, to the
8    * extent that NIST may hold copyright, to use, copy, modify, create derivative
9    * works, and distribute this software and its documentation without fee is hereby
10   * granted on a non-exclusive basis, provided that this notice and disclaimer
11   * of warranty appears in all copies.
12   *
13   * THE SOFTWARE IS PROVIDED 'AS IS' WITHOUT ANY WARRANTY OF ANY KIND, EITHER
14   * EXPRESSED, IMPLIED, OR STATUTORY, INCLUDING, BUT NOT LIMITED TO, ANY WARRANTY
15   * THAT THE SOFTWARE WILL CONFORM TO SPECIFICATIONS, ANY IMPLIED WARRANTIES OF
16   * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, AND FREEDOM FROM
17   * INFRINGEMENT, AND ANY WARRANTY THAT THE DOCUMENTATION WILL CONFORM TO THE
18   * SOFTWARE, OR ANY WARRANTY THAT THE SOFTWARE WILL BE ERROR FREE.  IN NO EVENT
19   * SHALL NIST BE LIABLE FOR ANY DAMAGES, INCLUDING, BUT NOT LIMITED TO, DIRECT,
20   * INDIRECT, SPECIAL OR CONSEQUENTIAL DAMAGES, ARISING OUT OF, RESULTING FROM,
21   * OR IN ANY WAY CONNECTED WITH THIS SOFTWARE, WHETHER OR NOT BASED UPON WARRANTY,
22   * CONTRACT, TORT, OR OTHERWISE, WHETHER OR NOT INJURY WAS SUSTAINED BY PERSONS OR
23   * PROPERTY OR OTHERWISE, AND WHETHER OR NOT LOSS WAS SUSTAINED FROM, OR AROSE OUT
24   * OF THE RESULTS OF, OR USE OF, THE SOFTWARE OR SERVICES PROVIDED HEREUNDER.
25   */
26  
27  package gov.nist.secauto.decima.module.logging;
28  
29  import org.apache.logging.log4j.Level;
30  import org.apache.logging.log4j.LogManager;
31  import org.apache.logging.log4j.core.LoggerContext;
32  import org.apache.logging.log4j.core.config.Configuration;
33  import org.apache.logging.log4j.core.config.ConfigurationFactory;
34  import org.apache.logging.log4j.core.config.ConfigurationSource;
35  import org.apache.logging.log4j.core.config.LoggerConfig;
36  import org.apache.logging.log4j.core.config.Order;
37  import org.apache.logging.log4j.core.config.plugins.Plugin;
38  import org.apache.logging.log4j.core.config.xml.XmlConfiguration;
39  
40  @Plugin(name = "DecimaLoggingConfigurationFactory", category = "ConfigurationFactory")
41  @Order(10)
42  public class DecimaLoggingConfigurationFactory
43      extends ConfigurationFactory {
44  
45    /**
46     * Sets the logging level of the root logger.
47     * 
48     * @param logLevel
49     *          the new {@link Level} to use
50     */
51    public static void changeRootLogLevel(Level logLevel) {
52  
53      final LoggerContext ctx = (LoggerContext) LogManager.getContext(false);
54      final Configuration config = ctx.getConfiguration();
55  
56      config.getRootLogger().setLevel(logLevel);
57      ctx.updateLoggers();
58    }
59  
60    /**
61     * Valid file extensions for XML files.
62     */
63    public static final String[] SUFFIXES = new String[] { ".xml", "*" };
64  
65    private Level logLevel = Level.INFO;
66  
67    public Level getLogLevel() {
68      return logLevel;
69    }
70  
71    public void setLogLevel(Level logLevel) {
72      this.logLevel = logLevel;
73    }
74  
75    @Override
76    public Configuration getConfiguration(LoggerContext loggerContext, ConfigurationSource source) {
77      return new DecimaXmlConfiguration(loggerContext, source);
78    }
79  
80    @Override
81    protected String[] getSupportedTypes() {
82      return SUFFIXES;
83    }
84  
85    public class DecimaXmlConfiguration
86        extends XmlConfiguration {
87  
88      /** the default serial version UID. */
89      @SuppressWarnings("unused")
90      private static final long serialVersionUID = 1L;
91  
92      public DecimaXmlConfiguration(LoggerContext loggerContext, ConfigurationSource configSource) {
93        super(loggerContext, configSource);
94      }
95  
96      @Override
97      protected void doConfigure() {
98        super.doConfigure();
99  
100       // final LoggerContext ctx = (LoggerContext) LogManager.getContext(false);
101       LoggerConfig rootConfig = getRootLogger();
102       rootConfig.setLevel(getLogLevel());
103     }
104   }
105 }