io.castled.pubsub.MessageSubscriber Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of commons Show documentation

Castled Commons Library

There is a newer version: 1.0.0

package io.castled.pubsub;

import com.google.common.collect.Maps;
import com.google.inject.Singleton;
import io.castled.pubsub.registry.Message;

import java.util.List;
import java.util.Map;
import java.util.concurrent.CopyOnWriteArrayList;

@Singleton
public class MessageSubscriber {
    private final Map> messageListeners = Maps.newConcurrentMap();

    public void subscribe(MessageType messageType, MessageListener messageListener) {
        this.messageListeners.computeIfAbsent(messageType, key -> new CopyOnWriteArrayList<>());
        this.messageListeners.get(messageType).add(messageListener);
    }

    public void consumeMessage(Message message) {
        for (MessageListener messageListener : messageListeners.get(message.getType())) {
            messageListener.handleMessage(message);
        }
    }
}