Java Stream收集到泛型集合：利用Supplier工厂模式

import java.util.Collection;
import java.util.stream.Collectors;

public class CollectionUtils {
    public static <T extends Collection<?>> T limit(T collection, long limit) {
        // 如何在这里创建一个T类型的实例？
        // return collection.stream().limit(limit).collect(Collectors.toCollection(???));
        return null; // 占位符
    }
}

import java.util.Collection;
import java.util.List;
import java.util.ArrayList;
import java.util.function.Supplier;
import java.util.stream.Collectors;

public class CollectionUtils {

    /**
     * 截取集合的前N个元素，并返回一个新的指定类型的集合。
     *
     * @param collection 原始集合
     * @param limit      要保留的元素数量
     * @param factory    用于创建新集合实例的工厂函数
     * @param <E>        集合中元素的类型
     * @param <T>        集合的类型，必须是Collection的子类
     * @return 包含截取后元素的新集合
     */
    public static <E, T extends Collection<E>> T limit(T collection, long limit, Supplier<T> factory) {
        return collection.stream()
                .limit(limit)
                .collect(Collectors.toCollection(factory));
    }

    public static void main(String[] args) {
        // 示例用法
        List<Integer> originalList = List.of(1, 2, 3, 4, 5);

        // 将List截取为包含前3个元素的ArrayList
        ArrayList<Integer> limitedArrayList = limit(originalList, 3, ArrayList::new);
        System.out.println("Limited ArrayList: " + limitedArrayList); // 输出: [1, 2, 3]

        // 如果原始集合是Set，可以截取为HashSet
        Collection<String> originalSet = new java.util.HashSet<>(List.of("A", "B", "C", "D"));
        java.util.HashSet<String> limitedHashSet = limit(originalSet, 2, java.util.HashSet::new);
        System.out.println("Limited HashSet: " + limitedHashSet); // 输出: [A, B] 或 [B, A] 等，取决于Set的内部顺序

        // 也可以直接返回原始类型，如果原始类型有无参构造函数
        // 注意：List.of() 返回的是不可变List，不能直接用ArrayList::new来限制
        // List<String> immutableList = List.of("X", "Y", "Z");
        // ArrayList<String> limitedFromImmutable = limit(immutableList, 1, ArrayList::new);
        // System.out.println("Limited from Immutable: " + limitedFromImmutable); // 输出: [X]
    }
}