All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • io.smallrye.reactive.messaging.amqp.AmqpMessage Maven / Gradle / Ivy

    The newest version!
    package io.smallrye.reactive.messaging.amqp;

import static io.smallrye.reactive.messaging.providers.locals.ContextAwareMessage.captureContextMetadata;

import java.util.ArrayList;
import java.util.concurrent.CompletionStage;
import java.util.function.BiFunction;
import java.util.function.Function;

import org.apache.qpid.proton.amqp.Binary;
import org.apache.qpid.proton.amqp.messaging.AmqpSequence;
import org.apache.qpid.proton.amqp.messaging.AmqpValue;
import org.apache.qpid.proton.amqp.messaging.Data;
import org.apache.qpid.proton.amqp.messaging.Section;
import org.apache.qpid.proton.message.Message;
import org.apache.qpid.proton.message.MessageError;
import org.eclipse.microprofile.reactive.messaging.Metadata;

import io.smallrye.reactive.messaging.amqp.ce.AmqpCloudEventHelper;
import io.smallrye.reactive.messaging.amqp.fault.AmqpFailureHandler;
import io.smallrye.reactive.messaging.ce.CloudEventMetadata;
import io.smallrye.reactive.messaging.providers.MetadataInjectableMessage;
import io.smallrye.reactive.messaging.providers.helpers.VertxContext;
import io.smallrye.reactive.messaging.providers.locals.ContextAwareMessage;
import io.vertx.core.json.JsonObject;
import io.vertx.mutiny.core.Context;
import io.vertx.mutiny.core.buffer.Buffer;

public class AmqpMessage implements ContextAwareMessage, MetadataInjectableMessage {

    protected static final String APPLICATION_JSON = "application/json";
    protected final io.vertx.amqp.AmqpMessage message;
    protected Metadata metadata;
    protected final IncomingAmqpMetadata amqpMetadata;
    private final Context context;
    protected final AmqpFailureHandler onNack;

    private final T payload;

    @Deprecated
    public static  AmqpMessageBuilder builder() {
        return new AmqpMessageBuilder<>();
    }

    public AmqpMessage(io.vertx.mutiny.amqp.AmqpMessage delegate, Context context, AmqpFailureHandler onNack,
            boolean cloudEventEnabled, Boolean tracingEnabled) {
        this(delegate.getDelegate(), context, onNack, cloudEventEnabled, tracingEnabled);
    }

    public AmqpMessage(io.vertx.mutiny.amqp.AmqpMessage delegate, Context context,
            OutgoingAmqpMetadata amqpMetadata) {
        this.message = delegate.getDelegate();
        this.context = context;
        this.amqpMetadata = null;
        this.onNack = null;
        //noinspection unchecked
        this.payload = (T) convert(message);
        this.metadata = Metadata.of(amqpMetadata);
    }

    @SuppressWarnings("unchecked")
    public AmqpMessage(io.vertx.amqp.AmqpMessage msg, Context context, AmqpFailureHandler onNack,
            boolean cloudEventEnabled, Boolean tracingEnabled) {
        this.message = msg;
        this.context = context;
        this.amqpMetadata = new IncomingAmqpMetadata(this.message);
        this.onNack = onNack;

        ArrayList meta = new ArrayList<>();
        meta.add(this.amqpMetadata);
        if (cloudEventEnabled) {
            // Cloud Event detection
            AmqpCloudEventHelper.CloudEventMode mode = AmqpCloudEventHelper.getCloudEventMode(msg);
            switch (mode) {
                case NOT_A_CLOUD_EVENT:
                    payload = (T) convert(message);
                    break;
                case STRUCTURED:
                    CloudEventMetadata event = AmqpCloudEventHelper
                            .createFromStructuredCloudEvent(msg);
                    meta.add(event);
                    payload = event.getData();
                    break;
                case BINARY:
                    payload = (T) convert(message);
                    meta.add(AmqpCloudEventHelper.createFromBinaryCloudEvent(msg, this));
                    break;
                default:
                    payload = (T) convert(message);
            }
        } else {
            payload = (T) convert(message);
        }

        this.metadata = captureContextMetadata(meta);
    }

