Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
package com.paypal.butterfly.cli.logging;
import ch.qos.logback.classic.Level;
import ch.qos.logback.classic.LoggerContext;
import ch.qos.logback.classic.encoder.PatternLayoutEncoder;
import ch.qos.logback.classic.spi.ILoggingEvent;
import ch.qos.logback.core.Appender;
import ch.qos.logback.core.ConsoleAppender;
import ch.qos.logback.core.OutputStreamAppender;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;
/**
* Logback specific logging configurator
*
* @author facarvalho
*/
@Component
public class LogbackLogConfigurator extends LogConfigurator {
private static final LoggerContext loggerContext = (LoggerContext) LoggerFactory.getILoggerFactory();
private boolean verboseMode = false;
@Override
public void setLoggerLevel(String logger, org.slf4j.event.Level level) {
if(level == null) {
throw new IllegalArgumentException("level argument cannot be null");
}
loggerContext.getLogger(logger).setLevel(getLogbackLogLevel(level));
}
@Override
public void setLoggerLevel(Class logger, org.slf4j.event.Level level) {
if(level == null) {
throw new IllegalArgumentException("level argument cannot be null");
}
loggerContext.getLogger(logger).setLevel(getLogbackLogLevel(level));
}
private Level getLogbackLogLevel(org.slf4j.event.Level slf4jLevel) {
if(slf4jLevel.equals(org.slf4j.event.Level.INFO)) return Level.INFO;
if(slf4jLevel.equals(org.slf4j.event.Level.DEBUG)) return Level.DEBUG;
if(slf4jLevel.equals(org.slf4j.event.Level.WARN)) return Level.WARN;
if(slf4jLevel.equals(org.slf4j.event.Level.ERROR)) return Level.ERROR;
throw new IllegalArgumentException("Unknown log level");
}
@Override
public void setVerboseMode(boolean verboseMode) {
this.verboseMode = verboseMode;
if (verboseMode) {
PatternLayoutEncoder patternLayoutEncoder = new PatternLayoutEncoder();
patternLayoutEncoder.setPattern("[%d{HH:mm:ss.SSS}] [%highlight(%level)] %msg%n");
patternLayoutEncoder.setContext(loggerContext);
patternLayoutEncoder.start();
Appender<ILoggingEvent> consoleAppender = new ConsoleAppender();
((OutputStreamAppender) consoleAppender).setEncoder(patternLayoutEncoder);
consoleAppender.setContext(loggerContext);
consoleAppender.start();
loggerContext.getLogger("com.paypal.butterfly.cli").detachAppender("CONSOLE");
loggerContext.getLogger("ROOT").addAppender(consoleAppender);
// } else {
// TODO
}
}
@Override
public void setLogToFile(boolean on) {
loggerContext.getLogger("ROOT").detachAppender("FILE");
}
@Override
public boolean isVerboseMode() {
return verboseMode;
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment