io.quarkiverse.reactive.messaging.nats.jetstream.tracing.JetStreamInstrument Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of quarkus-messaging-nats-jetstream Show documentation

There is a newer version: 3.17.0

Show newest version

package io.quarkiverse.reactive.messaging.nats.jetstream.tracing;

import static io.opentelemetry.instrumentation.api.instrumenter.messaging.MessageOperation.RECEIVE;
import static io.opentelemetry.instrumentation.api.instrumenter.messaging.MessagingAttributesExtractor.create;
import static io.smallrye.reactive.messaging.tracing.TracingUtils.getOpenTelemetry;

import jakarta.enterprise.context.ApplicationScoped;
import jakarta.enterprise.inject.Instance;
import jakarta.inject.Inject;

import io.opentelemetry.api.OpenTelemetry;
import io.opentelemetry.instrumentation.api.instrumenter.Instrumenter;
import io.opentelemetry.instrumentation.api.instrumenter.InstrumenterBuilder;
import io.opentelemetry.instrumentation.api.instrumenter.messaging.MessageOperation;
import io.opentelemetry.instrumentation.api.instrumenter.messaging.MessagingAttributesExtractor;
import io.opentelemetry.instrumentation.api.instrumenter.messaging.MessagingAttributesGetter;
import io.opentelemetry.instrumentation.api.instrumenter.messaging.MessagingSpanNameExtractor;

@ApplicationScoped
public class JetStreamInstrument {
    private final Instance openTelemetryInstance;

    @Inject
    public JetStreamInstrument(Instance openTelemetryInstance) {
        this.openTelemetryInstance = openTelemetryInstance;
    }

    public Instrumenter publisher() {
        final var attributesExtractor = new JetStreamTraceAttributesExtractor();
        MessagingAttributesGetter messagingAttributesGetter = attributesExtractor
                .getMessagingAttributesGetter();

        InstrumenterBuilder builder = Instrumenter.builder(getOpenTelemetry(openTelemetryInstance),
                "io.smallrye.reactive.messaging.jetstream",
                MessagingSpanNameExtractor.create(messagingAttributesGetter, MessageOperation.SEND));

        return builder.addAttributesExtractor(create(messagingAttributesGetter, MessageOperation.SEND))
                .addAttributesExtractor(attributesExtractor)
                .buildProducerInstrumenter(JetStreamTraceTextMapSetter.INSTANCE);
    }

    public Instrumenter receiver() {
        final var attributesExtractor = new JetStreamTraceAttributesExtractor();
        MessagingAttributesGetter messagingAttributesGetter = attributesExtractor
                .getMessagingAttributesGetter();
        InstrumenterBuilder builder = Instrumenter.builder(getOpenTelemetry(openTelemetryInstance),
                "io.smallrye.reactive.messaging.jetstream",
                MessagingSpanNameExtractor.create(messagingAttributesGetter, RECEIVE));

        return builder.addAttributesExtractor(attributesExtractor)
                .addAttributesExtractor(MessagingAttributesExtractor.create(messagingAttributesGetter, RECEIVE))
                .buildConsumerInstrumenter(JetStreamTraceTextMapGetter.INSTANCE);
    }

}