Java 8:在Stream上执行reduce操作

Question

我有一个包含键值对的java.util.stream.Stream,如:

<1,3> <1,5> <3,1> <4,2> <4,7> <4,8>

现在我想合并所有具有相同密钥的条目:

<1,[3,5]>  <3,[1]> <4,[2,7,8]>

数据已经排序,因此只需要合并连续的数据集.

现在我正在寻找一种方法来像上面那样转换流的内容,而不是将所有数据集加载到内存中.

我更喜欢使用包含值列表而不是单个值的不同对象类型来获取java.util.stream.Stream作为结果.

我唯一的方法是执行合并的自定义迭代器,但转换为迭代器并返回流似乎非常难看.

它的最佳方法是什么？

Answer 1

这是SteamEx的解决方案。

int[][] datasets = { { 1, 3 }, { 1, 5 }, { 3, 1 }, { 4, 2 }, { 4, 7 }, { 4, 8 } };

StreamEx.of(datasets) //
        .collapse((a, b) -> a[0] == b[0], groupingBy(a -> a[0], mapping(a -> a[1], toList()))) //
        .forEach(System.out::println);

你可以int[]用你的dataset对象替换。我们可以添加peek来验证它是否是延迟加载/计算：

StreamEx.of(datasets) //
        .peek(System.out::println) //
        .collapse((a, b) -> a[0] == b[0], groupingBy(a -> a[0], mapping(a -> a[1], toList()))) //
        .limit(1) //
        .forEach(System.out::println);