com.taobao.middleware.logger.option.Log4jActivateOption Maven / Gradle / Ivy

Go to download
package com.taobao.middleware.logger.option;

import com.taobao.middleware.logger.Logger;
import com.taobao.middleware.logger.support.LoggerHelper;
import org.apache.log4j.*;

import java.io.File;
import java.util.Enumeration;

/**
 * ActivateOption的Log4j实现
 * 
 * @author zhuyong 2014年3月20日 上午10:24:36
 */
public class Log4jActivateOption extends AbstractActiveOption {

    protected org.apache.log4j.Logger logger;

    public Log4jActivateOption(org.apache.log4j.Logger logger) {
        this.logger = logger;
    }

    @Override
    public void activateConsoleAppender(String target, String encoding) {
        org.apache.log4j.ConsoleAppender appender = new org.apache.log4j.ConsoleAppender();
        appender.setLayout(new PatternLayout(LoggerHelper.getPattern()));
        appender.setTarget(target);
        appender.setEncoding(encoding);
        appender.activateOptions();

        logger.removeAllAppenders();
        logger.addAppender(appender);
    }

    @Override
    public void activateAppender(String productName, String file, String encoding) {
        org.apache.log4j.Appender appender = getLog4jDailyRollingFileAppender(productName, file, encoding);
        logger.removeAllAppenders();
        logger.addAppender(appender);

        setProductName(productName);
    }

    @Override
    public void activateAsyncAppender(String productName, String file, String encoding) {
        AsyncAppender asynAppender = new AsyncAppender();
        asynAppender.setName(productName + "." + file.replace(File.separatorChar, '.') + ".AsyncAppender");
        org.apache.log4j.Appender appender = getLog4jDailyRollingFileAppender(productName, file, encoding);
        asynAppender.addAppender(appender);
        logger.removeAllAppenders();
        logger.addAppender(asynAppender);

        setProductName(productName);
    }

    @Override
    public void activateAppenderWithTimeAndSizeRolling(String productName, String file, String encoding, String size) {
        activateAppender(productName, file, encoding);
    }

    @Override
    public void setLevel(com.taobao.middleware.logger.Level level) {
        logger.setLevel(org.apache.log4j.Level.toLevel(level.getName()));
    }

    @Override
    public void setAdditivity(boolean additivity) {
        logger.setAdditivity(additivity);
    }

    protected org.apache.log4j.Appender getLog4jDailyRollingFileAppender(String productName, String file,
                                                                         String encoding) {
        DailyRollingFileAppender appender = new DailyRollingFileAppender();
        appender.setName(productName + "." + file.replace(File.separatorChar, '.') + ".Appender");
        appender.setLayout(new PatternLayout(LoggerHelper.getPattern(productName)));
        appender.setAppend(true);
        appender.setFile(LoggerHelper.getLogFileP(productName, file));
        appender.setEncoding(encoding);
        appender.activateOptions();

        return appender;
    }

    @Override
    public void activateAppender(Logger logger) {
        if (!(logger.getDelegate() instanceof org.apache.log4j.Logger)) {
            throw new IllegalArgumentException("logger must be org.apache.log4j.Logger, but it's "
                                               + logger.getDelegate().getClass());
        }
        activateAppender((org.apache.log4j.Logger) logger.getDelegate());

        setProductName(logger.getProductName());
    }

    protected void activateAppender(org.apache.log4j.Logger logger) {
        this.logger.removeAllAppenders();

        Enumeration enums = logger.getAllAppenders();
        while (enums != null && enums.hasMoreElements()) {
            this.logger.addAppender((Appender) enums.nextElement());
        }
    }

    @Override
    public void activateAppenderWithTimeAndSizeRolling(String productName, String file, String encoding, String size,
                                                       String datePattern) {
        Appender appender = getLog4jRollingFileAppender(productName, file, encoding, size, datePattern, -1);

        logger.removeAllAppenders();
        logger.addAppender(appender);

        setProductName(productName);
    }

    @Override
    public void activateAppenderWithTimeAndSizeRolling(String productName, String file, String encoding, String size,
                                                       String datePattern, int maxBackupIndex) {
        Appender appender = getLog4jRollingFileAppender(productName, file, encoding, size, datePattern, maxBackupIndex);
        logger.removeAllAppenders();
        logger.addAppender(appender);

        setProductName(productName);
    }

    protected org.apache.log4j.Appender getLog4jRollingFileAppender(String productName, String file, String encoding,
                                                                    String size, String datePattern, int maxBackupIndex) {
        RollingFileAppender appender = new RollingFileAppender();
        appender.setName(productName + "." + file.replace(File.separatorChar, '.') + ".Appender");
        appender.setLayout(new PatternLayout(LoggerHelper.getPattern(productName)));
        appender.setAppend(true);
        appender.setFile(LoggerHelper.getLogFileP(productName, file));
        appender.setEncoding(encoding);
        appender.setMaxFileSize(size);
        if (maxBackupIndex >= 0) {
            // 等于0表示直接truck
            appender.setMaxBackupIndex(maxBackupIndex);
        }
        appender.activateOptions();

        return appender;
    }

    @Override
    public void activateAppenderWithSizeRolling(String productName, String file, String encoding, String size,
                                                int maxBackupIndex) {
        Appender appender = getLog4jRollingFileAppender(productName, file, encoding, size, null, maxBackupIndex);
        logger.removeAllAppenders();
        logger.addAppender(appender);

        setProductName(productName);
    }
}