io.smallrye.reactive.messaging.amqp.tracing.AmqpOpenTelemetryInstrumenter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of smallrye-reactive-messaging-amqp Show documentation
There is a newer version: 4.25.0
Show newest version
package io.smallrye.reactive.messaging.amqp.tracing;

import jakarta.enterprise.inject.Instance;

import org.eclipse.microprofile.reactive.messaging.Message;

import io.opentelemetry.api.OpenTelemetry;
import io.opentelemetry.instrumentation.api.incubator.semconv.messaging.MessageOperation;
import io.opentelemetry.instrumentation.api.incubator.semconv.messaging.MessagingAttributesExtractor;
import io.opentelemetry.instrumentation.api.incubator.semconv.messaging.MessagingAttributesGetter;
import io.opentelemetry.instrumentation.api.incubator.semconv.messaging.MessagingSpanNameExtractor;
import io.opentelemetry.instrumentation.api.instrumenter.Instrumenter;
import io.opentelemetry.instrumentation.api.instrumenter.InstrumenterBuilder;
import io.smallrye.reactive.messaging.amqp.AmqpMessage;
import io.smallrye.reactive.messaging.tracing.TracingUtils;

public class AmqpOpenTelemetryInstrumenter {
    private final Instrumenter, Void> instrumenter;

    private AmqpOpenTelemetryInstrumenter(Instrumenter, Void> instrumenter) {
        this.instrumenter = instrumenter;
    }

    public static AmqpOpenTelemetryInstrumenter createForConnector(Instance openTelemetryInstance) {
        return create(TracingUtils.getOpenTelemetry(openTelemetryInstance), false);
    }

    public static AmqpOpenTelemetryInstrumenter createForSender(Instance openTelemetryInstance) {
        return create(TracingUtils.getOpenTelemetry(openTelemetryInstance), true);
    }

    private static AmqpOpenTelemetryInstrumenter create(OpenTelemetry openTelemetry, boolean sender) {
        MessageOperation messageOperation = sender ? MessageOperation.PUBLISH : MessageOperation.RECEIVE;
        AmqpAttributesExtractor amqpAttributesExtractor = new AmqpAttributesExtractor();
        MessagingAttributesGetter, Void> messagingAttributesGetter = amqpAttributesExtractor
                .getMessagingAttributesGetter();
        InstrumenterBuilder, Void> builder = Instrumenter.builder(openTelemetry,
                "io.smallrye.reactive.messaging",
                MessagingSpanNameExtractor.create(messagingAttributesGetter, messageOperation));

        builder
                .addAttributesExtractor(
                        MessagingAttributesExtractor.create(messagingAttributesGetter, messageOperation))
                .addAttributesExtractor(amqpAttributesExtractor)
                .buildProducerInstrumenter(AmqpMessageTextMapSetter.INSTANCE);

        Instrumenter, Void> instrumenter;
        if (sender) {
            instrumenter = builder.buildProducerInstrumenter(AmqpMessageTextMapSetter.INSTANCE);
        } else {
            instrumenter = builder.buildConsumerInstrumenter(AmqpMessageTextMapGetter.INSTANCE);
        }
        return new AmqpOpenTelemetryInstrumenter(instrumenter);
    }

    public Message traceIncoming(Message m, AmqpMessage trace) {
        return TracingUtils.traceIncoming(instrumenter, m, (AmqpMessage) trace);
    }

    public void traceOutgoing(Message msg, AmqpMessage