Java - Stream -每N个元素收集一次

Question

我正在尝试学习java流。我能够进行简单的迭代/过滤/映射/收集等。

当我尝试收集每 3 个元素并打印时，如本例所示，[收集每 3 个元素并打印等等......]

    List<String> list = Arrays.asList("a","b","c","d","e","f","g","h","i","j");

    int count=0;
    String append="";
    for(String l: list){
        if(count>2){
            System.out.println(append);
            System.out.println("-------------------");
            append="";
            count=0;
        }
        append = append + l;
        count++;
    }
    System.out.println(append);

输出：

abc
-------------------
def
-------------------
ghi
-------------------
j

我没有得到任何线索，如何使用流做到这一点。我应该实现自己的收集器来实现这一点吗？

Answer 1

您可以创建自己的收集器。最简单的方法是调用 Collector.of（）。

由于您的用例需要按顺序处理值，因此这里有一个根本不支持并行处理的实现。

public static Collector<String, List<List<String>>, List<List<String>>> blockCollector(int blockSize) {
    return Collector.of(
            ArrayList<List<String>>::new,
            (list, value) -> {
                List<String> block = (list.isEmpty() ? null : list.get(list.size() - 1));
                if (block == null || block.size() == blockSize)
                    list.add(block = new ArrayList<>(blockSize));
                block.add(value);
            },
            (r1, r2) -> { throw new UnsupportedOperationException("Parallel processing not supported"); }
    );
}

试验

List<String> input = Arrays.asList("a","b","c","d","e","f","g","h","i","j");
List<List<String>> output = input.stream().collect(blockCollector(3));
output.forEach(System.out::println);

输出

[a, b, c]
[d, e, f]
[g, h, i]
[j]

Answer 2

如果您的项目中有 Guava，则可以使用 Iterables.partition 方法：

import com.google.common.collect.Iterables;
import com.google.common.collect.Streams;
...

Stream<List<String>> stream = Streams.stream(Iterables.partition(list, 3));

Answer 3

您实际上可以使用 IntStream 来模拟列表的分页。

List<String> list = Arrays.asList("a","b","c","d","e","f","g","h","i","j");

int pageSize = 3;

IntStream.range(0, (list.size() + pageSize - 1) / pageSize)
        .mapToObj(i -> list.subList(i * pageSize, Math.min(pageSize * (i + 1), list.size())))
        .forEach(System.out::println);

哪些输出：

[a, b, c]
[d, e, f]
[g, h, i]
[j]

如果您想要生成字符串，您可以使用< code>String.join,因为您处理的是< code >列表

.mapToObj(i -> String.join("", list.subList(i * pageSize, Math.min(pageSize * (i + 1), list.size()))))