com.epam.jdi.light.logger.JDILogger Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of jdi-light Show documentation
There is a newer version: 1.6.0
package com.epam.jdi.light.logger;

/**
 * Created by Roman Iovlev on 14.02.2018
 * Email: [email protected]; Skype: roman.iovlev
 */

import com.epam.jdi.tools.func.JAction;
import com.epam.jdi.tools.func.JFunc;
import com.epam.jdi.tools.map.MapArray;
import org.apache.logging.log4j.Level;
import org.apache.logging.log4j.Logger;
import org.apache.logging.log4j.Marker;
import org.apache.logging.log4j.MarkerManager;

import java.util.ArrayList;
import java.util.List;

import static com.epam.jdi.light.logger.LogLevels.*;
import static com.epam.jdi.tools.StringUtils.format;
import static java.lang.Thread.currentThread;
import static org.apache.logging.log4j.LogManager.getLogger;
import static org.apache.logging.log4j.core.config.Configurator.setRootLevel;

public class JDILogger implements ILogger {
    private static MapArray loggers = new MapArray<>();
    private static Marker jdiMarker = MarkerManager.getMarker("JDI");

    public static JDILogger instance(String name) {
        if (!loggers.has(name))
            loggers.add(name, new JDILogger(name));
        return loggers.get(name);
    }

    public JDILogger() {
        this("JDI");
    }
    public JDILogger(String name) {
        logger = getLogger(name);
        this.name = name;
    }
    public JDILogger(Class clazz) {
        this(clazz.getSimpleName());
    }

    private LogLevels logLevel = INFO;

    public LogLevels getLogLevel() {
        return logLevel;
    }
    public void setLogLevel(LogLevels logLevel) {
        this.logLevel = logLevel;
        setRootLevel(getLog4j2Level(logLevel));
    }
    private LogLevels currentLevel = INFO;
    private int logOffDeepness = 0;
    public void logOff() {
        if (logOffDeepness == 0) {
            currentLevel = logLevel;
            logLevel = OFF;
        }
        logOffDeepness ++;
    }
    public void logOn() {
        logOffDeepness --;
        if (logOffDeepness > 0) return;
        if (logOffDeepness == 0)
            logLevel = currentLevel;
        if (logOffDeepness <0)
            throw new RuntimeException("Log Off Deepness to high. Please check that each logOff has appropriate logOn");
    }
    public void dropLogOff() {
        logOffDeepness = 0;
        logLevel =currentLevel;
    }
    public void logOff(JAction action) {
        logOff(() -> { action.invoke(); return null; });
    }
    public  T logOff(JFunc func) {
        LogLevels currentLevel = logLevel;
        if (logLevel == OFF) {
            try { return func.invoke();
            } catch (Exception ex) { throw new RuntimeException(ex); }
        }
        logLevel = OFF;
        T result;
        try{ result = func.invoke(); }
        catch (Exception ex) {throw new RuntimeException(ex); }
        logLevel = currentLevel;
        return result;
    }
    private String name;
    private Logger logger;
    private List multiThread = new ArrayList<>();
    private String getRecord(String record) {
        if (!multiThread.contains(currentThread().getId()))
            multiThread.add(currentThread().getId());
        return multiThread.size() > 1
                ? "[" + currentThread().getId() + "]" + record
                : record;
    }

    public String getName() {
        return name;
    }

    public void step(String s, Object... args) {
        if (logLevel.equalOrLessThan(STEP))
            logger.log(Level.forName("STEP", 350), jdiMarker, getRecord(format(s, args)));
    }
    public void trace(String s, Object... args) {
        if (logLevel.equalOrLessThan(TRACE))
            logger.trace(jdiMarker, getRecord(format(s, args)));
    }
    public void debug(String s, Object... args) {
        if (logLevel.equalOrLessThan(DEBUG))
            logger.debug(jdiMarker, getRecord(format(s, args)));
    }
    public void info(String s, Object... args) {
        if (logLevel.equalOrLessThan(INFO))
            logger.info(jdiMarker, getRecord(format(s, args)));
    }
    public void error(String s, Object... args) {
        logger.error(jdiMarker, getRecord(format(s, args)));
    }

    public void toLog(String msg) {
        toLog(msg, currentLevel);
    }
    public void toLog(String msg, LogLevels level) {
        if (logLevel.equalOrLessThan(level))
            switch (level) {
                case ERROR: error(msg); break;
                case STEP: step(msg); break;
                case INFO: info(msg); break;
                case DEBUG: debug(msg); break;
                default: throw new RuntimeException("Unknown log level: " + level);
            }
    }
}