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

import static io.nats.client.support.NatsJetStreamConstants.MSG_ID_HDR;

import java.time.ZonedDateTime;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;

import io.nats.client.Message;
import io.nats.client.api.MessageInfo;
import io.nats.client.impl.Headers;
import io.smallrye.common.constraint.NotNull;
import lombok.Builder;

@Builder
public record JetStreamIncomingMessageMetadata(String stream,
        String subject,
        String messageId,
        Map> headers,
        Long deliveredCount,
        String consumer,
        Long streamSequence,
        Long consumerSequence,
        ZonedDateTime timestamp) {

    public static JetStreamIncomingMessageMetadata of(@NotNull Message message) {
        final var headers = Optional.ofNullable(message.getHeaders());
        return JetStreamIncomingMessageMetadata.builder()
                .stream(message.metaData().getStream())
                .subject(message.getSubject())
                .messageId(headers.map(h -> h.getFirst(MSG_ID_HDR)).orElse(null))
                .headers(headers.map(JetStreamIncomingMessageMetadata::headers).orElseGet(HashMap::new))
                .deliveredCount(message.metaData().deliveredCount())
                .consumer(message.metaData().getConsumer())
                .streamSequence(message.metaData().streamSequence())
                .consumerSequence(message.metaData().consumerSequence())
                .timestamp(message.metaData().timestamp())
                .build();
    }

    public static JetStreamIncomingMessageMetadata of(MessageInfo message) {
        final var headers = Optional.ofNullable(message.getHeaders());
        return JetStreamIncomingMessageMetadata.builder()
                .stream(message.getStream())
                .subject(message.getSubject())
                .messageId(headers.map(h -> h.getFirst(MSG_ID_HDR)).orElse(null))
                .headers(headers.map(JetStreamIncomingMessageMetadata::headers).orElseGet(HashMap::new))
                .deliveredCount(null)
                .consumer(null)
                .streamSequence(message.getSeq())
                .consumerSequence(null)
                .timestamp(message.getTime())
                .build();
    }

    public static Map> headers(Headers messageHeaders) {
        final var headers = new HashMap>();
        messageHeaders.entrySet().forEach(entry -> headers.put(entry.getKey(), entry.getValue()));
        return headers;
    }

    @Override
    public String toString() {
        return "JetStreamIncomingMessageMetadata{" +
                "stream='" + stream + '\'' +
                ", subject='" + subject + '\'' +
                ", messageId='" + messageId + '\'' +
                ", headers=" + headers +
                ", deliveredCount=" + deliveredCount +
                '}';
    }
}