如何将按起始记录拆分文件的Java代码转为Java 8 Stream实现？

当然可以用Java 8 Stream来重构这段代码！不过因为Stream是无状态的，而你的需求是基于REQ00起始标记的有状态分组（每遇到标记就开启新的数据块），所以我们需要自定义一个Collector来处理这种状态跟踪逻辑，下面是具体实现：

完整Stream实现代码 #

import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;
import java.util.StringBuilder;
import java.util.stream.Collector;

public class FileSplitter {
    public static void main(String[] args) {
        try {
            List<StringBuilder> result = Files.lines(Paths.get("your-file-path"))
                    .collect(createBlockCollector());
            
            // 这里可以处理拆分后的结果
            result.forEach(block -> System.out.println("Block content:\n" + block));
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    private static Collector<String, ?, List<StringBuilder>> createBlockCollector() {
        return Collector.of(
                // 供应器：初始化状态容器，同时保存结果列表和当前正在构建的块
                () -> new Object() {
                    List<StringBuilder> blocks = new ArrayList<>();
                    StringBuilder currentBlock = null;
                },
                // 累加器：处理每一行，逻辑和原代码完全对齐
                (container, line) -> {
                    if (line.startsWith("REQ00")) {
                        container.currentBlock = new StringBuilder();
                        container.blocks.add(container.currentBlock);
                    }
                    // 注意：和原代码逻辑一致，如果第一行不是REQ00开头，会抛出NPE
                    if (container.currentBlock != null) {
                        container.currentBlock.append(line);
                    }
                },
                // 组合器：并行流场景下合并两个容器的结果
                (container1, container2) -> {
                    container1.blocks.addAll(container2.blocks);
                    return container1;
                },
                // 终结器：返回最终的拆分结果列表
                container -> container.blocks
        );
    }
}

代码说明 #

1. 自定义Collector的核心逻辑：

   • 用一个匿名内部类作为状态容器，同时维护拆分后的结果列表blocks和当前正在构建的currentBlock，解决Stream无状态的限制。
   • 累加器的逻辑和你原来的代码完全一致：遇到REQ00开头的行就新建StringBuilder并加入列表，然后将当前行追加到当前块中。
   • 组合器是为了兼容并行流场景（虽然处理文件时并行流的收益有限，但Collector要求必须实现），只是简单合并两个容器的结果列表。

2. 和原代码的一致性：

   • 完全保留了原代码的行为：如果文件第一行不是REQ00开头，会抛出NullPointerException。如果你想优化这个场景，可以在累加器里加个判断（比如当currentBlock为null时自动新建一个块）。

3. 优势：

   • 用Stream API让代码更符合函数式编程风格，同时封装了状态跟踪逻辑，让主流程更简洁。

内容的提问来源于stack exchange，提问作者Stefano R.