Spring Sleuth 在 SOAP 客户端调用中传播自定义追踪字段的策略

spring:
  sleuth:
    async:
      enabled: true # 启用异步操作的追踪上下文传播
    baggage:
      remote-fields: # 定义需要远程传播的自定义字段
        - Caller-Id

Request headers: {Accept=[application/json], X-B3-TraceId=[...], X-B3-SpanId=[...], X-B3-Sampled=[1], caller-id=[value]}

SOAP Headers - {Authorization=[Bearer...]}

import org.springframework.cloud.sleuth.Tracer;
import org.springframework.cloud.sleuth.CurrentTraceContext;
import org.springframework.cloud.sleuth.Baggage; // 用于获取当前行李字段

import javax.xml.ws.handler.MessageContext;
import javax.xml.ws.handler.soap.SOAPHandler;
import javax.xml.ws.handler.soap.SOAPMessageContext;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Collections;
import java.util.Set;
import javax.xml.namespace.QName;

/**
 * SleuthSoapClientHandler 是一个 JAX-WS SOAP 处理器，用于在出站 SOAP 消息中注入 Sleuth 追踪信息和自定义行李字段。
 */
public class SleuthSoapClientHandler implements SOAPHandler<SOAPMessageContext> {

    private final Tracer tracer;

    /**
     * 构造函数，通过 Spring 注入 Tracer 实例。
     * @param tracer Spring Cloud Sleuth 的 Tracer 实例。
     */
    public SleuthSoapClientHandler(Tracer tracer) {
        this.tracer = tracer;
    }

    @Override
    public boolean handleMessage(SOAPMessageContext context) {
        // 判断消息是否是出站消息（即客户端发往服务端的请求）
        Boolean outboundProperty = (Boolean) context.get(MessageContext.MESSAGE_OUTBOUND_PROPERTY);

        if (outboundProperty != null && outboundProperty) { // 如果是出站消息
            // 获取或创建 HTTP 请求头 Map
            Map<String, List<String>> httpHeaders = (Map<String, List<String>>) context.get(MessageContext.HTTP_REQUEST_HEADERS);
            if (httpHeaders == null) {
                httpHeaders = new HashMap<>();
                context.put(MessageContext.HTTP_REQUEST_HEADERS, httpHeaders);
            }

            // 1. 注入 Sleuth 追踪 ID (X-B3-TraceId, X-B3-SpanId, X-B3-Sampled)
            CurrentTraceContext currentTraceContext = tracer.currentTraceContext();
            if (currentTraceContext != null && currentTraceContext.context() != null) {
                String traceId = currentTraceContext.context().traceId();
                String spanId = currentTraceContext.context().spanId();
                httpHeaders.put("X-B3-TraceId", Collections.singletonList(traceId));
                httpHeaders.put("X-B3-SpanId", Collections.singletonList(spanId));
                httpHeaders.put("X-B3-Sampled", Collections.singletonList("1")); // 假设已采样
            }

            // 2. 注入自定义行李字段 (Baggage Fields)
            // 遍历所有当前活跃的行李字段，并将其值作为 HTTP 头注入
            Baggage.current().getAllFields().forEach(baggageField -> {
                if (baggageField.value() != null) {
                    // Sleuth 行李字段通常以其名称作为 HTTP 头名称进行传播
                    httpHeaders.put(baggageField.name(), Collections.singletonList(baggageField.value()));
                }
            });
            // 如果只需要注入特定的行李字段，可以这样获取：
            // BaggageField callerIdField = Baggage.current().getBaggage("Caller-Id");
            // if (callerIdField != null && callerIdField.value() != null) {
            //     httpHeaders.put("Caller-Id", Collections.singletonList(callerIdField.value()));
            // }
        }
        return true; // 继续处理消息
    }

    @Override
    public boolean handleFault(SOAPMessageContext context) {
        // 处理故障消息（如果需要），这里简单返回 true
        return true;
    }

    @Override
    public void close(MessageContext context) {
        // 资源清理（如果需要）
    }

    @Override
    public Set<QName> getHeaders() {
        // 此处理器不处理特定的 SOAP 头，只处理 HTTP 头，因此返回 null
        return null;
    }
}

import com.example.MyWebService; // 替换为你的 WebService 接口
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.ClassPathResource;
import org.springframework.remoting.jaxws.JaxWsPortProxyFactoryBean;
import org.springframework.cloud.sleuth.Tracer;

import javax.xml.ws.handler.Handler;
import java.util.ArrayList;
import java.util.List;

@Configuration
public class SoapClientConfig {

    @Autowired
    private Tracer tracer; // Spring Boot 会自动注入 Sleuth 的 Tracer 实例

    @Bean
    public MyWebService myWebService() {
        JaxWsPortProxyFactoryBean factory = new JaxWsPortProxyFactoryBean();