在 Golang 中用于执行命令的库是 os/exec,exec.Command 函数返回一个 Cmd 对象,根据不同的需求,可以将命令的执行分为三种情况

  1. 只执行命令,不获取结果
  2. 执行命令,并获取结果(不区分 stdout 和 stderr)
  3. 执行命令,并获取结果(区分 stdout 和 stderr)

第一种:只执行命令,不获取结果

直接调用 Cmd 对象的 Run 函数,返回的只有成功和失败,获取不到任何输出的结果。

 1package main
 2
 3import (
 4    "log"
 5    "os/exec"
 6)
 7
 8func main() {
 9    cmd := exec.Command("ls", "-l", "/var/log/")
10    err := cmd.Run()
11    if err != nil {
12        log.Fatalf("cmd.Run() failed with %s\n", err)
13    }
14} 

第二种:执行命令,并获取结果

有时候我们执行一个命令就是想要获取输出结果,此时你可以调用 Cmd 的 CombinedOutput 函数。

 1package main
 2
 3import (
 4"fmt"
 5"log"
 6"os/exec"
 7)
 8
 9func main() {
10    cmd := exec.Command("ls", "-l", "/var/log/")
11    out, err := cmd.CombinedOutput()
12    if err != nil {
13        fmt.Printf("combined out:\n%s\n", string(out))
14        log.Fatalf("cmd.Run() failed with %s\n", err)
15    }
16    fmt.Printf("combined out:\n%s\n", string(out))
17} 

CombinedOutput 函数,只返回 out,并不区分 stdout 和 stderr。如果你想区分他们,可以直接看第三种方法。

 1$ go run demo.go 
 2combined out:
 3total 11540876
 4-rw-r--r--  2 root       root         4096 Oct 29  2018 yum.log
 5drwx------  2 root       root           94 Nov  6 05:56 audit
 6-rw-r--r--  1 root       root    185249234 Nov 28  2019 message
 7-rw-r--r--  2 root       root        16374 Aug 28 10:13 boot.log

不过在那之前,我却发现一个小问题:有时候,shell 命令能执行,并不代码 exec 也能执行。

