如何使用 Golang 更新控制台中的多行

How to update multiple line in console with Golang

我写了一个 go 程序,它同时执行许多任务(在 goroutines 中),并在控制台上显示这些任务的状态,就像这样 -

[Ongoing] Task1
[Ongoing] Task2
[Ongoing] Task3

现在这些任务中的任何一个都可以先完成或最后完成。我想在同一个地方更新这些任务的状态。如果 Task2 首先完成,那么它应该显示如下内容-

[Ongoing] Task1
[Done] Task2
[Ongoing] Task3

我尝试了 uilive 库,但它总是像这样更新最后一行。 (我觉得不是更新多行)

[Ongoing] Task1
[Ongoing] Task2
[Done] Task2

如何实现?

Stdout 是一个流,因此您不能处理已经打印的行或字符并在以后更改它。 CLI 库中的此类更新是通过向后移动光标(通过打印转义序列 - 影响用户终端的不可打印字符序列)并覆盖文本来进行的。为此,需要一个参考点,以便图书馆(或您)知道光标在哪里以及打印了多少行等。

一种可能的方法是创建一个单独的 goroutine 来处理所有终端输出打印,并让其他执行实际工作的 goroutine 只向它传递更新(例如通过通道)。在此类例程中集中终端的“状态”应该可以更轻松地使用上述技术进行更新。

虽然不是适合您情况的 drop-in 解决方案,但我建议您查看 mpb - 一个允许呈现多个异步更新进度条的库。也许您可以以类似的方式设计您的解决方案或将其用作基础,因为它已经处理了 OS 之间的差异等

我正在寻求实现类似的功能并遇到了这个问题。不确定您是否仍在寻找解决方案,但我找到了一个 :)

结果是 uilive can very well be used to update multiple lines, by making use of the Newline()*uilive.Writer 上的函数。在他们的存储库中编辑示例以在多行中写入下载进度,我们得到:

writer := uilive.New()       // writer for the first line
writer2 := writer.Newline()  // writer for the second line
// start listening for updates and render
writer.Start()

for i := 0; i <= 100; i++ {
    fmt.Fprintf(writer, "Downloading File 1.. %d %%\n", i)
    fmt.Fprintf(writer2, "Downloading File 2.. %d %%\n", i)
    time.Sleep(time.Millisecond * 5)
}

fmt.Fprintln(writer, "Finished downloading both files :)")
writer.Stop() // flush and stop rendering

为了实现跟踪多个并发 运行 任务的进度,正如@blami 的回答也指出的那样,一个好的方法是有一个单独的 goroutine 来处理所有终端输出。

如果您有兴趣,我已经在 mllint, my tool for linting machine learning projects. It runs all its underlying linters (i.e. tasks) in parallel and prints their progress (running / done) in the manner that you describe in your question. See this file for the implementation of that. Specifically, the printTasks 函数中实现了这个确切的功能,从系统中的任务列表中读取并在新行上打印每个任务的显示名称和状态。

请注意,特别是对于多行打印,控制刷新打印缓冲区变得很重要,因为您不希望编写器在向缓冲区写入更新的过程中自动刷新。因此,我将 writer 的 RefreshDuration 设置为较大的值(例如 time.Hour)并在打印所有行后调用 writer Flush(),即在 printTasks 的末尾。