大家好,我是 @洛竹
本文首发于 洛竹的官方网站
本文同步于公众号『洛竹早茶馆』,转载请联系作者。
创作不易,养成习惯,素质三连!
mpb 是一个用于在终端应用程序中渲染进度条的 Go 库。
特性
- 多条进度条:支持多个进度条
- 动态总数:进度条运行时设置总数
- 动态添加/删除:动态添加或删除进度条
- 取消:取消整个渲染过程
- 预定义装饰器:时间、基于 EWMA 的 ETA、百分比、字节计数器。
- 装饰器的宽度同步:在多个进度条之间同步装饰器的宽度。
使用
渲染单个进度条
package main
import (
"math/rand"
"time"
"github.com/vbauerster/mpb/v6"
"github.com/vbauerster/mpb/v6/decor"
)
func main() {
// 使用自定义的宽度初始化进度条容器
p := mpb.New(mpb.WithWidth(64))
total := 100
name := "Single Bar:"
// 添加单个进度条,这个进度条会继承容器的宽度
bar := p.Add(int64(total),
// 使用自定义的样式填充进度条
mpb.NewBarFiller("╢▌▌░╟"),
mpb.PrependDecorators(
// 右边一格显示我们的名字
decor.Name(name, decor.WC{W: len(name) + 1, C: decor.DidentRight}),
// 用 `done` 消息替换 ETA 装饰器,`OnComplete` 事件。
decor.OnComplete(
decor.AverageETA(decor.ET_STYLE_GO, decor.WC{W: 4}), "done",
),
),
mpb.AppendDecorators(decor.Percentage()),
)
// 模拟一些工作
max := 100 * time.Millisecond
for i := 0; i < total; i++ {
time.Sleep(time.Duration(rand.Intn(10)+1) * max / 10)
bar.Increment()
}
// 等着我们的进度条完成并清理
p.Wait()
}
渲染多进度条
var wg sync.WaitGroup
// 传递 &wg(可选的),所以 p 最终会等等它
p := mpb.New(mpb.WithWaitGroup(&wg))
total, numBars := 100, 3
wg.Add(numBars)
for i := 0; i < numBars; i++ {
name := fmt.Sprintf("Bar#%d:", i)
bar := p.AddBar(int64(total),
mpb.PrependDecorators(
// 简单的 name 装饰器
decor.Name(name),
// decor.DSyncWidth 使列宽同步化。
decor.Percentage(decor.WCSyncSpace),
),
mpb.AppendDecorators(
// 用 `done` 消息替换 ETA 装饰器,`OnComplete` 事件。
decor.OnComplete(
decor.EwmaETA(decor.ET_STYLE_GO, 60), "done",
),
),
)
// 模拟一些工作
go func() {
defer wg.Done()
rng := rand.New(rand.NewSource(time.Now().UnixNano()))
max := 100 * time.Millisecond
for i := 0; i < total; i++ {
start := time.Now()
time.Sleep(time.Duration(rng.Intn(10)+1) * max / 10)
bar.Increment()
bar.DecoratorEwmaUpdate(time.Since(start))
}
}()
}
p.Wait()