如何在 Java 并行流中使用 print("\r"+progressMessage)？

Question

我正在尝试通过编写类似的内容来显示我的并行流的进度

int total=myList.size();
AtomicInteger counter=new AtomicInteger(0);

List<String> myResult=IntStream.range(0, total)
                    .parallel()
                    .mapToObj(i-> modify(myList.get(i)))
                    .peek(i->System.out.print("\r"+(counter.getAndIncrement()*100/total)+"%"))
                    .collect(toList());

我的问题来自“\r”。鉴于它是并行化的，当并发事件发生时，真正到达行首所需的“\r”数量可能会有所不同。所以有时我可以读到“70% 71%”...

Answer 1

将进度记录与进度输出操作分开：

int total = myList.size();
AtomicInteger counter = new AtomicInteger(0);
ScheduledExecutorService es = Executors.newScheduledThreadPool(1);

ScheduledFuture<?> f = es.scheduleWithFixedDelay(new Runnable() {
    int lastReported = -1;
    public void run() {
        int newValue = counter.get();
        if(newValue != lastReported) {
            lastReported = newValue;
            System.out.append("\r"+newValue*100/total+"%").flush();
        }
    }
}, 100, 100, TimeUnit.MILLISECONDS);

List<String> myResult = IntStream.range(0, total)
    .parallel()
    .mapToObj(i -> modify(myList.get(i)))
    .peek(i -> counter.incrementAndGet())
    .collect(toList());

f.cancel(true);
es.shutdown();
System.out.print("\r100%");

现在，打印始终由一个线程完成。这种分离解决了更多的问题。 对每个元素执行打印操作会显着减慢您的实际处理速度。通过使用预定的打印作业，您可以自己控制开销，即您不需要比人类阅读速度更快的打印，并且如果自上次更新以来百分比没有变化，您也不需要执行昂贵的打印操作。

请注意，当所有元素都通过peek 操作时，流操作尚未完全完成（至少有一个合并操作待处理），但这是您可以使用当前 API 获得的最佳进度估计。