exec 是 os 包中的一個子包,它可用于使用 Go 運(yùn)行外部命令。Go exec 命令教程展示了如何在 Golang 中執(zhí)行 shell 命令和程序。
要使用這個包,我們需要按如下方式導(dǎo)入:
1.使用 GoLang exec 包運(yùn)行命令
我們可以運(yùn)行任何我們希望的命令。就像我們使用 CMD、bash 或其他一些 shell 來運(yùn)行命令一樣,它可以運(yùn)行這些命令。
這是運(yùn)行 ls? 命令的示例。新建一個 main.go :
package main
import (
"fmt"
"os/exec"
)
func main() {
cmd := exec.Command("ls")
e := cmd.Run()
CheckError(e)
}
func CheckError(e error) {
if e != nil {
fmt.Println(e)
}
}
Run? 函數(shù)啟動指定命令并等待它完成,而 Start 啟動指定命令但不等待它完成;我們需要使用Wait with Start。
然后新建一個 go.mod 文件:
$ go mod init main.go
go: creating new go.mod: module main.go
go: to add module requirements and sums:
go mod tidy
現(xiàn)在,程序?qū)⑦\(yùn)行,但我們不會看到控制臺的任何輸出。原因是命令運(yùn)行,輸出沒有發(fā)送到標(biāo)準(zhǔn)輸出。
所以,我們需要修復(fù)它。添加下面顯示的兩行以查看控制臺的任何輸出。
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
輸出將顯示當(dāng)前目錄中的文件。
package main
import (
"fmt"
"os"
"os/exec"
)
func main() {
cmd := exec.Command("ls", "-lah")
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
e := cmd.Run()
CheckError(e)
}
func CheckError(e error) {
if e != nil {
fmt.Println(e)
}
}
然后我們再程序,可以看到標(biāo)準(zhǔn)臺輸出如下的文件:
$ go run main.go
total 16
drwxr-xr-x 4 yuzhou_1su staff 128B 5 15 22:56 .
drwxr-xr-x 23 yuzhou_1su staff 736B 5 15 22:53 ..
-rw-r--r-- 1 yuzhou_1su staff 24B 5 15 22:56 go.mod
-rw-r--r-- 1 yuzhou_1su staff 248B 5 15 23:18 main.go
利用直接 ls 直接運(yùn)行該命令,可以看到結(jié)果正確:
$ ls -alh
total 16
drwxr-xr-x 4 yuzhou_1su staff 128B 5 15 22:56 .
drwxr-xr-x 23 yuzhou_1su staff 736B 5 15 22:53 ..
-rw-r--r-- 1 yuzhou_1su staff 24B 5 15 22:56 go.mod
-rw-r--r-- 1 yuzhou_1su staff 248B 5 15 23:18 main.go
2.為不同的操作系統(tǒng)指定命令
我們可以指定針對不同操作系統(tǒng)運(yùn)行不同的命令(例如 Linux 上的 bash 命令)。這是一個例子。
if runtime.GOOS == "linux" {
cmd = exec.Command("ls")
}
為此,我們還需要導(dǎo)入運(yùn)行時包。
要查看所有可能的操作系統(tǒng),我們可以運(yùn)行 go tool dist list ,它將顯示所有可能的操作系統(tǒng)和 ARCH 組合。
3.Go exec 命令捕獲輸出
輸出運(yùn)行命令并返回其標(biāo)準(zhǔn)輸出:
package main
import (
"fmt"
"log"
"os/exec"
)
func main() {
out, err := exec.Command("ls", "-l").Output()
if err != nil {
log.Fatal(err)
}
fmt.Println(string(out))
}
運(yùn)行該程序:
$ go run main.go
total 16
-rw-r--r-- 1 yuzhou_1su staff 24 5 15 22:56 go.mod
-rw-r--r-- 1 yuzhou_1su staff 180 5 15 23:33 main.go
4.Go cmd.StdinPipe
管道允許我們將一個命令的輸出發(fā)送到另一個命令。 StdinPipe 返回一個管道,該管道將在命令啟動時連接到命令的標(biāo)準(zhǔn)輸入。
package main
import (
"fmt"
"io"
"log"
"os/exec"
)
func main() {
cmd := exec.Command("cat")
stdin, err := cmd.StdinPipe()
if err != nil {
log.Fatal(err)
}
go func() {
defer stdin.Close()
io.WriteString(stdin, "an old falcon")
}()
out, err := cmd.CombinedOutput()
if err != nil {
log.Fatal(err)
}
fmt.Printf("%s\n", out)
}
在代碼示例中,我們將字符串寫入 goroutine 內(nèi)的標(biāo)準(zhǔn)輸入。
cmd := exec.Command("cat")
cat 命令將給定的文件連接到標(biāo)準(zhǔn)輸出。當(dāng)沒有給定文件或帶有 - 時,該命令讀取標(biāo)準(zhǔn)輸入并將其打印到標(biāo)準(zhǔn)輸出。
stdin, err := cmd.StdinPipe()
我們得到 cat 命令的標(biāo)準(zhǔn)輸入管道。
go func() {
defer stdin.Close()
io.WriteString(stdin, "an old falcon")
}()
在 goroutine 內(nèi)部,我們將一個字符串寫入標(biāo)準(zhǔn)輸入管道。
$ go run stdinpipe.go
an old falcon
5.Go cmd.StdoutPipe
StdoutPipe 返回一個管道,該管道將在命令啟動時連接到命令的標(biāo)準(zhǔn)輸出。
package main
import (
"fmt"
"io/ioutil"
"log"
"os/exec"
"strings"
)
func upper(data string) string {
return strings.ToUpper(data)
}
func main() {
cmd := exec.Command("echo", "an old falcon")
stdout, err := cmd.StdoutPipe()
if err != nil {
log.Fatal(err)
}
if err := cmd.Start(); err != nil {
log.Fatal(err)
}
data, err := ioutil.ReadAll(stdout)
if err != nil {
log.Fatal(err)
}
if err := cmd.Wait(); err != nil {
log.Fatal(err)
}
fmt.Printf("%s\n", upper(string(data)))
}
該示例通過管道讀取 echo 命令的輸出并將其轉(zhuǎn)換為大寫字母。
cmd := exec.Command("echo", "an old falcon")
要運(yùn)行的命令是帶有單個字符串參數(shù)的 echo 命令。
stdout, err := cmd.StdoutPipe()
我們得到標(biāo)準(zhǔn)輸出管道。
if err := cmd.Start(); err != nil {
log.Fatal(err)
}
該命令使用 Start 函數(shù)執(zhí)行;它不會等待它完成。
data, err := ioutil.ReadAll(stdout)
我們從管道中讀取數(shù)據(jù)。
if err := cmd.Wait(); err != nil {
log.Fatal(err)
}
Wait 等待命令退出并等待任何復(fù)制到 stdin 或從 stdout 或 stderr 復(fù)制完成。它在看到命令退出后關(guān)閉管道。
運(yùn)行該程序:
$ go run stdoutpipe.go
AN OLD FALCON
6.總結(jié)
os/exec? 包運(yùn)行外部命令。它包裝了 os.StartProcess 以便更輕松地重新映射標(biāo)準(zhǔn)輸入和標(biāo)準(zhǔn)輸出、將 I/O 與管道連接以及進(jìn)行其他調(diào)整。
參考鏈接:https://zetcode.com/golang/exec-command/