develop.toolkit.struct.Sorter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of develop-toolkit Show documentation

开发工具箱

There is a newer version: 1.0.6

package develop.toolkit.struct;

import lombok.NonNull;

import java.util.*;

/**
 * 分拣器
 *
 * @author qiushui on 2018-07-09.
 * @since 0.1
 */
public class Sorter {

	private CollectionInMap map = new CollectionInMap<>(key -> new LinkedList());

	private SortFunction sortFunction;

	public Sorter(@NonNull SortFunction sortFunction) {
		this.sortFunction = sortFunction;
	}

	/**
	 * 分拣
	 * @param collection
	 */
	public void sort(Collection collection) {
		for (V item : collection) {
			K key = sortFunction.sort(item);
			map.addItemSoft(key, item);
		}
	}

	/**
	 * 清空
	 */
	public void clear() {
		map.clear();
	}

	/**
	 * 返回单列列表
	 * @param key
	 * @return
	 */
	public List getSingleList(K key) {
		if (map.containsKey(key)) {
			return new ArrayList<>(map.get(key));
		} else {
			return Collections.EMPTY_LIST;
		}
	}

	/**
	 * 返回所有键
	 *
	 * @return
	 */
	public Set allKey() {
		return map.keySet();
	}

	@FunctionalInterface
	public interface SortFunction {

		/**
		 * 返回需要投递的key
		 *
		 * @param item
		 * @return
		 */
		K sort(V item);
	}
}