io.smallrye.reactive.messaging.amqp.tracing.AmqpAttributesExtractor 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 java.util.Collections;
import java.util.List;

import io.opentelemetry.api.common.AttributesBuilder;
import io.opentelemetry.context.Context;
import io.opentelemetry.instrumentation.api.incubator.semconv.messaging.MessagingAttributesGetter;
import io.opentelemetry.instrumentation.api.instrumenter.AttributesExtractor;
import io.smallrye.reactive.messaging.amqp.AmqpMessage;

public class AmqpAttributesExtractor implements AttributesExtractor, Void> {
    private final MessagingAttributesGetter, Void> messagingAttributesGetter;

    public AmqpAttributesExtractor() {
        this.messagingAttributesGetter = new AmqpMessagingAttributesGetter();
    }

    public MessagingAttributesGetter, Void> getMessagingAttributesGetter() {
        return messagingAttributesGetter;
    }

    @Override
    public void onStart(
            final AttributesBuilder attributes,
            final Context parentContext,
            final AmqpMessage amqpMessage) {

    }

    @Override
    public void onEnd(
            final AttributesBuilder attributes,
            final Context context,
            final AmqpMessage amqpMessage,
            final Void unused,
            final Throwable error) {
    }

    private static class AmqpMessagingAttributesGetter implements MessagingAttributesGetter, Void> {
        // Required
        @Override
        public String getSystem(final AmqpMessage amqpMessage) {
            return "AMQP 1.0";
        }

        // Required
        @Override
        public String getDestination(final AmqpMessage amqpMessage) {
            return amqpMessage.getAddress();
        }

        @Override
        public boolean isTemporaryDestination(final AmqpMessage amqpMessage) {
            return false;
        }

        // Recommended
        @Override
        public String getConversationId(final AmqpMessage amqpMessage) {
            Object correlationId = amqpMessage.getCorrelationId();
            return correlationId instanceof String ? (String) correlationId : null;
        }

        // Recommended
        @Override
        public String getMessageId(final AmqpMessage amqpMessage, final Void unused) {
            Object messageId = amqpMessage.getMessageId();
            return messageId instanceof String ? (String) messageId : null;
        }

        @Override
        public List getMessageHeader(AmqpMessage amqpMessage, String name) {
            return Collections.emptyList();
        }

        @Override
        public String getDestinationTemplate(AmqpMessage amqpMessage) {
            return null;
        }

        @Override
        public boolean isAnonymousDestination(AmqpMessage amqpMessage) {
            return false;
        }

        @Override
        public Long getMessageBodySize(AmqpMessage amqpMessage) {
            return null;
        }

        @Override
        public Long getMessageEnvelopeSize(AmqpMessage amqpMessage) {
            return null;
        }

        @Override
        public String getClientId(AmqpMessage amqpMessage) {
            return null;
        }

        @Override
        public Long getBatchMessageCount(AmqpMessage amqpMessage, Void unused) {
            return null;
        }
    }
}