com.fireflysource.common.sys.SystemLogger Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of firefly-common Show documentation

The newest version!

package com.fireflysource.common.sys;

import com.fireflysource.common.bytecode.JavassistClassProxyFactory;
import com.fireflysource.common.lifecycle.ShutdownTasks;
import com.fireflysource.common.slf4j.LazyLogger;
import org.slf4j.ILoggerFactory;
import org.slf4j.LoggerFactory;
import org.slf4j.MDC;

import java.io.Closeable;
import java.io.IOException;

/**
 * @author Pengtao Qiu
 */
public class SystemLogger {

    private static final LazyLogger system = LazyLogger.create("firefly-system");

    static {
        ShutdownTasks.register(SystemLogger::stop);
    }

    public static LazyLogger create(Class clazz) {
        try {
            String className = clazz.getSimpleName();
            return JavassistClassProxyFactory.INSTANCE.createProxy(system, ((handler, originalInstance, args) -> {
                try (MDC.MDCCloseable ignored = MDC.putCloseable("class", className)) {
                    return handler.invoke(originalInstance, args);
                }
            }), null);
        } catch (Throwable e) {
            system.error("create system logger exception", e);
            throw new IllegalStateException(e);
        }
    }

    public static void stop() {
        ILoggerFactory iLoggerFactory = LoggerFactory.getILoggerFactory();
        if (iLoggerFactory instanceof Closeable) {
            try {
                ((Closeable) iLoggerFactory).close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}