All Downloads are FREE. Search and download functionalities are using the official Maven repository.

net.lulihu.ObjectKit.CollectionKit Maven / Gradle / Ivy

package net.lulihu.ObjectKit;


import net.lulihu.Assert;
import net.lulihu.Assert0;
import net.lulihu.exception.ToolBoxException;
import net.lulihu.functional.CollectionConsumption;
import net.lulihu.functional.ConsumerResult;

import java.lang.reflect.Array;
import java.util.*;
import java.util.Map.Entry;

/**
 * 集合相关工具类,包括数组
 */
public class CollectionKit {

    private CollectionKit() {
        // 静态类不可实例化
    }

    /**
     * 数组取字段封装为另一个集合
     *
     * @param list     集合对象
     * @param consumer 表达式执行者
     * @param       泛型
     * @return 返回拼接结果
     */
    public static  List listToList(E[] list, ConsumerResult consumer) {
        return listToList(Arrays.asList(list), consumer);
    }

    /**
     * 集合取字段封装为另一个集合
     *
     * @param list     集合对象
     * @param consumer 表达式执行者
     * @param       泛型
     * @return 返回拼接结果
     */
    public static  List listToList(Iterable list, ConsumerResult consumer) {
        Assert0.toolBox().notNull(consumer, "表达式不可以为空");

        List result = newArrayList();
        if (null == list) return result;

        for (E item : list) {
            result.add(consumer.accept(item));
        }
        return result;
    }

    /**
     * 以 conjunction 为分隔符将集合转换为字符串
     *
     * @param          被处理的集合
     * @param collection  集合
     * @param conjunction 分隔符
     * @return 连接后的字符串
     */
    public static  String join(Iterable collection, String conjunction) {
        return join(conjunction, collection, Object::toString);
    }

    /**
     * 以 conjunction 为分隔符将集合转换为字符串
     *
     * @param          被处理的集合
     * @param conjunction 分隔符
     * @param collection  集合
     * @param consumption 取值表达式
     * @return 连接后的字符串
     */
    public static  String join(String conjunction, Iterable collection, ConsumerResult consumption) {
        if (collection == null) return "";

        StringBuilder sb = new StringBuilder();
        boolean isFirst = true;
        for (T item : collection) {
            if (isFirst) {
                isFirst = false;
            } else {
                sb.append(conjunction);
            }
            sb.append(consumption.accept(item));
        }
        return sb.toString();
    }

    /**
     * 以 conjunction 为分隔符将数组转换为字符串
     *
     * @param          被处理的集合
     * @param array       数组
     * @param conjunction 分隔符
     * @return 连接后的字符串
     */
    public static  String join(String conjunction, T... array) {
        if (isEmpty(array)) return "";

        StringBuilder sb = new StringBuilder();
        for (T item : array) {
            sb.append(conjunction).append(item);
        }
        return sb.delete(0, conjunction.length()).toString();
    }

    /**
     * 将Set排序(根据Entry的值)
     *
     * @param set 被排序的Set
     * @return 排序后的Set
     */
    public static List> sortEntrySetToList(Set> set) {
        List> list = new LinkedList<>(set);
        list.sort((o1, o2) -> {
            if (o1.getValue() > o2.getValue()) {
                return 1;
            }
            if (o1.getValue() < o2.getValue()) {
                return -1;
            }
            return 0;
        });
        return list;
    }

    /**
     * 切取部分数据
     *
     * @param       集合元素类型
     * @param stack    原数据
     * @param partSize 每部分数据的长度
     * @return 切取出的数据或null
     */
    public static  List popPart(Stack stack, int partSize) {
        if (isEmpty(stack)) return null;

        List current = new ArrayList<>();
        int size = stack.size() < partSize ? stack.size() : partSize;
        // 切割
        for (int i = 0; i < size; i++) {
            current.add(stack.pop());
        }
        return current;
    }

    /**
     * 切取部分数据
     *
     * @param       集合元素类型
     * @param surplus  原数据
     * @param partSize 每部分数据的长度
     * @return 切取出的数据或null
     */
    public static  List popPart(Deque surplus, int partSize) {
        if (isEmpty(surplus)) return null;

        final List current = new ArrayList<>();
        int size = surplus.size() >= partSize ? partSize : surplus.size();
        // 切割
        for (int i = 0; i < size; i++) {
            current.add(surplus.pop());
        }
        return current;
    }


