TypeDictionary.java

/**
 * Copyright 2016-2017 Sixt GmbH & Co. Autovermietung KG
 * Licensed under the Apache License, Version 2.0 (the "License"); you may
 * not use this file except in compliance with the License. You may obtain a
 * copy of the License at http://www.apache.org/licenses/LICENSE-2.0
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations
 * under the License.
 */

package com.sixt.service.framework.kafka.messaging;

import com.google.protobuf.Parser;

import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

public class TypeDictionary {

    // synchronized because put may be executed in different thread than read access
    // if synchronization is found too heavy for this, extract interface and implement an immutable dictionary and another modifiable one
    private final Map<MessageType, Parser<com.google.protobuf.Message>> parsers = Collections.synchronizedMap(new HashMap<>());
    private final Map<MessageType, MessageHandler<? extends com.google.protobuf.Message>> handlers = Collections.synchronizedMap(new HashMap<>());


    public TypeDictionary() {

    }

    public TypeDictionary(Map<MessageType, MessageHandler<? extends com.google.protobuf.Message>> handlers, Map<MessageType, Parser<com.google.protobuf.Message>> parsers) {
        this.handlers.putAll(handlers);
        this.parsers.putAll(parsers);
    }

    /**
     * @param type
     * @return null if no MessageHandler was found for the type, otherwise the handler
     */
    public MessageHandler<? extends com.google.protobuf.Message> messageHandlerFor(MessageType type) {
        return handlers.get(type);
    }

    /**
     * @param type
     * @return null if no Parser was found for the type, otherwise the parser
     */
    public Parser<com.google.protobuf.Message> parserFor(MessageType type) {
        return parsers.get(type);
    }


    public MessageHandler<? extends com.google.protobuf.Message> putHandler(MessageType type, MessageHandler<? extends com.google.protobuf.Message> handler) {
        return handlers.put(type, handler);
    }

    public Parser putParser(MessageType type, Parser<com.google.protobuf.Message> parser) {
        return parsers.put(type, parser);
    }

    public void putAllParsers(Map<MessageType, Parser<com.google.protobuf.Message>> parsers) {
        this.parsers.putAll(parsers);
    }

    public void putAllHandlers(Map<MessageType, MessageHandler<? extends com.google.protobuf.Message>> handlers) {
        this.handlers.putAll(handlers);
    }

}