pl.allegro.tech.hermes.consumers.consumer.sender.MessageSenderFactory Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of hermes-consumers Show documentation

Fast and reliable message broker built on top of Kafka.

There is a newer version: 2.10.0

package pl.allegro.tech.hermes.consumers.consumer.sender;

import pl.allegro.tech.hermes.api.EndpointAddress;
import pl.allegro.tech.hermes.api.Subscription;
import pl.allegro.tech.hermes.common.exception.EndpointProtocolNotSupportedException;
import pl.allegro.tech.hermes.common.exception.InternalProcessingException;
import pl.allegro.tech.hermes.consumers.consumer.ResilientMessageSender;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class MessageSenderFactory {

    private final Map protocolProviders = new HashMap<>();

    public MessageSenderFactory(List providers) {
        for (ProtocolMessageSenderProvider provider : providers) {
            for (String protocol : provider.getSupportedProtocols()) {
                addSupportedProtocol(protocol, provider);
            }
        }
    }

    public MessageSender create(Subscription subscription, ResilientMessageSender resilientMessageSender) {
        EndpointAddress endpoint = subscription.getEndpoint();

        ProtocolMessageSenderProvider provider = protocolProviders.get(endpoint.getProtocol());
        if (provider == null) {
            throw new EndpointProtocolNotSupportedException(endpoint);
        }
        return provider.create(subscription, resilientMessageSender);
    }

    private void addSupportedProtocol(String protocol, ProtocolMessageSenderProvider provider) {
        if (!protocolProviders.containsKey(protocol)) {
            startProvider(provider);
            protocolProviders.put(protocol, provider);
        }
    }

    private void startProvider(ProtocolMessageSenderProvider provider) {
        try {
            provider.start();
        } catch (Exception e) {
            throw new InternalProcessingException("Something went wrong while starting message sender provider", e);
        }
    }

    public void closeProviders() {
        protocolProviders.values().forEach(provider -> {
            try {
                provider.stop();
            } catch (Exception e) {
                throw new InternalProcessingException("Something went wrong while stopping message sender provider", e);
            }
        });
    }
}