详解Go语言中用 os/exec 执行命令的五种方法

完整系列教程详见：

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

第一种：只执行命令，不获取结果

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

package mainimport ("log""os/exec")func main() {cmd := exec.command("ls", "-l", "/var/log/")err := cmd.run()if err != nil {log.fatalf("cmd.run() failed with %s\n", err)}}

第二种：执行命令，并获取结果

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

package mainimport ("fmt""log""os/exec")func main() {cmd := exec.command("ls", "-l", "/var/log/")out, err := cmd.combinedoutput()if err != nil { fmt.printf("combined out:\n%s\n", string(out))log.fatalf("cmd.run() failed with %s\n", err)}fmt.printf("combined out:\n%s\n", 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 nov 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/ 目录下的 lo逆问g 后缀名的文件呢？有点 linux 基础的同学，都会用这个命令

$ ls -l /var/log/*.logtotal 1延毕是什么意思1540-rw-r--r-- 2 root root  4096 oct 29 2018 /var/log/yum.log-rw-r--r-- 2 root root 16374 aug 28 10:13 /var/log/boot.log

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

package mainimport ("fmt""log""os/exec")func main() {cmd := exec.command("ls", "-l", "/var/log/*.log")out, err := cmd.combinedoutput()if err != nil { fmt.printf("combined out:\n%s\n", string(out))log.fatalf("cmd.run() failed with %s\n", err)}fmt.printf("combined out:\n%s\n", 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", "/var/log/*.log")

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

$ ls -l "/var/log/*.log"ls: 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", "-l", "/var/log/*.log")var stdout, stderr bytes.buffercmd.stdout = &stdout // 标准输出cmd.stderr = &stderr // 标准错误err := cmd.run()outstr, errstr := string(stdout.bytes()), string(stderr.bytes())fmt.printf("out:\n%s\nerr:\n%s\n", outstr, errstr)if err != nil {log.fatalf("cmd.run() failed with %s\n", err)}}

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

$ 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 | 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 库的 tenv 函数来设置的环境变量，是作用于整个进程的生命周期的。

package mainimport ("fmt""log""os""os少数民族最多的省/exec")func ma读书笔记摘抄大全30篇in() {os.tenv("name", "wangbm")cmd := exec.command("echo", os.expandenv("$name"))out, err := cmd.combinedoutput()if err != nil {log.fatalf("cmd.run() failed with %s\n", err)}fmt.printf("%s", out)}

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

$ go run 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 = cmdenvreturn nil}func main() {cmd1 := exec.command("bash", "/home/wangbm/demo.sh") changeyourcmdenvironment(cmd1) // 添加环境变量到 cmd1 命令: name=wangbmout1, _ := cmd1.combinedoutput()fmt.printf("output: %s", out1)cmd2 := exec.command("bash", "/home/wangbm/demo.sh")out2, _ := cmd2.combinedoutput()fmt.printf("output: %s", out2)}

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

$ go run demo.go output: wangbmoutput: 

到此这篇关于go语言中用 os/exec 执行命令的五种方法的文章就介绍到这了,更多相关go语言执行命令内容请搜索www.887551.com以前的文章或继续浏览下面的相关文章希望大家以后多多支持www.887551.com！