Go - 使用 goroutine 执行一个 Bash 命令 n 次并存储并打印其结果

Question

总的来说，我对Golang 很陌生，我正在尝试执行bash command with its arguments n times，然后将Output 存储在一个变量中，然后将Print 存储在其中。

我可以只做一次，或者只使用loops，如下所示：

package main

import (
    "fmt"
    "os/exec"
    "os"
    "sync"
)


func main() {

    //Default Output
    var (
        cmdOut []byte
        err    error
    )

    //Bash Command
    cmd  := "./myCmd"
   //Arguments to get passed to the command
   args := []string{"arg1", "arg2", "arg3"}

    //Execute the Command
    if cmdOut, err = exec.Command(cmd, args...).Output(); err != nil {
        fmt.Fprintln(os.Stderr, "There was an error running "+cmd+" "+args[0]+args[1]+args[2], err)
        os.Exit(1)
    }
    //Store it
    sha := string(cmdOut)
    //Print it
    fmt.Println(sha)
}

这很好用，我可以轻松阅读output。

现在，我想使用 goroutines 重复同样的操作 n 次。

我尝试采用与回答 How would you define a pool of goroutines to be executed at once in Golang? 的人完全相同的方法，但我无法使其发挥作用。

这就是我到目前为止所尝试的：

package main

import (
    "fmt"
    "os/exec"
    "sync"
)


func main() {

    //Bash Command
    cmd  := "./myCmd"
    //Arguments to get passed to the command
     args := []string{"arg1", "arg2", "arg3"}

    //Common Channel for the goroutines
    tasks := make(chan *exec.Cmd, 64)

    //Spawning 4 goroutines
    var wg sync.WaitGroup
    for i := 0; i < 4; i++ {
        wg.Add(1)
        go func() {
            for cmd := range tasks {
                cmd.Run()
            }
            wg.Done()
        }()
    }

    //Generate Tasks
    for i := 0; i < 10; i++ {
        tasks <- exec.Command(cmd, args...)
        //Here I should somehow print the result of the latter command
    }
    close(tasks)

    // wait for the workers to finish
    wg.Wait()

    fmt.Println("Done")

}

但是，我真的不知道如何存储执行命令的i-result 并打印它。

我怎样才能做到这一点？

在此先感谢，如果您对问题有任何澄清，请发表评论。

Answer 1

所以以下解决了您的问题

• 您可以调用 Cmd.Output() 来获取命令的输出。 否则您可以将 Cmd.StdOutPipe 管道连接到 byte.Buffer 示例并从中阅读。

• 您的 goroutine 逻辑错误。它只会执行 4 次，因为等待组将是 Done() 并且 main 将退出。您正确地从 main 关闭通道以向工作人员发出退出 for range 循环的信号。 wg.Done 应该在那之后被调用，所以我推迟了它。

• 当您使用 go 命令执行匿名函数时，尽量不要从父作用域捕获任何内容。它可能导致灾难。而是找出你想要的参数并将它们传递给函数。

`

package main 

import (
    "fmt"
    "os/exec"
    "sync"
)
func main() {
    cmd := "./foo.sh"
    //Arguments to get passed to the command
    args := []string{"bar", "baz"}

    //Common Channel for the goroutines
    tasks := make(chan *exec.Cmd, 64)

    //Spawning 4 goroutines
    var wg sync.WaitGroup
    for i := 0; i < 4; i++ {
            wg.Add(1)
            go func(num int, w *sync.WaitGroup) {
                    defer w.Done()
                    var (
                            out []byte
                            err error
                    )
                    for cmd := range tasks { // this will exit the loop when the channel closes
                            out, err = cmd.Output()
                            if err != nil {
                                    fmt.Printf("can't get stdout:", err)
                            }
                            fmt.Printf("goroutine %d command output:%s", num, string(out))
                    }
            }(i, &wg)
    }
    //Generate Tasks
    for i := 0; i < 10; i++ {
            tasks <- exec.Command(cmd, args...)
    }
    close(tasks)

    // wait for the workers to finish
    wg.Wait()

    fmt.Println("Done")

}

`