org.voovan.tools.MultiMap Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of voovan-framework Show documentation
Voovan is a java framwork and it not depends on any third-party framework.
There is a newer version: 4.3.8
package org.voovan.tools;

/**
 * 多值 Map
 *
 * @author helyho
 *         
 * Voovan Framework.
 * WebSite: https://github.com/helyho/Voovan
 * Licence: Apache v2 License
 */

import org.voovan.tools.json.JSON;

import java.util.*;

public class MultiMap
        extends HashMap>
{
    public MultiMap() {}

    public MultiMap(Map> map)
    {
        super(map);
    }

    /**
     * 获取值的集合
     * @param key 键
     * @return 值的集合
     */
    public List getValues(K key)
    {
        List vals = (List)super.get(key);
        if ((vals == null) || (vals.isEmpty())) {
            return null;
        }
        return vals;
    }

    /**
     * 获取值
     * @param key 键
     * @param i 值的索引
     * @return 值
     */
    public V getValue(K key, int i)
    {
        List vals = getValues(key);
        if (vals == null) {
            return null;
        }
        if ((i == 0) && (vals.isEmpty())) {
            return null;
        }
        return (V)vals.get(i);
    }

    /**
     * 增加值
     * @param key   键
     * @param value  值
     * @return  值的集合
     */
    public List putValue(K key, V value)
    {
        if (value == null) {
            return (List)super.put(key, null);
        }
        List vals = new ArrayList();
        vals.add(value);
        return (List)super.put((K)key, vals);
    }

    /**
     * 一次插入多个键值
     * @param input 键/值(List)的 Map 对象
     */
    public void putAllValues(Map input)
    {
        for (Map.Entry entry : input.entrySet()) {
            putValue((K)entry.getKey(), entry.getValue());
        }
    }

    /**
     * 一次插入一个键值
     * @param key 键
     * @param values 值的集合
     * @return 值的集合
     */
    public List putValues(K key, List values)
    {
        return (List)super.put(key, values);
    }

    /**
     * 一次插入一个键值
     * @param key 键
     * @param values 值的数组
     * @return 值的集合
     */
    @SafeVarargs
    public final List putValues(K key, V... values)
    {
        List list = new ArrayList();
        list.addAll(Arrays.asList(values));
        return (List)super.put(key, list);
    }

    /**
     * 一次插入一个键值
     * @param key 键
     * @param value 值
     */
    public void add(K key, V value)
    {
        List lo = (List)get(key);
        if (lo == null) {
            lo = new ArrayList();
        }
        lo.add(value);
        super.put(key, lo);
    }

    /**
     * 增加键的一系列的值
     * @param key 键
     * @param values 值的集合
     */
    public void addValues(K key, List values)
    {
        List lo = (List)get(key);
        if (lo == null) {
            lo = new ArrayList();
        }
        lo.addAll(values);
        super.put(key, lo);
    }

    /**
     * 增加键的一系列的值
     * @param key 键
     * @param values 值的集合
     */
    public void addValues(K key, V[] values)
    {
        List lo = (List)get(key);
        if (lo == null) {
            lo = new ArrayList();
        }
        lo.addAll(Arrays.asList(values));
        super.put(key, lo);
    }

    /**
     * 增加键/值
     * @param map 键/值(List)的 Map 对象
     * @return 是否成功 true:成功,false:失败
     */
    public boolean addAllValues(MultiMap map)
    {
        boolean merged = false;
        if ((map == null) || (map.isEmpty())) {
            return merged;
        }
        for (Map.Entry> entry : map.entrySet())
        {
            K name = (K)entry.getKey();
            List values = (List)entry.getValue();
            if (containsKey(name)) {
                merged = true;
            }
            addValues(name, values);
        }
        return merged;
    }

    /**
     * 移除某个值
     * @param key  键
     * @param value 值
     * @return  成功:true , 失败:false
     */
    public boolean removeValue(K key, V value)
    {
        List lo = (List)get(key);
        if ((lo == null) || (lo.isEmpty())) {
            return false;
        }
        boolean ret = lo.remove(value);
        if (lo.isEmpty()) {
            remove(key);
        } else {
            super.put(key, lo);
        }
        return ret;
    }

    /**
     * 是否包含某个值
     * @param value 值
     * @return 成功:true , 失败:false
     */
    public boolean containsValues(V value)
    {
        for (List vals : values()) {
            if ((vals.size() == 1) && (vals.contains(value))) {
                return true;
            }
        }
        return false;
    }

    public String toString()
    {
        return JSON.toJSON(this);
    }

}