    @Override
    public CompletionStage ack(Metadata metadata) {
        // We must switch to the context having created the message.
        // This context is passed when this instance of message is created.
        // It's more a Vert.x AMQP client issue which should ensure calling `accepted` on the right context.
        return VertxContext.runOnContext(context.getDelegate(), f -> {
            message.accepted();
            this.runOnMessageContext(() -> f.complete(null));
        });
    }

    @Override
    public CompletionStage nack(Throwable reason, Metadata metadata) {
        return onNack.handle(this, context, reason);
    }

    @Override
    public T getPayload() {
        return payload;
    }

    @Override
    public Metadata getMetadata() {
        return metadata;
    }

    private Object convert(io.vertx.amqp.AmqpMessage msg) {
        Object body = msg.unwrap().getBody();
        if (body instanceof AmqpValue) {
            Object value = ((AmqpValue) body).getValue();
            if (value instanceof Binary) {
                Binary bin = (Binary) value;
                byte[] bytes = new byte[bin.getLength()];
                System.arraycopy(bin.getArray(), bin.getArrayOffset(), bytes, 0, bin.getLength());
                return bytes;
            }
            return value;
        }

        if (body instanceof AmqpSequence) {
            return ((AmqpSequence) body).getValue();
        }

        if (body instanceof Data) {
            Binary bin = ((Data) body).getValue();
            byte[] bytes = new byte[bin.getLength()];
            System.arraycopy(bin.getArray(), bin.getArrayOffset(), bytes, 0, bin.getLength());

            if (APPLICATION_JSON.equalsIgnoreCase(msg.contentType())) {
                return Buffer.buffer(bytes).toJson();
            }
            return bytes;
        }

        return body;
    }

    public Message unwrap() {
        return message.unwrap();
    }

    public boolean isDurable() {
        return amqpMetadata.isDurable();
    }

    public long getDeliveryCount() {
        return amqpMetadata.getDeliveryCount();
    }

    public int getPriority() {
        return amqpMetadata.getPriority();
    }

    public long getTtl() {
        return amqpMetadata.getTtl();
    }

    public Object getMessageId() {
        return amqpMetadata.getId();
    }

    public long getGroupSequence() {
        return amqpMetadata.getGroupSequence();
    }

    public long getCreationTime() {
        return amqpMetadata.getCreationTime();
    }

    public String getAddress() {
        return amqpMetadata.getAddress();
    }

    public String getGroupId() {
        return amqpMetadata.getGroupId();
    }

    public String getContentType() {
        return amqpMetadata.getContentType();
    }

    public long getExpiryTime() {
        return amqpMetadata.getExpiryTime();
    }

    public Object getCorrelationId() {
        return amqpMetadata.getCorrelationId();
    }

    public String getContentEncoding() {
        return amqpMetadata.getContentEncoding();
    }

    public String getSubject() {
        return amqpMetadata.getSubject();
    }

    public JsonObject getApplicationProperties() {
        return amqpMetadata.getProperties();
    }

    public Section getBody() {
        return message.unwrap().getBody();
    }

    public MessageError getError() {
        return message.unwrap().getError();
    }

    public io.vertx.mutiny.amqp.AmqpMessage getAmqpMessage() {
        return new io.vertx.mutiny.amqp.AmqpMessage(message);
    }

    @Override
    public Function> getAckWithMetadata() {
        return this::ack;
    }

    @Override
    public BiFunction> getNackWithMetadata() {
        return this::nack;
    }

    @Override
    public synchronized void injectMetadata(Object metadataObject) {
        metadata = metadata.with(metadataObject);
    }

}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy