org.reactivecommons.async.rabbit.config.ConnectionManager Maven / Gradle / Ivy

Go to download
package org.reactivecommons.async.rabbit.config;

import lombok.Builder;
import lombok.Getter;
import lombok.Setter;
import org.reactivecommons.async.commons.DiscardNotifier;
import org.reactivecommons.async.rabbit.HandlerResolver;
import org.reactivecommons.async.rabbit.communications.ReactiveMessageListener;
import org.reactivecommons.async.rabbit.communications.ReactiveMessageSender;

import java.util.Map;
import java.util.TreeMap;
import java.util.function.BiConsumer;

public class ConnectionManager {
    private final Map connections = new TreeMap<>();

    @Builder
    @Getter
    public static class DomainConnections {
        private final ReactiveMessageListener listener;
        private final ReactiveMessageSender sender;
        private final HandlerResolver handlerResolver;
        private final ConnectionFactoryProvider provider;
        @Setter
        private DiscardNotifier discardNotifier;
    }

    public void forSender(BiConsumer consumer) {
        connections.forEach((key, conn) -> consumer.accept(key, conn.getSender()));
    }

    public void forListener(BiConsumer consumer) {
        connections.forEach((key, conn) -> consumer.accept(key, conn.getListener()));
    }

    public void setDiscardNotifier(String domain, DiscardNotifier discardNotifier) {
        connections.get(domain).setDiscardNotifier(discardNotifier);
    }

    public ConnectionManager addDomain(String domain, ReactiveMessageListener listener, ReactiveMessageSender sender,
                                       HandlerResolver resolver) {
        connections.put(domain, DomainConnections.builder()
                .listener(listener)
                .sender(sender)
                .handlerResolver(resolver)
                .build());
        return this;
    }

    public ReactiveMessageSender getSender(String domain) {
        return connections.get(domain).getSender();
    }

    public ReactiveMessageListener getListener(String domain) {
        return connections.get(domain).getListener();
    }

    public DiscardNotifier getDiscardNotifier(String domain) {
        return connections.get(domain).getDiscardNotifier();
    }

    public HandlerResolver getHandlerResolver(String domain) {
        return connections.get(domain).getHandlerResolver();
    }
    public ConnectionFactoryProvider getProvider(String domain) {
        return connections.get(domain).getProvider();
    }
}