com.cookingfox.util.uid.UidKeyTranslator Maven / Gradle / Ivy

Go to download
package com.cookingfox.util.uid;

import java.util.LinkedHashMap;
import java.util.Map;

/**
 * A class that can be used to manage a map of {@link Uid} to user-defined objects and 'translate'
 * from one to the other format. This is useful when an application uses unique ids that require
 * different data types in certain contexts. Use the {@link #addToDictionary(Map)} method to add
 * specifications for this format to the translator's dictionary.
 *
 * @param  Indicates the data type of the item (key) that the unique id is mapped to.
 */
public class UidKeyTranslator {

    //----------------------------------------------------------------------------------------------
    // PROPERTIES
    //----------------------------------------------------------------------------------------------

    /**
     * A map of unique ids to a user-defined data type.
     */
    protected final Map dictionary = new LinkedHashMap<>();

    //----------------------------------------------------------------------------------------------
    // CONSTRUCTORS
    //----------------------------------------------------------------------------------------------

    /**
     * Creates a new instance.
     */
    public UidKeyTranslator() {
    }

    /**
     * Creates a new instance, calling {@link #addToDictionary(Map)} for the provided map.
     *
     * @param map The values to add to the dictionary.
     * @throws UidKeyTranslatorException when the map contains invalid keys or values.
     */
    public UidKeyTranslator(Map map) throws UidKeyTranslatorException {
        addToDictionary(map);
    }

    //----------------------------------------------------------------------------------------------
    // PUBLIC METHODS
    //----------------------------------------------------------------------------------------------

    /**
     * Adds a UID-key map to the dictionary.
     *
     * @param map The values to add to the dictionary.
     * @throws UidKeyTranslatorException when the map contains invalid keys or values.
     */
    @SuppressWarnings("SuspiciousMethodCalls")
    public void addToDictionary(Map map) throws UidKeyTranslatorException {
        for (Map.Entry entry : map.entrySet()) {
            final Object key = entry.getKey();

            if (!Uid.class.isInstance(key)) {
                String type = (key == null ? null : key.getClass().getSimpleName());

                throw new UidKeyTranslatorException(String.format("The map can only contain keys " +
                        "that are Uid instances (%s)", type));
            } else if (dictionary.containsKey(key)) {
                throw new UidKeyTranslatorException("The following key is already present in the " +
                        "dictionary: " + key);
            }
        }

        dictionary.putAll(map);
    }

    /**
     * Returns the key that is mapped to this unique id.
     *
     * @param uid The uid to translate.
     * @return The key that is mapped to the provided uid.
     * @throws UidKeyTranslatorException when the provided uid is not in the dictionary.
     */
    public T fromUid(Uid uid) throws UidKeyTranslatorException {
        T key = dictionary.get(uid);

        if (null == key) {
            throw new UidKeyTranslatorException("Provided Uid is not in the dictionary: " + uid);
        }

        return key;
    }

    /**
     * Accepts a map of unique ids to values and returns a new map which replaces the unique ids
     * with the keys from the dictionary.
     *
     * @param map Map of Uid -> value.
     * @return Map of Key -> value.
     * @throws UidKeyTranslatorException when a provided uid is not in the dictionary.
     */
    public Map fromUidMap(Map map) throws UidKeyTranslatorException {
        Map result = new LinkedHashMap<>();

        for (Map.Entry entry : map.entrySet()) {
            result.put(fromUid(entry.getKey()), entry.getValue());
        }

        return result;
    }

    /**
     * Returns the unique id that is mapped to this key.
     *
     * @param key The key to translate to Uid.
     * @return The uid that is mapped to the provided key.
     * @throws UidKeyTranslatorException when the provided key is not in the dictionary.
     */
    public Uid toUid(T key) throws UidKeyTranslatorException {
        for (Map.Entry entry : dictionary.entrySet()) {
            if (key.equals(entry.getValue())) {
                return entry.getKey();
            }
        }

        throw new UidKeyTranslatorException("Provided key is not in the dictionary: " + key);
    }

    /**
     * Accepts a map of keys to values and returns a new map which replaces the keys with the unique
     * ids from the dictionary.
     *
     * @param map Map of Key -> value.
     * @return Map of Uid -> value.
     * @throws UidKeyTranslatorException when the provided key is not in the dictionary.
     */
    public Map toUidMap(Map map) throws UidKeyTranslatorException {
        Map result = new LinkedHashMap<>();

        for (Map.Entry entry : map.entrySet()) {
            result.put(toUid(entry.getKey()), entry.getValue());
        }

        return result;
    }

    @Override
    public String toString() {
        return dictionary.toString();
    }

}