    /**
     * 新建一个HashSet
     *
     * @return HashSet对象
     */
    public static  HashSet newHashSet() {
        return new HashSet<>();
    }

    /**
     * 新建一个HashSet
     *
     * @return HashSet对象
     */
    @SafeVarargs
    public static  HashSet newHashSet(T... ts) {
        return new HashSet<>(Arrays.asList(ts));
    }

    /**
     * 新建一个ArrayList
     *
     * @return ArrayList对象
     */
    public static  ArrayList newArrayList() {
        return new ArrayList<>();
    }

    /**
     * 新建一个ArrayList
     *
     * @return ArrayList对象
     */
    @SafeVarargs
    public static  ArrayList newArrayList(T... values) {
        return new ArrayList<>(Arrays.asList(values));
    }

    /**
     * 将新元素添加到已有数组中
* 添加新元素会生成一个新的数组,不影响原数组 * * @param buffer 已有数组 * @param newElement 新元素 * @return 新数组 */ public static T[] append(T[] buffer, T newElement) { T[] t = resize(buffer, buffer.length + 1, newElement.getClass()); t[buffer.length] = newElement; return t; } /** * 生成一个新的重新设置大小的数组 * * @param buffer 原数组 * @param newSize 新的数组大小 * @param componentType 数组元素类型 * @return 调整后的新数组 */ public static T[] resize(T[] buffer, int newSize, Class componentType) { T[] newArray = newArray(componentType, newSize); System.arraycopy(buffer, 0, newArray, 0, buffer.length >= newSize ? newSize : buffer.length); return newArray; } /** * 新建一个空数组 * * @param componentType 元素类型 * @param newSize 大小 * @return 空数组 */ @SuppressWarnings("unchecked") public static T[] newArray(Class componentType, int newSize) { return (T[]) Array.newInstance(componentType, newSize); } /** * 生成一个新的重新设置大小的数组
* 新数组的类型为原数组的类型 * * @param buffer 原数组 * @param newSize 新的数组大小 * @return 调整后的新数组 */ public static T[] resize(T[] buffer, int newSize) { return resize(buffer, newSize, buffer.getClass().getComponentType()); } /** * 将多个数组合并在一起
* 忽略null的数组 * * @param arrays 数组集合 * @return 合并后的数组 */ @SafeVarargs public static T[] addAll(T[]... arrays) { if (isEmpty(arrays)) return null; if (arrays.length == 1) return arrays[0]; int length = 0; for (T[] array : arrays) { if (array == null) { continue; } length += array.length; } T[] result = newArray(arrays.getClass().getComponentType().getComponentType(), length); length = 0; for (T[] array : arrays) { if (array == null) continue; System.arraycopy(array, 0, result, length, array.length); length += array.length; } return result; } /** * 克隆数组 * * @param array 被克隆的数组 * @return 新数组 */ public static List clone(List array) { if (array == null) return newArrayList(); return new ArrayList<>(array); } /** * 克隆数组 * * @param array 被克隆的数组 * @return 新数组 */ public static T[] clone(T[] array) { return array == null ? null : array.clone(); } /** * 生成一个数字列表
* 自动判定正序反序 * * @param excludedEnd 结束的数字(不包含) * @return 数字列表 */ public static int[] range(int excludedEnd) { return range(0, excludedEnd, 1); } /** * 生成一个数字列表
* 自动判定正序反序 * * @param includedStart 开始的数字(包含) * @param excludedEnd 结束的数字(不包含) * @return 数字列表 */ public static int[] range(int includedStart, int excludedEnd) { return range(includedStart, excludedEnd, 1); } /** * 生成一个数字列表
* 自动判定正序反序 * * @param includedStart 开始的数字(包含) * @param excludedEnd 结束的数字(不包含) * @param step 步进 * @return 数字列表 */ public static int[] range(int includedStart, int excludedEnd, int step) { if (includedStart > excludedEnd) { int tmp = includedStart; includedStart = excludedEnd; excludedEnd = tmp; } if (step <= 0) { step = 1; } int deviation = excludedEnd - includedStart; int length = deviation / step; if (deviation % step != 0) { length += 1; } int[] range = new int[length]; for (int i = 0; i < length; i++) { range[i] = includedStart; includedStart += step; } return range; } /** * 截取数组的部分 * * @param list 被截取的数组 * @param start 开始位置(包含) * @param end 结束位置(不包含) * @return 截取后的数组,当开始位置超过最大时,返回null */ public static List sub(List list, int start, int end) { if (isEmpty(list)) return null; if (start < 0) { start = 0; } if (end < 0) { end = 0; } if (start > end) { int tmp = start; start = end; end = tmp; } final int size = list.size(); if (end > size) { if (start >= size) { return null; } end = size; } if (start == 0 && end == size) return list; return list.subList(start, end); } /** * 截取集合的部分 * * @param list 被截取的数组 * @param start 开始位置(包含) * @param end 结束位置(不包含) * @return 截取后的数组,当开始位置超过最大时,返回null */ public static List sub(Collection list, int start, int end) { if (list == null || list.isEmpty()) { return null; } return sub(new ArrayList(list), start, end); } /** * 数组是否为空 * * @param array 数组 * @return 是否为空 */ public static boolean isEmpty(T[] array) { return array == null || array.length == 0; } /** * 数组是否为非空 * * @param array 数组 * @return 是否为非空 */ public static boolean isNotEmpty(T[] array) { return !isEmpty(array); } /** * 集合是否为空 * * @param collection 集合 * @return 是否为空 */ public static boolean isEmpty(Collection collection) { return collection == null || collection.isEmpty(); } /** * 集合是否为空 * * @param iterable 集合 * @return 是否为空 */ public static boolean isEmpty(Iterable iterable) { return iterable == null || !iterable.iterator().hasNext(); } /** * 如果集合为空则执行表达式返回结果,如果集合不为空,则直接返回 * * @param collection 集合 * @return 是否为空 */ public static Collection isEmpty(Collection collection, CollectionConsumption accept) { if (isNotEmpty(collection)) return collection; return accept.accept(); } /** * 集合是否为非空 * * @param collection 集合 * @return 是否为非空 */ public static boolean isNotEmpty(Collection collection) { return !isEmpty(collection); } /** * 映射键值(参考Python的zip()函数)
* 例如:
* keys = [a,b,c,d]
* values = [1,2,3,4]
* 则得到的Map是 {a=1, b=2, c=3, d=4}
* 如果两个数组长度不同,则只对应最短部分 * * @param keys 键列表 * @param values 值列表 * @return Map */ public static Map zip(T[] keys, K[] values) { if (isEmpty(keys) || isEmpty(values)) return null; int size = Math.min(keys.length, values.length); Map map = new HashMap<>(); for (int i = 0; i < size; i++) { map.put(keys[i], values[i]); } return map; } /** * 映射键值(参考Python的zip()函数)
* 例如:
* keys = a,b,c,d
* values = 1,2,3,4
* delimiter = , * 则得到的Map是 {a=1, b=2, c=3, d=4}
* 如果两个数组长度不同,则只对应最短部分 * * @param keys 键列表 * @param values 值列表 * @return Map */ public static Map zip(String keys, String values, String delimiter) { return zip(StrKit.split(keys, delimiter), StrKit.split(values, delimiter)); } /** * 映射键值(参考Python的zip()函数)
* 例如:
* keys = [a,b,c,d]
* values = [1,2,3,4]
* 则得到的Map是 {a=1, b=2, c=3, d=4}
* 如果两个数组长度不同,则只对应最短部分 * * @param keys 键列表 * @param values 值列表 * @return Map */ public static Map zip(Collection keys, Collection values) { if (isEmpty(keys) || isEmpty(values)) { return null; } final List keyList = new ArrayList<>(keys); final List valueList = new ArrayList<>(values); final int size = Math.min(keys.size(), values.size()); final Map map = new HashMap<>(); for (int i = 0; i < size; i++) { map.put(keyList.get(i), valueList.get(i)); } return map; } /** * 数组中是否包含元素 * * @param array 数组 * @param value 被检查的元素 * @return 是否包含 */ public static boolean contains(T[] array, T value) { final Class componentType = array.getClass().getComponentType(); boolean isPrimitive = false; if (null != componentType) isPrimitive = componentType.isPrimitive(); for (T t : array) { if (t == value) { return true; } else if (!isPrimitive && null != value && value.equals(t)) { return true; } } return false; } /** * 将集合转换为排序后的TreeSet * * @param collection 集合 * @param comparator 比较器 * @return treeSet */ public static TreeSet toTreeSet(Collection collection, Comparator comparator) { final TreeSet treeSet = new TreeSet<>(comparator); treeSet.addAll(collection); return treeSet; } /** * 排序集合 * * @param collection 集合 * @param comparator 比较器 * @return treeSet */ public static List sort(Collection collection, Comparator comparator) { List list = new ArrayList<>(collection); list.sort(comparator); return list; } //------------- 基本类型的数组转换为包装类型数组 /** * 数组转集合 * * @param values 数组 * @return 同样的集合 */ public static List wrap(T... values) { if (isEmpty(values)) return newArrayList(); return newArrayList(values); } /** * 集合转数组 * * @param values 数组 * @return 同样的数组 */ public static Object[] wrapList(final List values) { if (isEmpty(values)) return new Object[0]; int size = values.size(); Object[] result = new Object[size]; for (int i = 0; i < size; i++) { result[i] = values.get(i); } return result; } /** * 集合转数组 * * @param values 数组 * @return 同样的集合 */ public static String[] wrapStr(List values) { if (isEmpty(values)) return null; final int length = values.size(); String[] array = new String[length]; for (int i = 0; i < length; i++) { array[i] = values.get(i); } return array; } /** * 集合转数组 * * @param values 数组 * @return 同样的集合 */ public static Integer[] wrap(List values) { final int length = values.size(); Integer[] array = new Integer[length]; for (int i = 0; i < length; i++) { array[i] = values.get(i); } return array; } /** * 将基本类型数组包装为包装类型 * * @param values 基本类型数组 * @return 包装类型数组 */ public static Integer[] wrap(int... values) { final int length = values.length; Integer[] array = new Integer[length]; for (int i = 0; i < length; i++) { array[i] = values[i]; } return array; } /** * 将基本类型数组包装为包装类型 * * @param values 基本类型数组 * @return 包装类型数组 */ public static Long[] wrap(long... values) { final int length = values.length; Long[] array = new Long[length]; for (int i = 0; i < length; i++) { array[i] = values[i]; } return array; } /** * 将基本类型数组包装为包装类型 * * @param values 基本类型数组 * @return 包装类型数组 */ public static Character[] wrap(char... values) { final int length = values.length; Character[] array = new Character[length]; for (int i = 0; i < length; i++) { array[i] = values[i]; } return array; } /** * 将基本类型数组包装为包装类型 * * @param values 基本类型数组 * @return 包装类型数组 */ public static Byte[] wrap(byte... values) { final int length = values.length; Byte[] array = new Byte[length]; for (int i = 0; i < length; i++) { array[i] = values[i]; } return array; } /** * 将基本类型数组包装为包装类型 * * @param values 基本类型数组 * @return 包装类型数组 */ public static Short[] wrap(short... values) { final int length = values.length; Short[] array = new Short[length]; for (int i = 0; i < length; i++) { array[i] = values[i]; } return array; } /** * 将基本类型数组包装为包装类型 * * @param values 基本类型数组 * @return 包装类型数组 */ public static Float[] wrap(float... values) { final int length = values.length; Float[] array = new Float[length]; for (int i = 0; i < length; i++) { array[i] = values[i]; } return array; } /** * 将基本类型数组包装为包装类型 * * @param values 基本类型数组 * @return 包装类型数组 */ public static Double[] wrap(double... values) { final int length = values.length; Double[] array = new Double[length]; for (int i = 0; i < length; i++) { array[i] = values[i]; } return array; } /** * 将基本类型数组包装为包装类型 * * @param values 基本类型数组 * @return 包装类型数组 */ public static Boolean[] wrap(boolean... values) { final int length = values.length; Boolean[] array = new Boolean[length]; for (int i = 0; i < length; i++) { array[i] = values[i]; } return array; } /** * 判定给定对象是否为数组类型 * * @param obj 对象 * @return 是否为数组类型 */ public static boolean isArray(Object obj) { return obj.getClass().isArray(); } /** * 数组或集合转String * * @param obj 集合或数组对象 * @return 数组字符串,与集合转字符串格式相同 */ public static String toString(Object obj) { if (null == obj) { return null; } if (isArray(obj)) try { return Arrays.deepToString((Object[]) obj); } catch (Exception e) { final String className = obj.getClass().getComponentType().getName(); switch (className) { case "long": return Arrays.toString((long[]) obj); case "int": return Arrays.toString((int[]) obj); case "short": return Arrays.toString((short[]) obj); case "char": return Arrays.toString((char[]) obj); case "byte": return Arrays.toString((byte[]) obj); case "boolean": return Arrays.toString((boolean[]) obj); case "float": return Arrays.toString((float[]) obj); case "double": return Arrays.toString((double[]) obj); default: throw new ToolBoxException(e); } } return obj.toString(); } }