io.smallrye.reactive.messaging.gcp.pubsub.PubSubMessage Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of smallrye-reactive-messaging-gcp-pubsub Show documentation

A module for Smallrye reactive messaging integration with GCP Pub/Sub Service

The newest version!

package io.smallrye.reactive.messaging.gcp.pubsub;

import static io.smallrye.reactive.messaging.gcp.pubsub.i18n.PubSubMessages.msg;

import java.util.Objects;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CompletionStage;
import java.util.function.Function;

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

import com.google.cloud.pubsub.v1.AckReplyConsumer;
import com.google.pubsub.v1.PubsubMessage;

public class PubSubMessage implements Message {

    private final PubsubMessage message;

    private final AckReplyConsumer ackReplyConsumer;

    public PubSubMessage(final PubsubMessage message) {
        this.message = Objects.requireNonNull(message, msg.isRequired("message"));
        this.ackReplyConsumer = null;
    }

    public PubSubMessage(final PubsubMessage message, final AckReplyConsumer ackReplyConsumer) {
        this.message = Objects.requireNonNull(message, msg.isRequired("message"));
        this.ackReplyConsumer = Objects.requireNonNull(ackReplyConsumer, msg.isRequired("ackReplyConsumer"));
    }

    public PubsubMessage getMessage() {
        return message;
    }

    @Override
    public String getPayload() {
        return message.getData().toStringUtf8();
    }

    @Override
    public CompletionStage ack(Metadata metadata) {
        if (ackReplyConsumer != null) {
            ackReplyConsumer.ack();
        }
        return CompletableFuture.completedFuture(null);
    }

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

    @Override
    public boolean equals(final Object o) {
        if (this == o) {
            return true;
        }
        if (o == null || getClass() != o.getClass()) {
            return false;
        }
        final PubSubMessage that = (PubSubMessage) o;
        return Objects.equals(message, that.message);
    }

    @Override
    public int hashCode() {
        return Objects.hash(message);
    }

    @Override
    public String toString() {
        return "PubSubMessage[" +
                "message=" + message +
                ']';
    }
}