java-8 - 返回泛型类型，它是基于传递给参数的类型的特定集合

Question

我正在尝试编写一个默认情况下返回一个集合的泛型类，但如果调用函数需要特定类型，例如 List 或 Set，则该方法应该能够产生它。这在java中可能吗？如果我做得不对，请纠正我。

public <U extends Collection> U saveOrUpdate(Iterable<T> entities, Class<U> klass) {
    Iterable<T> savedEntities = this.repository.save(entities);
    Type type = klass.getClass().getGenericSuperclass();
    ParameterizedType pt = (ParameterizedType) type;
    if (pt.getTypeName().equalsIgnoreCase(Set.class.getName())) {
        return StreamSupport.stream(savedEntities.spliterator(), false)
                .collect(Collectors.toSet());
    } else if (pt.getTypeName().equalsIgnoreCase(List.class.getName())) {
        return StreamSupport.stream(savedEntities.spliterator(), false)
                .collect(Collectors.toList());
    } else {
        return StreamSupport.stream(savedEntities.spliterator(), false)
                .collect(Collectors.toCollection());
    }
}

Answer 1

我认为您的方法没有类型安全的解决方案 - 您可以将返回的集合转换为 U ， 我猜...

而不是通过 Class , 如果你传入 Supplier 不是更容易吗？为 Collectors#toCollection ?

public <U extends Collection<T>> U saveOrUpdate(Iterable<T> entities, Supplier<U> supplier) {
    Iterable<T> savedEntities = this.repository.save(entities);
    return StreamSupport.stream(savedEntities.spliterator(), false)
                        .collect(Collectors.toCollection(supplier));
}

然后你就可以完全控制结果类型，例如:

saveOrUpdate(entities, HashSet::new);  // constructor as Supplier

它甚至会进行类型检查。