详解Go语言中用 os/exec 执行命令的五种方法
完整系列教程详见:http://golang.iswbm.com
在Golang中用于执行命令的库是os/exec,exec.Command函数返回一个Cmd对象,根据不同的需求,可以将命令的执行分为三种情况
- 只执行命令,不获取结果
- 执行命令,并获取结果(不区分stdout和stderr)
- 执行命令,并获取结果(区分stdout和stderr)
第一种:只执行命令,不获取结果
直接调用Cmd对象的Run函数,返回的只有成功和失败,获取不到任何输出的结果。
packagemain import( "log" "os/exec" ) funcmain(){ cmd:=exec.Command("ls","-l","/var/log/") err:=cmd.Run() iferr!=nil{ log.Fatalf("cmd.Run()failedwith%s\n",err) } }
第二种:执行命令,并获取结果
有时候我们执行一个命令就是想要获取输出结果,此时你可以调用Cmd的CombinedOutput函数。
packagemain import( "fmt" "log" "os/exec" ) funcmain(){ cmd:=exec.Command("ls","-l","/var/log/") out,err:=cmd.CombinedOutput() iferr!=nil{ fmt.Printf("combinedout:\n%s\n",string(out)) log.Fatalf("cmd.Run()failedwith%s\n",err) } fmt.Printf("combinedout:\n%s\n",string(out)) }
CombinedOutput函数,只返回out,并不区分stdout和stderr。如果你想区分他们,可以直接看第三种方法。
$gorundemo.go combinedout: total11540876 -rw-r--r--2rootroot4096Oct292018yum.log drwx------2rootroot94Nov605:56audit -rw-r--r--1rootroot185249234Nov282019message -rw-r--r--2rootroot16374Aug2810:13boot.log
不过在那之前,我却发现一个小问题:有时候,shell命令能执行,并不代码exec也能执行。
比如我只想查看/var/log/目录下的log后缀名的文件呢?有点Linux基础的同学,都会用这个命令
$ls-l/var/log/*.log total11540 -rw-r--r--2rootroot4096Oct292018/var/log/yum.log -rw-r--r--2rootroot16374Aug2810:13/var/log/boot.log
按照这个写法将它放入到exec.Command
packagemain import( "fmt" "log" "os/exec" ) funcmain(){ cmd:=exec.Command("ls","-l","/var/log/*.log") out,err:=cmd.CombinedOutput() iferr!=nil{ fmt.Printf("combinedout:\n%s\n",string(out)) log.Fatalf("cmd.Run()failedwith%s\n",err) } fmt.Printf("combinedout:\n%s\n",string(out)) }
什么情况?居然不行,报错了。
$gorundemo.go combinedout: ls:cannotaccess/var/log/*.log:Nosuchfileordirectory 2020/11/1119:46:00cmd.Run()failedwithexitstatus2 exitstatus1
为什么会报错呢?shell明明没有问题啊
其实很简单,原来ls-l/var/log/*.log并不等价于下面这段代码。
exec.Command("ls","-l","/var/log/*.log")
上面这段代码对应的Shell命令应该是下面这样,如果你这样子写,ls就会把参数里的内容当成具体的文件名,而忽略通配符*
$ls-l"/var/log/*.log" ls:cannotaccess/var/log/*.log:Nosuchfileordirectory
第三种:执行命令,并区分stdout和stderr
上面的写法,无法实现区分标准输出和标准错误,只要换成下面种写法,就可以实现。
packagemain import( "bytes" "fmt" "log" "os/exec" ) funcmain(){ cmd:=exec.Command("ls","-l","/var/log/*.log") varstdout,stderrbytes.Buffer cmd.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) iferr!=nil{ log.Fatalf("cmd.Run()failedwith%s\n",err) } }
输出如下,可以看到前面的报错内容被归入到标准错误里
$gorundemo.go out: err: ls:cannotaccess/var/log/*.log:Nosuchfileordirectory 2020/11/1119:59:31cmd.Run()failedwithexitstatus2 exitstatus1
第四种:多条命令组合,请使用管道
将上一条命令的执行输出结果,做为下一条命令的参数。在Shell中可以使用管道符|来实现。
比如下面这条命令,统计了message日志中ERROR日志的数量。
$grepERROR/var/log/messages|wc-l 19
类似的,在Golang中也有类似的实现。
packagemain import( "os" "os/exec" ) funcmain(){ 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() }
输出如下
$gorundemo.go 19
第五种:设置命令级别的环境变量
使用os库的Setenv函数来设置的环境变量,是作用于整个进程的生命周期的。
packagemain import( "fmt" "log" "os" "os/exec" ) funcmain(){ os.Setenv("NAME","wangbm") cmd:=exec.Command("echo",os.ExpandEnv("$NAME")) out,err:=cmd.CombinedOutput() iferr!=nil{ log.Fatalf("cmd.Run()failedwith%s\n",err) } fmt.Printf("%s",out) }
只要在这个进程里,NAME这个变量的值都会是wangbm,无论你执行多少次命令
$gorundemo.go wangbm
如果想把环境变量的作用范围再缩小到命令级别,也是有办法的。
为了方便验证,我新建个sh脚本,内容如下
$cat/home/wangbm/demo.sh echo$NAME $bash/home/wangbm/demo.sh#由于全局环境变量中没有NAME,所以无输出
另外,demo.go里的代码如下
packagemain import( "fmt" "os" "os/exec" ) funcChangeYourCmdEnvironment(cmd*exec.Cmd)error{ env:=os.Environ() cmdEnv:=[]string{} for_,e:=rangeenv{ cmdEnv=append(cmdEnv,e) } cmdEnv=append(cmdEnv,"NAME=wangbm") cmd.Env=cmdEnv returnnil } funcmain(){ cmd1:=exec.Command("bash","/home/wangbm/demo.sh") ChangeYourCmdEnvironment(cmd1)//添加环境变量到cmd1命令:NAME=wangbm out1,_:=cmd1.CombinedOutput() fmt.Printf("output:%s",out1) cmd2:=exec.Command("bash","/home/wangbm/demo.sh") out2,_:=cmd2.CombinedOutput() fmt.Printf("output:%s",out2) }
执行后,可以看到第二次执行的命令,是没有输出NAME的变量值。
$gorundemo.go output:wangbm output:
到此这篇关于Go语言中用os/exec执行命令的五种方法的文章就介绍到这了,更多相关Go语言执行命令内容请搜索毛票票以前的文章或继续浏览下面的相关文章希望大家以后多多支持毛票票!