溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
這篇文章給大家介紹os/exec 如何在Go語言中執行命令，內容非常詳細，感興趣的小伙伴們可以參考借鑒，希望對大家能有所幫助。
在 Golang 中用于執行命令的庫是 os/exec，exec.Command 函數返回一個 Cmd 對象，根據不同的需求，可以將命令的執行分為三種情況
第一種：只執行命令，不獲取結果
直接調用 Cmd 對象的 Run 函數，返回的只有成功和失敗，獲取不到任何輸出的結果。
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)
}
}第二種：執行命令，并獲取結果
有時候我們執行一個命令就是想要獲取輸出結果，此時你可以調用 Cmd 的 CombinedOutput 函數。
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 函數，只返回 out，并不區分 stdout 和 stderr。如果你想區分他們，可以直接看第三種方法。
$ 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
不過在那之前，我卻發現一個小問題：有時候，shell 命令能執行，并不代碼 exec 也能執行。
比如我只想查看 /var/log/ 目錄下的 log 后綴名的文件呢？有點 Linux 基礎的同學，都會用這個命令
$ 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
按照這個寫法將它放入到 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))
}什么情況？居然不行，報錯了。
$ 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
為什么會報錯呢？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 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 // 標準輸出
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 directory 2020/11/11 19:59:31 cmd.Run() failed with exit status 2 exit status 1
第四種：多條命令組合，請使用管道
將上一條命令的執行輸出結果，做為下一條命令的參數。在 Shell 中可以使用管道符 | 來實現。
比如下面這條命令，統計了 message 日志中 ERROR 日志的數量。
$ grep ERROR /var/log/messages | wc -l 19
類似的，在 Golang 中也有類似的實現。
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
第五種：設置命令級別的環境變量
使用 os 庫的 Setenv 函數來設置的環境變量，是作用于整個進程的生命周期的。
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)
}只要在這個進程里，NAME 這個變量的值都會是 wangbm，無論你執行多少次命令
$ go run demo.go wangbm
如果想把環境變量的作用范圍再縮小到命令級別，也是有辦法的。
為了方便驗證，我新建個 sh 腳本，內容如下
$ cat /home/wangbm/demo.sh echo $NAME $ bash /home/wangbm/demo.sh # 由于全局環境變量中沒有 NAME，所以無輸出
另外，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) // 添加環境變量到 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 的變量值。
$ go run demo.go output: wangbm output:
關于os/exec 如何在Go語言中執行命令就分享到這里了，希望以上內容可以對大家有一定的幫助，可以學到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
