package com.qianwen.core.websocket.holder;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import com.qianwen.core.websocket.handler.JsonMessageHandler;

public final class JsonMessageHandlerHolder {
    private static final Map<String, JsonMessageHandler> MESSAGE_HANDLER_MAP = new ConcurrentHashMap<>();

    private JsonMessageHandlerHolder() {
    }

    public static JsonMessageHandler getHandler(String type) {
        return MESSAGE_HANDLER_MAP.get(type);
    }

    public static void addHandler(JsonMessageHandler jsonMessageHandler) {
        MESSAGE_HANDLER_MAP.put(jsonMessageHandler.type(), jsonMessageHandler);
    }
}