001/**
002 * Logback: the reliable, generic, fast and flexible logging framework. Copyright (C) 1999-2015, QOS.ch. All rights reserved.
003 * <p>
004 * This program and the accompanying materials are dual-licensed under either the terms of the Eclipse Public License v1.0 as published by the Eclipse
005 * Foundation
006 * <p>
007 * or (per the licensee's choosing)
008 * <p>
009 * under the terms of the GNU Lesser General Public License version 2.1 as published by the Free Software Foundation.
010 */
011package ch.qos.logback.classic.util;
012
013import ch.qos.logback.classic.ClassicConstants;
014import ch.qos.logback.classic.LoggerContext;
015import ch.qos.logback.classic.spi.Configurator;
016import ch.qos.logback.classic.spi.ConfiguratorRank;
017import ch.qos.logback.core.CoreConstants;
018import ch.qos.logback.core.LogbackException;
019import ch.qos.logback.core.joran.spi.JoranException;
020import ch.qos.logback.core.spi.ContextAware;
021import ch.qos.logback.core.spi.ContextAwareImpl;
022import ch.qos.logback.core.status.InfoStatus;
023import ch.qos.logback.core.status.WarnStatus;
024import ch.qos.logback.core.util.EnvUtil;
025import ch.qos.logback.core.util.Loader;
026import ch.qos.logback.core.util.StatusListenerConfigHelper;
027import ch.qos.logback.core.util.VersionUtil;
028
029import java.util.Comparator;
030import java.util.List;
031
032// contributors
033// Ted Graham, Matt Fowles, see also http://jira.qos.ch/browse/LBCORE-32
034
035/**
036 * This class contains logback's logic for automatic configuration
037 *
038 * @author Ceki Gulcu
039 */
040public class ContextInitializer {
041
042    /**
043     *  @deprecated Please use ClassicConstants.AUTOCONFIG_FILE instead
044     */
045    final public static String AUTOCONFIG_FILE = ClassicConstants.AUTOCONFIG_FILE;
046    /**
047     * @deprecated Please use ClassicConstants.TEST_AUTOCONFIG_FILE instead
048     */
049    final public static String TEST_AUTOCONFIG_FILE = ClassicConstants.TEST_AUTOCONFIG_FILE;
050    /**
051     * @deprecated Please use ClassicConstants.CONFIG_FILE_PROPERTY instead
052     */
053    final public static String CONFIG_FILE_PROPERTY = ClassicConstants.CONFIG_FILE_PROPERTY;
054
055    String[] INTERNAL_CONFIGURATOR_CLASSNAME_LIST = {"ch.qos.logback.classic.util.DefaultJoranConfigurator", "ch.qos.logback.classic.BasicConfigurator"};
056
057    final LoggerContext loggerContext;
058
059    final ContextAware contextAware;
060
061    public ContextInitializer(LoggerContext loggerContext) {
062        this.loggerContext = loggerContext;
063        this.contextAware = new ContextAwareImpl(loggerContext, this);
064    }
065
066    public void autoConfig() throws JoranException {
067        autoConfig(Configurator.class.getClassLoader());
068    }
069
070
071    public void autoConfig(ClassLoader classLoader) throws JoranException {
072
073        // see https://github.com/qos-ch/logback/issues/715
074        classLoader = Loader.systemClassloaderIfNull(classLoader);
075
076        checkVersions();
077
078        StatusListenerConfigHelper.installIfAsked(loggerContext);
079
080
081        // invoke custom configurators
082        List<Configurator> configuratorList = ClassicEnvUtil.loadFromServiceLoader(Configurator.class, classLoader);
083        configuratorList.sort(rankComparator);
084        if (configuratorList.isEmpty()) {
085            contextAware.addInfo("No custom configurators were discovered as a service.");
086        } else {
087            printConfiguratorOrder(configuratorList);
088        }
089
090        for (Configurator c : configuratorList) {
091            if (invokeConfigure(c) == Configurator.ExecutionStatus.DO_NOT_INVOKE_NEXT_IF_ANY)
092                return;
093        }
094
095        // invoke internal configurators
096        for (String configuratorClassName : INTERNAL_CONFIGURATOR_CLASSNAME_LIST) {
097            contextAware.addInfo("Trying to configure with "+configuratorClassName);
098            Configurator c = instantiateConfiguratorByClassName(configuratorClassName, classLoader);
099            if(c == null)
100                continue;
101            if (invokeConfigure(c) == Configurator.ExecutionStatus.DO_NOT_INVOKE_NEXT_IF_ANY)
102                return;
103        }
104    }
105
106    private void checkVersions() {
107        try {
108            VersionUtil.checkForVersionEquality(loggerContext, this.getClass(), VersionUtil.class, "logback-classic", "logback-core");
109        }  catch(NoClassDefFoundError e) {
110            contextAware.addWarn("Missing ch.logback.core.util.VersionUtil class on classpath. The version of logback-core is probably older than 1.5.25.");
111        }
112    }
113
114    private Configurator instantiateConfiguratorByClassName(String configuratorClassName, ClassLoader classLoader) {
115        try {
116            Class<?> classObj = classLoader.loadClass(configuratorClassName);
117            return (Configurator) classObj.getConstructor().newInstance();
118        } catch (ReflectiveOperationException  e) {
119            contextAware.addInfo("Instantiation failure: " + e.toString());
120            return null;
121        }
122    }
123
124    /**
125     *
126     * @param configurator
127     * @return ExecutionStatus
128     */
129    private Configurator.ExecutionStatus invokeConfigure(Configurator configurator) {
130        try {
131            long start = System.currentTimeMillis();
132            contextAware.addInfo("Constructed configurator of type " + configurator.getClass());
133            configurator.setContext(loggerContext);
134            Configurator.ExecutionStatus status = configurator.configure(loggerContext);
135            printDuration(start, configurator, status);
136            return status;
137
138        } catch (Exception e) {
139            throw new LogbackException(String.format("Failed to initialize or to run Configurator: %s",
140                    configurator != null ? configurator.getClass().getCanonicalName() : "null"), e);
141        }
142    }
143
144    private void printConfiguratorOrder(List<Configurator> configuratorList) {
145        contextAware.addInfo("Here is a list of configurators discovered as a service, by rank: ");
146        for(Configurator c: configuratorList) {
147            contextAware.addInfo("  "+c.getClass().getName());
148        }
149        contextAware.addInfo("They will be invoked in order until ExecutionStatus.DO_NOT_INVOKE_NEXT_IF_ANY is returned.");
150    }
151
152    private void printDuration(long start, Configurator configurator, Configurator.ExecutionStatus executionStatus) {
153        long end = System.currentTimeMillis();
154        long diff = end - start;
155        contextAware.addInfo( configurator.getClass().getName()+".configure() call lasted "+diff + " milliseconds. ExecutionStatus="+executionStatus);
156    }
157
158    private Configurator.ExecutionStatus attemptConfigurationUsingJoranUsingReflexion(ClassLoader classLoader) {
159
160        try {
161            Class<?> djcClass = classLoader.loadClass("ch.qos.logback.classic.util.DefaultJoranConfigurator");
162            Configurator c = (Configurator) djcClass.newInstance();
163            c.setContext(loggerContext);
164            return c.configure(loggerContext);
165        } catch (ClassNotFoundException | InstantiationException | IllegalAccessException e) {
166            contextAware.addError("unexpected exception while instantiating DefaultJoranConfigurator", e);
167            return Configurator.ExecutionStatus.INVOKE_NEXT_IF_ANY;
168        }
169
170    }
171
172    Comparator<Configurator> rankComparator = new Comparator<Configurator>() {
173        @Override
174        public int compare(Configurator c1, Configurator c2) {
175
176            ConfiguratorRank r1 = c1.getClass().getAnnotation(ConfiguratorRank.class);
177            ConfiguratorRank r2 = c2.getClass().getAnnotation(ConfiguratorRank.class);
178
179            int value1 = r1 == null ? ConfiguratorRank.DEFAULT : r1.value();
180            int value2 = r2 == null ? ConfiguratorRank.DEFAULT : r2.value();
181
182            int result = compareRankValue(value1, value2);
183            // reverse the result for high to low sort
184            return (-result);
185        }
186    };
187
188    private int compareRankValue(int value1, int value2) {
189        if(value1 > value2)
190            return 1;
191        else if (value1 == value2)
192            return 0;
193        else return -1;
194
195    }
196}