[全]Go语言中用 os exec 执行命令的五种姿势

1、Go语言中用 os/exec 执行命令的五种姿势在 Golang 中用于执行命令的库是os/exec，exec.Command 函数返回一个Cmd对象，根据不同的需求，可以将命令的执行分为三种情况1. 只执行命令，不获取结果2. 执行命令，并获取结果（不区分 stdout 和 stderr）3. 执行命令，并获取结果（区分 stdout 和 stderr）第一种：只执行命令，不获取结果直接调用 Cmd 对象的 Run 函数，返回的只有成功和失败，获取不到任何输出的结果。package mainimport ( log os/exec)func main() cmd := exec.Comman

2、d(ls, -l, /var/log/) err := cmd.Run() if err != nil log.Fatalf(cmd.Run() failed with %sn, err) 第二种：执行命令，并获取结果有时候我们执行一个命令就是想要获取输出结果，此时你可以调用 Cmd 的 CombinedOutput 函数。package mainimport (fmtlogos/exec)func main() cmd := exec.Command(ls, -l, /var/log/) out, err := cmd.CombinedOutput() if err != nil fmt.P

3、rintf(combined out:n%sn, string(out) log.Fatalf(cmd.Run() failed with %sn, err) fmt.Printf(combined out:n%sn, string(out)CombinedOutput 函数，只返回 out，并不区分 stdout 和 stderr。如果你想区分他们，可以直接看第三种方法。$ go run demo.go combined out:total 11540876-rw-r-r- 2 root root 4096 Oct 29 2018 yum.logdrwx- 2 root root 94 No

4、v 6 05:56 audit-rw-r-r- 1 root root 185249234 Nov 28 2019 message-rw-r-r- 2 root root 16374 Aug 28 10:13 boot.log不过在那之前，我却发现一个小问题：有时候，shell 命令能执行，并不代码 exec 也能执行。比如我只想查看/var/log/目录下的 log 后缀名的文件呢？有点 Linux 基础的同学，都会用这个命令$ ls -l /var/log/*.logtotal 11540-rw-r-r- 2 root root 4096 Oct 29 2018 /var/log/yum.

5、log-rw-r-r- 2 root root 16374 Aug 28 10:13 /var/log/boot.log按照这个写法将它放入到exec.Commandpackage mainimport (fmtlogos/exec)func main() cmd := exec.Command(ls, -l, /var/log/*.log) out, err := cmd.CombinedOutput() if err != nil fmt.Printf(combined out:n%sn, string(out) log.Fatalf(cmd.Run() failed with %sn,

6、err) fmt.Printf(combined out:n%sn, string(out)什么情况？居然不行，报错了。$ go run demo.go combined out:ls: cannot access /var/log/*.log: No such file or directory2020/11/11 19:46:00 cmd.Run() failed with exit status 2exit status 1为什么会报错呢？shell 明明没有问题啊其实很简单，原来ls -l /var/log/*.log并不等价于下面这段代码。exec.Command(ls, -l, /

7、var/log/*.log)上面这段代码对应的 Shell 命令应该是下面这样，如果你这样子写，ls 就会把参数里的内容当成具体的文件名，而忽略通配符*$ ls -l /var/log/*.logls: cannot access /var/log/*.log: No such file or directory第三种：执行命令，并区分stdout 和 stderr上面的写法，无法实现区分标准输出和标准错误，只要换成下面种写法，就可以实现。package mainimport ( bytes fmt log os/exec)func main() cmd := exec.Command(ls,

8、 -l, /var/log/*.log) var stdout, stderr bytes.Buffer cmd.Stdout = &stdout / 标准输出 cmd.Stderr = &stderr / 标准错误 err := cmd.Run() outStr, errStr := string(stdout.Bytes(), string(stderr.Bytes() fmt.Printf(out:n%snerr:n%sn, outStr, errStr) if err != nil log.Fatalf(cmd.Run() failed with %sn, err) 输出如下，可以看到

9、前面的报错内容被归入到标准错误里$ go run demo.go out:err:ls: cannot access /var/log/*.log: No such file or directory2020/11/11 19:59:31 cmd.Run() failed with exit status 2exit status 1第四种：多条命令组合，请使用管道将上一条命令的执行输出结果，做为下一条命令的参数。在 Shell 中可以使用管道符|来实现。比如下面这条命令，统计了 message 日志中 ERROR 日志的数量。$ grep ERROR /var/log/messages |

10、wc -l19类似的，在 Golang 中也有类似的实现。package mainimport ( os os/exec)func main() c1 := exec.Command(grep, ERROR, /var/log/messages) c2 := exec.Command(wc, -l) c2.Stdin, _ = c1.StdoutPipe() c2.Stdout = os.Stdout _ = c2.Start() _ = c1.Run() _ = c2.Wait()输出如下$ go run demo.go 19第五种：设置命令级别的环境变量使用 os 库的 Setenv 函数

11、来设置的环境变量，是作用于整个进程的生命周期的。package mainimport ( fmt log os os/exec)func main() os.Setenv(NAME, wangbm) cmd := exec.Command(echo, os.ExpandEnv($NAME) out, err := cmd.CombinedOutput() if err != nil log.Fatalf(cmd.Run() failed with %sn, err) fmt.Printf(%s, out)只要在这个进程里，NAME这个变量的值都会是wangbm，无论你执行多少次命令$ go r

12、un demo.go wangbm如果想把环境变量的作用范围再缩小到命令级别，也是有办法的。为了方便验证，我新建个 sh 脚本，内容如下$ cat /home/wangbm/demo.shecho $NAME$ bash /home/wangbm/demo.sh # 由于全局环境变量中没有 NAME，所以无输出另外，demo.go 里的代码如下package mainimport ( fmt os os/exec)func ChangeYourCmdEnvironment(cmd * exec.Cmd) error env := os.Environ() cmdEnv := string for _, e := range env cmdEnv = append(cmdEnv, e) cmdEnv = append(cmdEnv, NAME=wangbm) cmd.Env = cmdEnv return nilfunc main() cmd1 := exec.Command(bash, /home/wangbm/demo.sh) ChangeYourCmdEnvironment(cmd1) / 添加环境变量到 cmd1 命令: NAME=wangbm out1, _ := cmd1.CombinedOutput() fmt.Pri