如何将计数器插入流<字符串> .forEach（）？

foreach java line-numbers java-stream

2022-09-01 11:18:57

FileWriter writer = new FileWriter(output_file);
    int i = 0;

    try (Stream<String> lines = Files.lines(Paths.get(input_file))) {
        lines.forEach(line -> {
            try {
                writer.write(i + " # " + line + System.lineSeparator());
            } catch (Exception e) {
                e.printStackTrace();
            }   
        }
                    );
        writer.close();
    }

我需要用行号写一行，所以我试图在.forEach（）中添加一个计数器，但我无法让它工作。我只是不知道把i ++放在哪里;进入代码，随机拧来拧去到目前为止没有帮助。

答案 1

您可以将用作可变计数器。AtomicIntegerfinal

public void test() throws IOException {
    // Make sure the writer closes.
    try (FileWriter writer = new FileWriter("OutFile.txt") ) {
        // Use AtomicInteger as a mutable line count.
        final AtomicInteger count = new AtomicInteger();
        // Make sure the stream closes.
        try (Stream<String> lines = Files.lines(Paths.get("InFile.txt"))) {
            lines.forEach(line -> {
                        try {
                            // Annotate with line number.
                            writer.write(count.incrementAndGet() + " # " + line + System.lineSeparator());
                        } catch (Exception e) {
                            e.printStackTrace();
                        }
                    }
            );
        }
    }
}

答案 2

这是一个很好的例子，说明您应该使用一个好的老式for循环。虽然专门给出了顺序流，但流可以无序地生成和处理，因此插入计数器并依赖其顺序是一个相当糟糕的习惯。如果您仍然真的想这样做，请记住，在任何地方都可以使用 lambda，您仍然可以使用完整的匿名类。匿名类是普通类，因此可以具有状态。Files.lines()

因此，在您的示例中，您可以这样做：

FileWriter writer = new FileWriter(output_file);

try (Stream<String> lines = Files.lines(Paths.get(input_file))) {
    lines.forEach(new Consumer<String>() {
        int i = 0;
        void accept(String line) {
            try {
                writer.write((i++) + " # " + line + System.lineSeparator());
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    });
    writer.close();
}