io.quarkiverse.reactive.messaging.nats.jetstream.mapper.DefaultMessageMapper 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.mapper;

import static io.smallrye.reactive.messaging.tracing.TracingUtils.traceIncoming;

import java.time.Duration;

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

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

import io.quarkiverse.reactive.messaging.nats.jetstream.ExponentialBackoff;
import io.quarkiverse.reactive.messaging.nats.jetstream.JetStreamIncomingMessage;
import io.quarkiverse.reactive.messaging.nats.jetstream.tracing.JetStreamInstrument;
import io.quarkiverse.reactive.messaging.nats.jetstream.tracing.JetStreamTrace;
import io.quarkus.arc.DefaultBean;
import io.vertx.mutiny.core.Context;

@ApplicationScoped
@DefaultBean
public class DefaultMessageMapper implements MessageMapper {

    public static final String MESSAGE_TYPE_HEADER = "message.type";

    private final PayloadMapper payloadMapper;
    private final JetStreamInstrument instrumenter;

    @Inject
    public DefaultMessageMapper(PayloadMapper payloadMapper,
            JetStreamInstrument instrumenter) {
        this.payloadMapper = payloadMapper;
        this.instrumenter = instrumenter;
    }

    @SuppressWarnings("unchecked")
    @Override
    public  org.eclipse.microprofile.reactive.messaging.Message of(
            io.nats.client.Message message,
            boolean tracingEnabled,
            Class payloadType,
            Context context,
            ExponentialBackoff exponentialBackoff,
            Duration ackTimeout) {
        try {
            final var incomingMessage = payloadType != null
                    ? new JetStreamIncomingMessage<>(message, payloadMapper.of(message, payloadType), context,
                            exponentialBackoff, ackTimeout)
                    : new JetStreamIncomingMessage<>(message,
                            payloadMapper.of(message).orElseGet(() -> message.getData()),
                            context,
                            exponentialBackoff, ackTimeout);
            if (tracingEnabled) {
                return (Message) traceIncoming(instrumenter.receiver(), incomingMessage,
                        JetStreamTrace.trace(incomingMessage));
            } else {
                return (Message) incomingMessage;
            }
        } catch (ClassCastException e) {
            throw new RuntimeException(e);
        }
    }

}