com.dottydingo.hyperion.client.MultiMap Maven / Gradle / Ivy

Go to download
package com.dottydingo.hyperion.client;

import java.util.*;

/**
 */
public class MultiMap
{
    private Map> map = new TreeMap>(new CaseInsensitiveComparator());

    public List get(String key)
    {
        if(key == null)
            throw new IllegalArgumentException("Null keys not allowed.");

        return map.get(key);
    }

    public String getFirst(String key)
    {
        if(key == null)
            throw new IllegalArgumentException("Null keys not allowed.");

        List vals = map.get(key);
        if(vals == null || vals.size() == 0)
            return null;

        return vals.get(0);
    }

    public void set(String key, String value)
    {
        if(key == null)
            throw new IllegalArgumentException("Null keys not allowed.");

        if(value == null)
            throw new IllegalArgumentException("Null values not allowed.");

        ArrayList vals = new ArrayList();
        vals.add(value);
        map.put(key,vals);

    }
    public void add(String key, String value)
    {
        if(key == null)
            throw new IllegalArgumentException("Null keys not allowed.");

        if(value == null)
            throw new IllegalArgumentException("Null values not allowed.");

        List vals = map.get(key);
        if(vals == null)
        {
            vals = new ArrayList();
            map.put(key,vals);
        }

        vals.add(value);
    }

    public Set getKeys()
    {
        return map.keySet();
    }

    public Set>> entries()
    {
        return map.entrySet();
    }

    public MultiMap merge(MultiMap multiMap)
    {
        MultiMap result = new MultiMap();
        result.map.putAll(this.map);
        result.map.putAll(multiMap.map);
        return result;
    }

    public boolean isEmpty()
    {
        return map.isEmpty();
    }

    private class CaseInsensitiveComparator implements Comparator
    {
        @Override
        public int compare(String o1, String o2)
        {
            return o1.toLowerCase().compareTo(o2.toLowerCase());
        }
    }
}