fun.fengwk.convention4j.springboot.starter.webflux.tracer.ServerWebExchangeExtract Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of convention4j-spring-boot-starter-webflux Show documentation

There is a newer version: 0.0.43

Show newest version

package fun.fengwk.convention4j.springboot.starter.webflux.tracer;

import com.google.auto.service.AutoService;
import fun.fengwk.convention4j.common.lang.StringUtils;
import fun.fengwk.convention4j.tracer.SpanContextImpl;
import fun.fengwk.convention4j.tracer.propagation.PropagationConstants;
import fun.fengwk.convention4j.tracer.propagation.extract.Extract;
import fun.fengwk.convention4j.tracer.util.TracerUtils;
import io.opentracing.SpanContext;
import io.opentracing.propagation.Format;
import org.springframework.http.HttpHeaders;
import org.springframework.web.server.ServerWebExchange;

/**
 * @author fengwk
 */
@AutoService(Extract.class)
public class ServerWebExchangeExtract implements Extract {

    public static final Format FORMAT = new Format<>() {};

    @Override
    public Format format() {
        return FORMAT;
    }

    @Override
    public SpanContext extract(ServerWebExchange exchange) {
        if (exchange == null) {
            return null;
        }
        HttpHeaders headers = exchange.getRequest().getHeaders();
        String traceId = headers.getFirst(PropagationConstants.TRACE_ID_HTTP_HEADER_NAME);
        String spanId = headers.getFirst(PropagationConstants.SPAN_ID_HTTP_HEADER_NAME);
        String baggage = headers.getFirst(PropagationConstants.BAGGAGE_HEADER_NAME);
        if (StringUtils.isNotBlank(traceId)) {
            return new SpanContextImpl(traceId, spanId, TracerUtils.deserializeHttpPropagationBaggage(baggage));
        }
        return null;
    }

}