比如我只想查看 /var/log/ 目录下的 log 后缀名的文件呢?有点 Linux 基础的同学,都会用这个命令

 1$ ls -l /var/log/*.log
 2total 11540
 3-rw-r--r--  2 root       root         4096 Oct 29  2018 /var/log/yum.log
 4-rw-r--r--  2 root       root        16374 Aug 28 10:13 /var/log/boot.log

按照这个写法将它放入到 exec.Command

 1package main
 2
 3import (
 4"fmt"
 5"log"
 6"os/exec"
 7)
 8
 9func main() {
10    cmd := exec.Command("ls", "-l", "/var/log/*.log")
11    out, err := cmd.CombinedOutput()
12    if err != nil {
13        fmt.Printf("combined out:\n%s\n", string(out))
14        log.Fatalf("cmd.Run() failed with %s\n", err)
15    }
16    fmt.Printf("combined out:\n%s\n", string(out))
17} 

什么情况?居然不行,报错了。

 1$ go run demo.go 
 2combined out:
 3ls: cannot access /var/log/*.log: No such file or directory
 4
 52020/11/11 19:46:00 cmd.Run() failed with exit status 2
 6exit status 1

为什么会报错呢?shell 明明没有问题啊

其实很简单,原来 ls -l /var/log/*.log 并不等价于下面这段代码。

 1exec.Command("ls", "-l", "/var/log/*.log")

上面这段代码对应的 Shell 命令应该是下面这样,如果你这样子写,ls 就会把参数里的内容当成具体的文件名,而忽略通配符 *

 1$ ls -l "/var/log/*.log"
 2ls: cannot access /var/log/*.log: No such file or directory

第三种:执行命令,并区分stdout 和 stderr

上面的写法,无法实现区分标准输出和标准错误,只要换成下面种写法,就可以实现。

 1package main
 2
 3import (
 4    "bytes"
 5    "fmt"
 6    "log"
 7    "os/exec"
 8)
 9
10func main() {
11    cmd := exec.Command("ls", "-l", "/var/log/*.log")
12    var stdout, stderr bytes.Buffer
13    cmd.Stdout = &stdout  // 标准输出 
14
15    cmd.Stderr = &stderr  // 标准错误 
16
17    err := cmd.Run()
18    outStr, errStr := string(stdout.Bytes()), string(stderr.Bytes())
19    fmt.Printf("out:\n%s\nerr:\n%s\n", outStr, errStr)
20    if err != nil {
21        log.Fatalf("cmd.Run() failed with %s\n", err)
22    }
23} 

输出如下,可以看到前面的报错内容被归入到标准错误里

 1$ go run demo.go 
 2out:
 3
 4err:
 5ls: cannot access /var/log/*.log: No such file or directory
 6
 72020/11/11 19:59:31 cmd.Run() failed with exit status 2
 8exit status 1

第四种:多条命令组合,请使用管道

将上一条命令的执行输出结果,做为下一条命令的参数。在 Shell 中可以使用管道符 | 来实现。

比如下面这条命令,统计了 message 日志中 ERROR 日志的数量。

 1$ grep ERROR /var/log/messages | wc -l
 219

类似的,在 Golang 中也有类似的实现。

 1package main
 2import (
 3    "os"
 4    "os/exec"
 5)
 6func main() {
 7    c1 := exec.Command("grep", "ERROR", "/var/log/messages")
 8    c2 := exec.Command("wc", "-l")
 9    c2.Stdin, _ = c1.StdoutPipe()
10    c2.Stdout = os.Stdout
11    _ = c2.Start()
12    _ = c1.Run()
13    _ = c2.Wait()
14} 

输出如下

 1$ go run demo.go 
 219

第五种:设置命令级别的环境变量

使用 os 库的 Setenv 函数来设置的环境变量,是作用于整个进程的生命周期

 1package main
 2import (
 3    "fmt"
 4    "log"
 5    "os"
 6    "os/exec"
 7)
 8func main() {
 9    os.Setenv("NAME", "wangbm")
10    cmd := exec.Command("echo", os.ExpandEnv("$NAME"))
11    out, err := cmd.CombinedOutput()
12    if err != nil {
13        log.Fatalf("cmd.Run() failed with %s\n", err)
14    }
15    fmt.Printf("%s", out)
16} 

只要在这个进程里,NAME 这个变量的值都会是 wangbm,无论你执行多少次命令

 1$ go run demo.go 
 2wangbm

如果想把环境变量的作用范围再缩小到命令级别,也是有办法的。

为了方便验证,我新建个 sh 脚本,内容如下

 1$ cat /home/wangbm/demo.sh
 2echo $NAME
 3$ bash /home/wangbm/demo.sh   # 由于全局环境变量中没有 NAME所以无输出 

另外,demo.go 里的代码如下

 1package main
 2import (
 3    "fmt"
 4    "os"
 5    "os/exec"
 6)
 7
 8func ChangeYourCmdEnvironment(cmd * exec.Cmd) error {
 9    env := os.Environ()
10    cmdEnv := []string{}
11
12    for _, e := range env {
13        cmdEnv = append(cmdEnv, e)
14    }
15    cmdEnv = append(cmdEnv, "NAME=wangbm")
16    cmd.Env = cmdEnv
17
18    return nil
19}
20
21func main() {
22    cmd1 := exec.Command("bash", "/home/wangbm/demo.sh")
23  ChangeYourCmdEnvironment(cmd1) // 添加环境变量到 cmd1 命令: NAME=wangbm 
24
25    out1, _ := cmd1.CombinedOutput()
26    fmt.Printf("output: %s", out1)
27
28    cmd2 := exec.Command("bash", "/home/wangbm/demo.sh")
29    out2, _ := cmd2.CombinedOutput()
30    fmt.Printf("output: %s", out2)
31} 

执行后,可以看到第二次执行的命令,是没有输出 NAME 的变量值。

 1$ go run demo.go 
 2output: wangbm
 3output: 
个人笔记记录 2021 ~ 2025