com.hibegin.common.util.LoggerUtil Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of simplewebserver Show documentation

Simple, flexible, less dependent, more extended. Less memory footprint, can quickly build Web project. Can quickly run embedded, Android devices

There is a newer version: 0.3.162

Show newest version

package com.hibegin.common.util;

import com.hibegin.http.server.util.PathUtil;

import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.text.SimpleDateFormat;
import java.util.Arrays;
import java.util.Date;
import java.util.Objects;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;
import java.util.logging.FileHandler;
import java.util.logging.Level;
import java.util.logging.Logger;
import java.util.logging.SimpleFormatter;

public class LoggerUtil {

    private static final SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
    private static final String LOG_FILE_SUFFIX = ".log";
    private static FileHandler fileHandler;
    private static final Lock loadLock = new ReentrantLock();

    private LoggerUtil() {
    }

    public static FileHandler buildFileHandle() {
        try {
            File fileName = new File(getLogFilePath());
            if (!fileName.exists()) {
                fileName.getParentFile().mkdirs();
            }
            FileHandler handler = new FileHandler(fileName.toString(), true);
            handler.setFormatter(new SimpleFormatter());
            return handler;
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    public static void initFileHandle(FileHandler _fileHandler) {
        fileHandler = _fileHandler;
    }

    public static FileHandler getFileHandler() {
        return fileHandler;
    }

    public static Logger getLogger(Class clazz) {
        loadLock.lock();
        try {
            Logger logger = Logger.getLogger(clazz.getName());
            //避免重复添加 handle
            if (Arrays.stream(logger.getHandlers()).anyMatch(x -> Objects.equals(x, fileHandler))) {
                return logger;
            }
            try {
                if (Objects.isNull(fileHandler)) {
                    fileHandler = buildFileHandle();
                }
                logger.addHandler(fileHandler);
                logger.setLevel(Level.ALL);
            } catch (Exception e) {
                logger.severe("Init logger error " + e.getMessage());
            }
            return logger;
        } finally {
            loadLock.unlock();
        }
    }

    private synchronized static String getLogFilePath() {
        StringBuilder logFilePath = new StringBuilder();
        logFilePath.append(PathUtil.getLogPath());
        File file = new File(logFilePath.toString());
        if (!file.exists()) {
            file.mkdir();
        }

        logFilePath.append(File.separatorChar);
        logFilePath.append(sdf.format(new Date()));
        logFilePath.append(LOG_FILE_SUFFIX);

        return logFilePath.toString();
    }

    /**
     * 记录完善的异常日志信息(包括堆栈信息)
     */
    public static String recordStackTraceMsg(Throwable e) {
        StringWriter stringWriter = new StringWriter();
        PrintWriter writer = new PrintWriter(stringWriter);
        e.printStackTrace(writer);
        StringBuffer buffer = stringWriter.getBuffer();
        return buffer.toString();
    }


}