本文介绍了如何在Go语言中使用os/exec包执行外部命令，并通过标准输入（stdin）向命令传递数据，以及从标准输出（stdout）接收数据。重点讲解了在使用goroutine处理stdin和stdout时可能遇到的并发问题，并提供使用sync.WaitGroup来解决这些问题的示例代码。
在使用Go语言执行外部命令时，os/exec包提供了强大的功能。其中，通过标准输入（stdin）传递数据，并从标准输出（stdout）接收数据是常见的需求。然而，在并发环境下处理这些输入输出流时，需要特别注意goroutine的同步问题，否则可能导致数据丢失或程序hang住。
使用os/exec包执行命令
首先，我们需要使用exec.Command函数创建一个Cmd对象，指定要执行的命令。然后，通过cmd.StdinPipe()和cmd.StdoutPipe()分别获取到命令的标准输入和标准输出的管道。
cmd := exec.Command("cat") // 这里以 cat 命令为例
stdin, err := cmd.StdinPipe()
if err != nil {
log.Panic(err)
}
stdout, err := cmd.StdoutPipe()
if err != nil {
log.Panic(err)
}
err = cmd.Start()
if err != nil {
log.Panic(err)
}通过Stdin传递数据
获取到stdin管道后，我们可以通过io.Copy函数将数据写入管道。为了避免阻塞，通常会在一个goroutine中执行写入操作。需要注意的是，在写入完成后，需要显式关闭stdin管道，通知命令输入结束。
立即学习“go语言免费学习笔记（深入）”；
func populateStdin(str string) func(io.WriteCloser) {
return func(stdin io.WriteCloser) {
defer stdin.Close()
io.Copy(stdin, bytes.NewBufferString(str))
}
}
go populateStdin("hello\n")(stdin)从Stdout接收数据
同样，从stdout管道读取数据也应该在一个goroutine中进行，避免阻塞主线程。可以使用io.Copy将stdout的数据读取到os.Stdout或者一个bytes.Buffer中。
go func() {
io.Copy(os.Stdout, stdout)
}()并发问题与sync.WaitGroup
上述代码看似简单，但在实际运行中，可能会遇到程序hang住，或者数据丢失的问题。这是因为cmd.Wait()函数会在命令执行完成后关闭stdin和stdout管道，即使管道中还有未读取的数据。如果写入stdin和读取stdout的goroutine还没有完成，就会导致数据丢失。
为了解决这个问题，我们需要使用sync.WaitGroup来同步goroutine。在启动写入stdin和读取stdout的goroutine之前，调用wg.Add(2)，表示有两个goroutine需要等待。在每个goroutine完成时，调用wg.Done()。最后，在cmd.Wait()之前，调用wg.Wait()，等待所有goroutine完成。
func runCatFromStdinWorks(populate_stdin_func func(io.WriteCloser)) {
cmd := exec.Command("cat")
stdin, err := cmd.StdinPipe()
if err != nil {
log.Panic(err)
}
stdout, err := cmd.StdoutPipe()
if err != nil {
log.Panic(err)
}
err = cmd.Start()
if err != nil {
log.Panic(err)
}
var wg sync.WaitGroup
wg.Add(2)
go func() {
defer wg.Done()
populate_stdin_func(stdin)
}()
go func() {
defer wg.Done()
io.Copy(os.Stdout, stdout)
}()
wg.Wait()
err = cmd.Wait()
if err != nil {
log.Panic(err)
}
}完整示例代码
下面是一个完整的示例代码，演示了如何使用sync.WaitGroup来解决并发问题：
package main
import (
"bytes"
"io"
"log"
"os"
"os/exec"
"sync"
)
func main() {
runCatFromStdinWorks(populateStdin("aaa\n"))
runCatFromStdinWorks(populateStdin("bbb\n"))
}
func populateStdin(str string) func(io.WriteCloser) {
return func(stdin io.WriteCloser) {
defer stdin.Close()
io.Copy(stdin, bytes.NewBufferString(str))
}
}
func runCatFromStdinWorks(populate_stdin_func func(io.WriteCloser)) {
cmd := exec.Command("cat")
stdin, err := cmd.StdinPipe()
if err != nil {
log.Panic(err)
}
stdout, err := cmd.StdoutPipe()
if err != nil {
log.Panic(err)
}
err = cmd.Start()
if err != nil {
log.Panic(err)
}
var wg sync.WaitGroup
wg.Add(2)
go func() {
defer wg.Done()
populate_stdin_func(stdin)
}()
go func() {
defer wg.Done()
io.Copy(os.Stdout, stdout)
}()
wg.Wait()
err = cmd.Wait()
if err != nil {
log.Panic(err)
}
}总结
在使用Go语言的os/exec包执行外部命令，并通过stdin和stdout进行数据交互时，需要特别注意并发问题。使用sync.WaitGroup可以有效地同步goroutine，避免数据丢失和程序hang住。在实际应用中，可以根据具体需求，灵活调整goroutine的同步策略。
