com.tvd12.ezymq.rabbitmq.setting.EzyRabbitRpcConsumerSetting 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.setting;

import com.rabbitmq.client.Channel;
import com.tvd12.ezymq.rabbitmq.handler.EzyRabbitRequestHandler;
import com.tvd12.ezymq.rabbitmq.handler.EzyRabbitRequestHandlers;
import com.tvd12.ezymq.rabbitmq.handler.EzyRabbitRequestInterceptor;
import com.tvd12.ezymq.rabbitmq.handler.EzyRabbitRequestInterceptors;
import lombok.Getter;

import java.util.Collection;
import java.util.Map;

@Getter
public class EzyRabbitRpcConsumerSetting extends EzyRabbitEndpointSetting {

    protected final int threadPoolSize;
    protected final String replyRoutingKey;
    protected final String requestQueueName;
    protected final EzyRabbitRequestHandlers requestHandlers;
    protected final EzyRabbitRequestInterceptors requestInterceptors;

    public EzyRabbitRpcConsumerSetting(
        Channel channel,
        String exchange,
        int prefetchCount,
        String replyRoutingKey,
        String requestQueueName,
        int threadPoolSize,
        EzyRabbitRequestHandlers requestHandlers,
        EzyRabbitRequestInterceptors requestInterceptors
    ) {
        super(channel, exchange, prefetchCount);
        this.replyRoutingKey = replyRoutingKey;
        this.requestQueueName = requestQueueName;
        this.threadPoolSize = threadPoolSize;
        this.requestHandlers = requestHandlers;
        this.requestInterceptors = requestInterceptors;
    }

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

    public static class Builder extends EzyRabbitEndpointSetting.Builder {

        protected int threadPoolSize = 1;
        protected String replyRoutingKey = "";
        protected String requestQueueName = null;
        protected final EzyRabbitSettings.Builder parent;
        protected final EzyRabbitRequestHandlers requestHandlers;
        protected final EzyRabbitRequestInterceptors requestInterceptors;

        public Builder() {
            this(null);
        }

        public Builder(EzyRabbitSettings.Builder parent) {
            this.parent = parent;
            this.requestHandlers = new EzyRabbitRequestHandlers();
            this.requestInterceptors = new EzyRabbitRequestInterceptors();
        }

        public Builder threadPoolSize(int threadPoolSize) {
            this.threadPoolSize = threadPoolSize;
            return this;
        }

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

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

        public Builder addRequestInterceptor(
            EzyRabbitRequestInterceptor requestInterceptor
        ) {
            this.requestInterceptors.addInterceptor(requestInterceptor);
            return this;
        }

        public Builder addRequestInterceptors(
            Collection requestInterceptors
        ) {
            this.requestInterceptors.addInterceptors(requestInterceptors);
            return this;
        }

        @SuppressWarnings("rawtypes")
        public Builder addRequestHandler(String cmd, EzyRabbitRequestHandler handler) {
            requestHandlers.addHandler(cmd, handler);
            return this;
        }

        @SuppressWarnings("rawtypes")
        public Builder addRequestHandlers(
            Map handlers
        ) {
            for (String cmd : handlers.keySet()) {
                EzyRabbitRequestHandler handler = handlers.get(cmd);
                addRequestHandler(cmd, handler);
            }
            return this;
        }

        public EzyRabbitSettings.Builder parent() {
            return parent;
        }

        @Override
        public EzyRabbitRpcConsumerSetting build() {
            return new EzyRabbitRpcConsumerSetting(
                channel,
                exchange,
                prefetchCount,
                replyRoutingKey,
                requestQueueName,
                threadPoolSize,
                requestHandlers,
                requestInterceptors
            );
        }
    }
}