java - 如何将 Stream<List<Long>> flatMapToLong？

Question

我有这个方法:

public static long sumDigits(final List<Long> list) {
    return list
            .stream()
            .map(l -> toDigits(l))
            .flatMapToLong(x -> x.stream())
            .sum()
}

toDigits 有这个签名:

List<Long> toDigits(long l)

在 flatMapToLong 行给出了这个错误

Type mismatch: cannot convert from Stream< Long > to LongStream

当我把它改成

flatMapToLong(x -> x)

我收到这个错误

Type mismatch: cannot convert from List< Long > to LongStream

唯一有效的就是这个

public static long sumDigits(final List<Long> list) {
    return list
            .stream()
            .map(l -> toDigits(l))
            .flatMap(x -> x.stream())
            .reduce(0L, (accumulator, add) -> Math.addExact(accumulator, add));
}

Answer 1

您传递给 flatMapToLong 的 Function 需要返回一个 LongStream :

return list
        .stream()
        .map(l -> toDigits(l))
        .flatMapToLong(x -> x.stream().mapToLong(l -> l))
        .sum();

如果需要，您也可以拆分 flatMapToLong:

return list
        .stream()
        .map(ClassOfToDigits::toDigits)
        .flatMap(List::stream)
        .mapToLong(Long::longValue)
        .sum();