im.shs.tick.auto.common.MultiSetMap Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tick-auto Show documentation

tick代码自动生成

The newest version!

package im.shs.tick.auto.common;

import java.util.*;

/**
 * MultiSetMap
 *
 * @author SimpleJuly
 */
public class MultiSetMap {
	private transient final Map> map;
	
	public MultiSetMap() {
		map = new HashMap<>();
	}
	
	private Set createSet() {
		return new HashSet<>();
	}
	
	/**
	 * put to MultiSetMap
	 * @param key 键
	 * @param value 值
	 * @return boolean
	 */
	public boolean put(K key, V value) {
		Set set = map.get(key);
		if (set == null) {
			set = createSet();
			if (set.add(value)) {
				map.put(key, set);
				return true;
			} else {
				throw new AssertionError("New set violated the set spec");
			}
		} else if (set.add(value)) {
			return true;
		} else {
			return false;
		}
	}

	/**
	 * 是否包含某个key
	 * @param key key
	 * @return 结果
	 */
	public boolean containsKey(K key) {
		return map.containsKey(key);
	}

	/**
	 * 是否包含 value 中的某个值
	 * @param value value
	 * @return 是否包含
	 */
	public boolean containsVal(V value) {
		Collection> values = map.values();
		return values.stream().anyMatch(vs -> vs.contains(value));
	}

	/**
	 * key 集合
	 * @return keys
	 */
	public Set keySet() {
		return map.keySet();
	}

	/**
	 * put list to MultiSetMap
	 * @param key 键
	 * @param set 值列表
	 * @return boolean
	 */
	public boolean putAll(K key, Set set) {
		if (set == null) {
			return false;
		} else {
			map.put(key, set);
			return true;
		}
	}
	
	/**
	 * get List by key
	 * @param key 键
	 * @return List
	 */
	public Set get(K key) {
		return map.get(key);
	}
	
	/**
	 * clear MultiSetMap
	 */
	public void clear() {
		map.clear();
	}

	/**
	 * isEmpty
	 * @return isEmpty
	 */
	public boolean isEmpty() {
		return map.isEmpty();
	}
}