dev.soffa.foundation.pubsub.PubSubMessengerImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of foundation-starter-pubsub Show documentation

Soffa Foundation

There is a newer version: 0.17.31

Show newest version

package dev.soffa.foundation.pubsub;

import dev.soffa.foundation.annotation.Publish;
import dev.soffa.foundation.message.Message;
import dev.soffa.foundation.message.MessageHandler;
import dev.soffa.foundation.message.pubsub.PubSubClient;
import dev.soffa.foundation.message.pubsub.PubSubMessenger;
import org.checkerframework.checker.nullness.qual.NonNull;

import javax.validation.constraints.NotNull;
import java.util.Map;
import java.util.concurrent.CompletableFuture;

public class PubSubMessengerImpl implements PubSubMessenger {

    public static final String DEFAULT = "default";
    private final Map clients;
    private final PubSubClient defaultClient;

    public PubSubMessengerImpl(Map clients) {
        this.clients = clients;
        if (clients.containsKey(DEFAULT)) {
            defaultClient = clients.get(DEFAULT);
        } else {
            defaultClient = clients.values().iterator().next();
        }
    }


    @Override
    public PubSubClient getDefaultClient() {
        return defaultClient;
    }

    @Override
    public PubSubClient getClient(String name) {
        return clients.get(name);
    }

    @Override
    public void subscribe(@NonNull String subject, boolean broadcast, MessageHandler messageHandler) {
        getDefaultClient().subscribe(subject, broadcast, messageHandler);
    }

    @Override
    public void subscribe(MessageHandler messageHandler) {
        getDefaultClient().subscribe(messageHandler);
    }

    @Override
    public  CompletableFuture request(@NonNull String subject, @NotNull Message message, Class returnType) {
        return getDefaultClient().request(subject, message, returnType);
    }

    @Override
    public void publish(@NonNull String subject, @NotNull Message message) {
        if (Publish.BROADCAST_TARGET.equalsIgnoreCase(subject)) {
            getDefaultClient().broadcast(message);
        } else {
            getDefaultClient().publish(subject, message);
        }
    }

    @Override
    public void publish(@NotNull Message message) {
        getDefaultClient().publish(message);
    }

    @Override
    public void broadcast(@NonNull String target, @NotNull Message message) {
        getDefaultClient().broadcast(target, message);
    }

    @Override
    public void broadcast(Message message) {
        getDefaultClient().broadcast(message);
    }

    @Override
    public void addBroadcastChannel(String value) {
        getDefaultClient().addBroadcastChannel(value);
    }


}