詳解Go語(yǔ)言中用 os/exec 執(zhí)行命令的五種方法
完整系列教程詳見(jiàn)：http://golang.iswbm.com
在 Golang 中用于執(zhí)行命令的庫(kù)是 os/exec，exec.Command 函數(shù)返回一個(gè) Cmd 對(duì)象，根據(jù)不同的需求，可以將命令的執(zhí)行分為三種情況
- 只執(zhí)行命令，不獲取結(jié)果
- 執(zhí)行命令，并獲取結(jié)果（不區(qū)分 stdout 和 stderr）
- 執(zhí)行命令，并獲取結(jié)果（區(qū)分 stdout 和 stderr）
第一種：只執(zhí)行命令，不獲取結(jié)果
直接調(diào)用 Cmd 對(duì)象的 Run 函數(shù)，返回的只有成功和失敗，獲取不到任何輸出的結(jié)果。
package main
import (
"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)
}
}
第二種：執(zhí)行命令，并獲取結(jié)果
有時(shí)候我們執(zhí)行一個(gè)命令就是想要獲取輸出結(jié)果，此時(shí)你可以調(diào)用 Cmd 的 CombinedOutput 函數(shù)。
package main
import (
"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 函數(shù)，只返回 out，并不區(qū)分 stdout 和 stderr。如果你想?yún)^(qū)分他們，可以直接看第三種方法。
$ go run demo.go combined out: total 11540876 -rw-r--r-- 2 root root 4096 Oct 29 2018 yum.log drwx------ 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
不過(guò)在那之前，我卻發(fā)現(xiàn)一個(gè)小問(wèn)題：有時(shí)候，shell 命令能執(zhí)行，并不代碼 exec 也能執(zhí)行。
比如我只想查看 /var/log/ 目錄下的 log 后綴名的文件呢？有點(diǎn) Linux 基礎(chǔ)的同學(xué)，都會(huì)用這個(gè)命令
$ ls -l /var/log/*.log total 11540 -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
按照這個(gè)寫(xiě)法將它放入到 exec.Command
package main
import (
"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))
}
什么情況？居然不行，報(bào)錯(cuò)了。
$ go run demo.go combined out: ls: cannot access /var/log/*.log: No such file or directory 2020/11/11 19:46:00 cmd.Run() failed with exit status 2 exit status 1
為什么會(huì)報(bào)錯(cuò)呢？shell 明明沒(méi)有問(wèn)題啊
其實(shí)很簡(jiǎn)單，原來(lái) ls -l /var/log/*.log 并不等價(jià)于下面這段代碼。
exec.Command("ls", "-l", "/var/log/*.log")
上面這段代碼對(duì)應(yīng)的 Shell 命令應(yīng)該是下面這樣，如果你這樣子寫(xiě)，ls 就會(huì)把參數(shù)里的內(nèi)容當(dāng)成具體的文件名，而忽略通配符 *
$ ls -l "/var/log/*.log" ls: cannot access /var/log/*.log: No such file or directory
第三種：執(zhí)行命令，并區(qū)分stdout 和 stderr
上面的寫(xiě)法，無(wú)法實(shí)現(xiàn)區(qū)分標(biāo)準(zhǔn)輸出和標(biāo)準(zhǔn)錯(cuò)誤，只要換成下面種寫(xiě)法，就可以實(shí)現(xiàn)。
package main
import (
"bytes"
"fmt"
"log"
"os/exec"
)
func main() {
cmd := exec.Command("ls", "-l", "/var/log/*.log")
var stdout, stderr bytes.Buffer
cmd.Stdout = &stdout // 標(biāo)準(zhǔn)輸出
cmd.Stderr = &stderr // 標(biāo)準(zhǔn)錯(cuò)誤
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)
}
}
輸出如下，可以看到前面的報(bào)錯(cuò)內(nèi)容被歸入到標(biāo)準(zhǔn)錯(cuò)誤里
$ go run demo.go out: err: ls: cannot access /var/log/*.log: No such file or directory 2020/11/11 19:59:31 cmd.Run() failed with exit status 2 exit status 1
第四種：多條命令組合，請(qǐng)使用管道
將上一條命令的執(zhí)行輸出結(jié)果，做為下一條命令的參數(shù)。在 Shell 中可以使用管道符 | 來(lái)實(shí)現(xiàn)。
比如下面這條命令，統(tǒng)計(jì)了 message 日志中 ERROR 日志的數(shù)量。
$ grep ERROR /var/log/messages | wc -l 19
類(lèi)似的，在 Golang 中也有類(lèi)似的實(shí)現(xiàn)。
package main
import (
"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
第五種：設(shè)置命令級(jí)別的環(huán)境變量
使用 os 庫(kù)的 Setenv 函數(shù)來(lái)設(shè)置的環(huán)境變量，是作用于整個(gè)進(jìn)程的生命周期的。
package main
import (
"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 %s\n", err)
}
fmt.Printf("%s", out)
}
只要在這個(gè)進(jìn)程里，NAME 這個(gè)變量的值都會(huì)是 wangbm，無(wú)論你執(zhí)行多少次命令
$ go run demo.go wangbm
如果想把環(huán)境變量的作用范圍再縮小到命令級(jí)別，也是有辦法的。
為了方便驗(yàn)證，我新建個(gè) sh 腳本，內(nèi)容如下
$ cat /home/wangbm/demo.sh echo $NAME $ bash /home/wangbm/demo.sh # 由于全局環(huán)境變量中沒(méi)有 NAME，所以無(wú)輸出
另外，demo.go 里的代碼如下
package main
import (
"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 nil
}
func main() {
cmd1 := exec.Command("bash", "/home/wangbm/demo.sh")
ChangeYourCmdEnvironment(cmd1) // 添加環(huán)境變量到 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)
}
執(zhí)行后，可以看到第二次執(zhí)行的命令，是沒(méi)有輸出 NAME 的變量值。
$ go run demo.go output: wangbm output:
到此這篇關(guān)于Go語(yǔ)言中用 os/exec 執(zhí)行命令的五種方法的文章就介紹到這了,更多相關(guān)Go語(yǔ)言執(zhí)行命令內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
go語(yǔ)言數(shù)組及結(jié)構(gòu)體繼承和初始化示例解析
這篇文章主要為大家介紹了go語(yǔ)言數(shù)組及結(jié)構(gòu)體繼承和初始化示例解析，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步早日升職加薪2022-04-04
Golang實(shí)現(xiàn)http重定向https
這篇文章介紹了Golang實(shí)現(xiàn)http重定向https的方法，文中通過(guò)示例代碼介紹的非常詳細(xì)。對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2022-07-07
本文主要介紹了詳解Golang五種原子性操作的用法，文中通過(guò)示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2021-09-09
Golang中切片長(zhǎng)度和容量的區(qū)別示例詳解
切片長(zhǎng)度與容量在Go中很常見(jiàn),切片長(zhǎng)度是切片中可用元素的數(shù)量,而切片容量是從切片中第一個(gè)元素開(kāi)始計(jì)算的底層數(shù)組中的元素?cái)?shù)量,這篇文章主要給大家介紹了關(guān)于Golang中切片長(zhǎng)度和容量區(qū)別的相關(guān)資料,需要的朋友可以參考下2024-01-01
這篇文章主要為大家介紹了Go中的新增對(duì)模糊測(cè)試的支持，文中還包含了一些功能實(shí)驗(yàn)性測(cè)試分析有需要的朋友可以借鑒參考下，希望能夠有所幫助2022-03-03
在?Go?語(yǔ)言中,切片拼接是一項(xiàng)常見(jiàn)的操作,但如果處理不當(dāng),可能會(huì)導(dǎo)致性能問(wèn)題或意外的副作用,本文將詳細(xì)介紹幾種高效的切片拼接方法,希望對(duì)大家有所幫助2024-01-01
使用Go實(shí)現(xiàn)偽靜態(tài)URL重寫(xiě)功能
在Web開(kāi)發(fā)中,偽靜態(tài)URL已成為優(yōu)化網(wǎng)站架構(gòu)和提升SEO的常用技術(shù)手段,偽靜態(tài)URL是一種介于動(dòng)態(tài)URL和靜態(tài)URL之間的解決方案,本文給大家介紹了如何使用Go實(shí)現(xiàn)偽靜態(tài)URL重寫(xiě)功能,需要的朋友可以參考下2024-08-08
GoFrame?gtree樹(shù)形結(jié)構(gòu)的使用技巧示例
這篇文章主要為大家介紹了GoFrame?gtree樹(shù)形結(jié)構(gòu)的使用技巧示例，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-06-06
Go語(yǔ)言編譯程序從后臺(tái)運(yùn)行,不出現(xiàn)dos窗口的操作
這篇文章主要介紹了Go語(yǔ)言編譯程序從后臺(tái)運(yùn)行,不出現(xiàn)dos窗口的操作，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2021-04-04
本文給大家介紹的這里是GO程序員的五個(gè)進(jìn)化階段，從最開(kāi)始的菜逼到最終的布道者，附上各種示例，一步步走向大神之路，推薦給小伙伴們，有需要的朋友可以參考下2015-03-03