如何实现将Stream收集回原泛型Collection类型的方法？

我想要编写一个简单方法，接收一个集合并截断其元素数量，代码实现如下：

public static <T extends Collection<?>> T limit(T collection, long limit){
    return collection.stream().limit(limit).collect(Collectors.toCollection(???));
}

请问是否可以将Stream收集回泛型T对应的Collection类型？

回答 #

可以实现，但核心要解决如何创建泛型T对应的集合实例的问题，具体分几种场景处理：

1. 利用反射自动创建实例（适用于有无参构造器的集合） #

如果泛型T对应的Collection实现（比如ArrayList、HashSet）提供了无参构造器，可以通过反射获取构造器来创建新实例，填充截断后的元素：

public static <T extends Collection<?>> T limit(T collection, long limit) {
    try {
        Supplier<T> collectionSupplier = () -> {
            try {
                return collection.getClass().getConstructor().newInstance();
            } catch (InstantiationException | IllegalAccessException | NoSuchMethodException | InvocationTargetException e) {
                throw new RuntimeException("无法创建目标集合实例", e);
            }
        };
        
        @SuppressWarnings("unchecked")
        T result = (T) collection.stream()
                .limit(limit)
                .collect(Collectors.toCollection(collectionSupplier));
        
        return result;
    } catch (RuntimeException e) {
        throw new IllegalArgumentException("当前集合类型不支持自动实例化", e);
    }
}

这种方式无需调用者额外传参，但缺点是如果集合没有无参构造器（比如Arrays.asList()返回的固定大小列表），会直接抛出异常。

2. 让调用者传入实例供应商（更灵活可靠） #

为兼容没有无参构造器的集合，最好让调用者传入Supplier<T>，明确指定如何创建目标类型的集合实例：

public static <T extends Collection<?>> T limit(T collection, long limit, Supplier<T> supplier) {
    @SuppressWarnings("unchecked")
    T result = (T) collection.stream()
            .limit(limit)
            .collect(Collectors.toCollection(supplier));
    
    return result;
}

调用示例：

// 截断ArrayList，传入ArrayList的构造方法引用
ArrayList<String> originalList = new ArrayList<>(List.of("a", "b", "c", "d"));
ArrayList<String> limitedList = limit(originalList, 2, ArrayList::new);

// 截断LinkedHashSet，传入对应构造器
LinkedHashSet<Integer> originalSet = new LinkedHashSet<>(Set.of(1,2,3,4,5));
LinkedHashSet<Integer> limitedSet = limit(originalSet, 3, LinkedHashSet::new);

这种方式完全避免了反射的限制，也更符合Java函数式编程风格。

3. 特殊集合的适配处理 #

对于不可变集合（比如Guava的ImmutableList），直接创建空实例再添加元素的方式不适用，需要结合对应集合的构建器处理：

public static <T extends ImmutableCollection<?>> T limit(T collection, long limit) {
    if (collection instanceof ImmutableList) {
        return (T) ((ImmutableList<?>) collection).stream()
                .limit(limit)
                .collect(ImmutableList.toImmutableList());
    }
    if (collection instanceof ImmutableSet) {
        return (T) ((ImmutableSet<?>) collection).stream()
                .limit(limit)
                .collect(ImmutableSet.toImmutableSet());
    }
    throw new IllegalArgumentException("不支持当前不可变集合类型");
}

关键注意点 #

• 由于Java泛型擦除，需要添加@SuppressWarnings("unchecked")抑制编译警告，只要supplier返回的确实是T类型实例，运行时就是安全的。
• 对于线程安全集合（比如CopyOnWriteArrayList），同样可以通过上述方式处理，只要确保构造器或供应商能正确创建实例。

内容的提问来源于stack exchange，提问作者Teddy Tsai