使用mpb库实现并发命令执行的进度条监控

在Go语言开发中，mpb库是一个非常实用的进度条显示工具。本文将介绍如何利用mpb库为并发执行的shell命令添加进度条监控功能。

问题背景

在实际开发中，我们经常需要执行多个shell命令并监控它们的执行进度。mpb库提供了丰富的进度条功能，但在并发场景下使用时需要注意一些细节。

初始方案分析

最初的实现方案存在几个关键问题：

1. 进度条提前完成：所有进度条会在最后一个命令完成时同时结束
2. 结果显示不正确：结果显示在进度条开始时就显示，而不是命令完成后
3. 同步问题：存在潜在的竞态条件

改进方案

经过分析，我们改进了实现方案，主要优化点包括：

1. 使用独立的goroutine管理每个命令的执行
2. 正确同步进度条状态更新
3. 确保结果显示在命令完成后

关键实现细节

命令执行器结构

type Runner struct {
    mu      sync.Mutex
    Command string
    Success bool
}

这个结构体封装了命令执行的相关信息，包括命令字符串、执行状态和互斥锁。

进度条更新机制

func (r *Runner) Exec(bar *mpb.Bar) {
    r.mu.Lock()
    defer r.mu.Unlock()

    cmd := exec.Command("bash")
    cmd.Stdin = strings.NewReader(r.Command)

    errbuf := &bytes.Buffer{}
    cmd.Stderr = errbuf

    done := make(chan bool)
    start := time.Now()
    ticker := time.NewTicker(20 * time.Millisecond)

    go func() {
        _, err := cmd.Output()
        if errbuf.String() == "" && err == nil {
            r.Success = true
        } else {
            fmt.Printf("Error: %s - %s\n", errbuf.String(), err.Error())
            bar.Abort(true)
        }
        done <- true
    }()

    running := true
    for running {
        select {
        case <-ticker.C:
            bar.EwmaIncrement(time.Since(start))
        case <-done:
            running = false
        }
    }
}

这段代码实现了命令执行和进度条更新的核心逻辑，使用ticker定期更新进度条状态。

主程序流程

func main() {
    var wg sync.WaitGroup

    bars := mpb.New(
        mpb.WithWaitGroup(&wg),
        mpb.WithWidth(1),
        mpb.WithAutoRefresh(),
    )

    barFiller := func() mpb.BarFillerBuilder {
        return mpb.SpinnerStyle("⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏")
    }

    for i, cmd := range os.Args {
        if i == 0 {
            continue
        }

        wg.Add(1)

        go func() {
            defer wg.Done()

            r := Runner{Command: cmd}
            bar := bars.New(1,
                barFiller(),
                mpb.PrependDecorators(decor.Name(fmt.Sprintf("%30s - ", cmd))),
                mpb.AppendDecorators(
                    decor.OnComplete(
                        decor.Elapsed(decor.ET_STYLE_GO), fmt.Sprintf("Result: %t", r.Success),
                    ),
                ),
            )

            r.Exec(bar)
        }()
    }

    wg.Wait()
    bars.Wait()
}

主程序创建多个goroutine并发执行命令，每个命令都有独立的进度条显示。

经验总结

1. 在并发场景下使用mpb时，必须确保进度条更新的同步性
2. 使用WaitGroup正确管理goroutine的生命周期
3. 进度条的装饰器应该在命令完成后才显示最终结果
4. 适当的更新频率可以平衡性能和平滑度

通过这种方式，我们可以为并发执行的shell命令提供直观的进度反馈，提升用户体验。