com.suchtool.nicelog.util.NiceLogTraceIdUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of nicelog-spring-boot-starter Show documentation
collect log automatically for SpringBoot project
The newest version!
package com.suchtool.nicelog.util;

import com.suchtool.nicelog.property.NiceLogProperty;
import com.suchtool.nicelog.util.log.context.NiceLogContext;
import com.suchtool.nicelog.util.log.context.NiceLogContextThreadLocal;
import com.suchtool.nicetool.util.spring.ApplicationContextHolder;
import org.apache.skywalking.apm.toolkit.trace.TraceContext;
import org.springframework.util.StringUtils;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;
import java.util.UUID;

/**
 * traceId工具类
 */
public class NiceLogTraceIdUtil {
    public static String readTraceId() {
        String traceId = null;

        NiceLogContext niceLogContext = NiceLogContextThreadLocal.read();
        if (niceLogContext != null) {
            traceId = niceLogContext.getTraceId();
            if (!StringUtils.hasText(traceId)) {
                traceId = readOrCreateTraceId();
                niceLogContext.setTraceId(traceId);
            }
        }

        if (!StringUtils.hasText(traceId)) {
            traceId = readOrCreateTraceId();
        }

        return traceId;
    }

    private static String readOrCreateTraceId() {
        String traceId = null;

        // 如果header里有traceId，则取出
        ServletRequestAttributes servletRequestAttributes =
                (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        if (servletRequestAttributes != null) {
            HttpServletRequest request = servletRequestAttributes.getRequest();
            NiceLogProperty niceLogProperty = ApplicationContextHolder.getContext().getBean(NiceLogProperty.class);
            String traceIdOfHeader = request.getHeader(niceLogProperty.getFeignTraceIdHeader());
            if (StringUtils.hasText(traceIdOfHeader)) {
                traceId = traceIdOfHeader;
            }
        }

        if (!StringUtils.hasText(traceId)) {
            // 如果使用了skywalking，则使用它的traceId
            String skywalkingTraceId = TraceContext.traceId();
            if (StringUtils.hasText(skywalkingTraceId)) {
                int length = skywalkingTraceId.length();

                // 如果太短，说明有问题。正常traceId超过25
                if (length > 25) {
                    traceId = skywalkingTraceId;
                }
            }
        }

        if (!StringUtils.hasText(traceId)) {
            // 通过uuid创建一个traceId
            traceId = UUID.randomUUID().toString().replace("-", "");
        }

        return traceId;
    }
}