com.lwq.fast.log.fastlogclient.logback.appender.util.MessageUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fast-log-client Show documentation
fast-log-client
The newest version!
package com.lwq.fast.log.fastlogclient.logback.appender.util;

import ch.qos.logback.classic.Level;
import ch.qos.logback.classic.spi.ILoggingEvent;
import ch.qos.logback.classic.spi.ThrowableProxy;
import cn.hutool.core.text.StrBuilder;
import cn.hutool.core.util.IdUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import com.lwq.fast.log.fastlogcore.entity.Message;
import com.lwq.fast.log.fastlogcore.trace.TraceThreadLocal;

import java.io.IOException;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.Date;

/**
 * @author 刘文强
 */
public class MessageUtil {



    private final static String rn = "\r\n";
    /**
     *格式化日志
     *
     * @param appName 应用名称
     * @param env 环境信息
     * @param iLoggingEvent
     * @return Message
     */
    public static Message formatMessage(String appName, String env, ILoggingEvent iLoggingEvent) {
        if (ObjectUtil.isNull(iLoggingEvent)){
            return null;
        }


        String content = iLoggingEvent.getFormattedMessage();
        StrBuilder strBuilder = StrBuilder.create(content);
        if (StrUtil.equals(Level.ERROR.levelStr, iLoggingEvent.getLevel().levelStr)){
            // 出现error 级别的日志，获取错误信息
            ThrowableProxy throwableProxy =  (ThrowableProxy) iLoggingEvent.getThrowableProxy();
            Object errorStackTrace = getErrorStackTrace(throwableProxy.getThrowable());
            strBuilder.append(errorStackTrace);
        }
        content = strBuilder.toString();

        return  Message.builder().env(env)
                .appName(appName)
                .threadName(iLoggingEvent.getThreadName())
                .level(iLoggingEvent.getLevel().levelStr)
                .traceId(StrUtil.isBlank(TraceThreadLocal.get()) ? IdUtil.simpleUUID() : TraceThreadLocal.get())
                .className(iLoggingEvent.getLoggerName())
                .content(content)
                .dateTime(new Date(iLoggingEvent.getTimeStamp()))
                .build();
    }


    /**
     * 获取错误堆栈信息
     *
     * @param errorObj
     * @return
     */
    private static Object getErrorStackTrace(Object errorObj) {
        if (errorObj instanceof Exception) {
            Exception exception = (Exception) errorObj;
            StringWriter stringWriter = new StringWriter();
            PrintWriter printWriter = new PrintWriter(stringWriter);
            String stackTrace = rn;
            try {
                exception.printStackTrace(printWriter);
                stackTrace = stringWriter.toString();
                return stackTrace;
            } catch (Exception e) {
                e.printStackTrace();
                return errorObj;
            } finally {
                try {
                    printWriter.close();
                    stringWriter.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        } else if(errorObj instanceof Error){
            Error error = (Error) errorObj;
            StringWriter stringWriter = new StringWriter();
            PrintWriter printWriter = new PrintWriter(stringWriter);
            String stackTrace = rn;
            try {
                error.printStackTrace(printWriter);
                stackTrace = stringWriter.toString();
                return stackTrace;
            } catch (Exception e) {
                e.printStackTrace();
                return errorObj;
            } finally {
                try {
                    printWriter.close();
                    stringWriter.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }else {
            return errorObj;
        }
    }


}