com.solace.spring.cloud.stream.binder.tracing.TracingImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-cloud-stream-binder-solace Show documentation

A Spring Cloud Stream Binder implementation using the Solace Java API (JCSMP)

The newest version!

package com.solace.spring.cloud.stream.binder.tracing;

import com.solacesystems.jcsmp.JCSMPFactory;
import com.solacesystems.jcsmp.SDTMap;
import io.micrometer.tracing.Span;
import io.micrometer.tracing.Tracer;
import io.micrometer.tracing.propagation.Propagator;
import lombok.RequiredArgsConstructor;
import lombok.SneakyThrows;
import org.springframework.lang.Nullable;
import org.springframework.messaging.Message;

import java.util.function.Consumer;

@RequiredArgsConstructor
public class TracingImpl {
    private final Tracer tracer;
    private final Propagator propagator;

    public SDTMap getTracingHeader() {
        Span span = tracer.currentSpan();
        if (span != null) {
            SDTMap tracingHeader = JCSMPFactory.onlyInstance().createMap();
            propagator.inject(span.context(), tracingHeader, this::putString);
            return tracingHeader;
        }
        return null;
    }


    public Consumer> wrapInTracingContext(SDTMap tracingHeader, Consumer> messageConsumer) {
        return (message) -> {
            Span.Builder extractedSpan = propagator.extract(tracingHeader, this::getString);
            Span span = extractedSpan.start();
            try (Tracer.SpanInScope ignored = tracer.withSpan(span)) {
                messageConsumer.accept(message);
            } finally {
                span.end();
            }
        };
    }

    @SneakyThrows
    private void putString(@Nullable SDTMap map, String key, String value) {
        if (map != null) {
            map.putString(key, value);
        }
    }

    @SneakyThrows
    private String getString(@Nullable SDTMap map, String key) {
        if (map != null) {
            return map.getString(key);
        }
        return null;
    }
}