使用 goroutine 执行 Bash 命令 n 次，并将结果存储和打印出来

Question

使用 goroutine 执行 Bash 命令 n 次，并将结果存储和打印出来

4

我对Golang比较新，在尝试执行一个带有参数的bash命令n次，然后将输出存储在变量中并进行打印。

我可以只执行一次，或者像下面这样使用循环：

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)
}

这个很好用，我能轻松地阅读输出。

现在，我想使用 goroutines 重复执行 n 次这个操作。

我尝试了遵循回答如何定义一个一次性在 Golang 中执行的 goroutine 池？的人的方法，但我无法让它工作。

到目前为止，这是我尝试过的：

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并将其打印出来。

我该如何做到这一点？

提前感谢您对问题的任何解释，如果需要进一步澄清，请留下评论。

- AndreaM16

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- ramrunner · Accepted Answer

因此，以下方法可以解决您的问题：

您可以调用Cmd.Output()方法获取命令的输出。或者，您可以将Cmd.StdOutPipe连接到一个byte.Buffer中，然后从其中读取。
您的goroutine逻辑是错误的。它只会执行4次，因为waitgroup被Done()标记并且主程序退出了。在主程序关闭通道来通知工作线程退出for range循环后，您应该调用wg.Done。我使用defer关键字来实现。
当您使用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")

}

`