com.tvd12.ezymq.rabbitmq.endpoint.EzyRabbitTopicServer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ezymq-rabbitmq Show documentation
The newest version!
package com.tvd12.ezymq.rabbitmq.endpoint;

import com.rabbitmq.client.AMQP.BasicProperties;
import com.rabbitmq.client.*;
import com.tvd12.ezyfox.io.EzyStrings;
import com.tvd12.ezyfox.util.EzyReturner;
import com.tvd12.ezyfox.util.EzyStartable;
import com.tvd12.ezymq.rabbitmq.handler.EzyRabbitMessageHandler;
import lombok.Setter;

import java.io.IOException;

public class EzyRabbitTopicServer
    extends EzyRabbitEndpoint
    implements EzyStartable {

    protected final String queueName;
    protected final Consumer consumer;
    @Setter
    protected EzyRabbitMessageHandler messageHandler;

    public EzyRabbitTopicServer(
        Channel channel,
        String exchange,
        String queueName
    ) throws IOException {
        super(channel, exchange);
        this.queueName = fetchQueueName(queueName);
        this.consumer = newConsumer();
    }

    public static Builder builder() {
        return new Builder();
    }

    protected String fetchQueueName(String queueName) throws IOException {
        if (EzyStrings.isNoContent(queueName)) {
            return channel.queueDeclare().getQueue();
        }
        return queueName;
    }

    @Override
    public void start() throws Exception {
        channel.basicConsume(queueName, true, consumer);
    }

    protected Consumer newConsumer() {
        return new DefaultConsumer(channel) {
            @Override
            public void handleDelivery(
                String consumerTag,
                Envelope envelope,
                BasicProperties properties,
                byte[] body
            ) {
                Delivery delivery = new Delivery(envelope, properties, body);
                messageHandler.handle(delivery);
            }
        };
    }

    public static class Builder extends EzyRabbitEndpoint.Builder {

        protected String queueName;

        public Builder queueName(String queueName) {
            this.queueName = queueName;
            return this;
        }

        @Override
        public EzyRabbitTopicServer build() {
            return EzyReturner.returnWithException(() ->
                new EzyRabbitTopicServer(
                    channel,
                    exchange,
                    queueName
                )
            );
        }
    }
}