io.github.novareseller.boot.interceptor.RetrofitTraceInterceptor Maven / Gradle / Ivy

Go to download

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

There is a newer version: 1.5.3.beta

package io.github.novareseller.boot.interceptor;

import cn.hutool.core.net.NetUtil;
import com.github.lianjiatech.retrofit.spring.boot.interceptor.BaseGlobalInterceptor;
import io.github.novareseller.log.constant.LogConstants;
import io.github.novareseller.log.context.LogContext;
import io.github.novareseller.log.context.SpanIdGenerator;
import io.github.novareseller.tool.utils.Validator;
import lombok.Getter;
import lombok.Setter;
import okhttp3.Request;
import okhttp3.Response;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import java.io.IOException;

/**
 * @author Bowen Huang
 * @Date 2021/5/5 17:45
 */
@Component
public class RetrofitTraceInterceptor extends BaseGlobalInterceptor {

    @Getter
    @Setter
    @Value("${spring.application.name}")
    private String appName;

    @Override
    protected Response doIntercept(Chain chain) throws IOException {
        Request request = chain.request();
        String traceId = LogContext.getTraceId();
        if (Validator.isNullOrEmpty(traceId)) {
            return chain.proceed(request);
        }

        String localHostName = NetUtil.getLocalHostName();
        String preIp = LogContext.getPreIp();

        Request newReq = request.newBuilder()
                .addHeader(LogConstants.LOG_TRACE_KEY, Validator.isNullOrEmpty(traceId) ? LogConstants.UNKNOWN : traceId)
                .addHeader(LogConstants.PRE_IVK_APP_KEY, Validator.isNullOrEmpty(appName) ? LogConstants.UNKNOWN : appName)
                .addHeader(LogConstants.PRE_IVK_APP_HOST, Validator.isNullOrEmpty(localHostName) ? LogConstants.UNKNOWN : localHostName)
                .addHeader(LogConstants.PRE_IP_KEY, Validator.isNullOrEmpty(preIp) ? LogConstants.UNKNOWN : preIp)
                .addHeader(LogConstants.LOG_SPAN_ID_KEY, SpanIdGenerator.generateNextSpanId())
                .build();
        return chain.proceed(newReq);
    }
}