Go語言入門exec的基本使用示例

Exec 是 os 包中的一個子包,它可用於使用 Go 運行外部命令。Go exec 命令教程展示瞭如何在 Golang 中執行 shell 命令和程序。

要使用這個包,我們需要按如下方式導入:

import "os/exec"

使用 GoLang exec 包運行命令

我們可以運行任何我們希望的命令。就像我們使用 CMD、bash 或其他一些 shell 來運行命令一樣,它可以運行這些命令。

這是運行 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 函數啟動指定命令並等待它完成,而 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

現在,程序將運行,但我們不會看到控制臺的任何輸出。原因是命令運行,輸出沒有發送到標準輸出。

$ go run main.go

所以,我們需要修復它。添加下面顯示的兩行以查看控制臺的任何輸出。

cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr

輸出將顯示當前目錄中的文件。

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)
  }
}

然後我們再程序,可以看到標準臺輸出如下的文件:

$ 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 直接運行該命令,可以看到結果正確:

$ 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

為不同的操作系統指定命令

我們可以指定針對不同操作系統運行不同的命令(例如 Linux 上的 bash 命令)。這是一個例子。

if runtime.GOOS == "linux" {
    cmd = exec.Command("ls")
}

為此,我們還需要導入運行時包。

要查看所有可能的操作系統,我們可以運行 go tool dist list ,它將顯示所有可能的操作系統和 ARCH 組合。

Go exec 命令捕獲輸出

輸出運行命令並返回其標準輸出:

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))
}

運行該程序:

$ 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

Go cmd.StdinPipe

管道允許我們將一個命令的輸出發送到另一個命令。 StdinPipe 返回一個管道,該管道將在命令啟動時連接到命令的標準輸入。

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 內的標準輸入。

cmd := exec.Command("cat")

cat 命令將給定的文件連接到標準輸出。當沒有給定文件或帶有 – 時,該命令讀取標準輸入並將其打印到標準輸出。

stdin, err := cmd.StdinPipe()

我們得到 cat 命令的標準輸入管道。

go func() {
    defer stdin.Close()
    io.WriteString(stdin, "an old falcon")
}()

在 goroutine 內部,我們將一個字符串寫入標準輸入管道。

$ go run stdinpipe.go 
an old falcon

Go cmd.StdoutPipe

StdoutPipe 返回一個管道,該管道將在命令啟動時連接到命令的標準輸出。

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 命令的輸出並將其轉換為大寫字母。

cmd := exec.Command("echo", "an old falcon")

要運行的命令是帶有單個字符串參數的 echo 命令。

stdout, err := cmd.StdoutPipe()

我們得到標準輸出管道。

if err := cmd.Start(); err != nil {
    log.Fatal(err)
}

該命令使用 Start 函數執行;它不會等待它完成。

data, err := ioutil.ReadAll(stdout)

我們從管道中讀取數據。

if err := cmd.Wait(); err != nil {
    log.Fatal(err)
}

Wait 等待命令退出並等待任何復制到 stdin 或從 stdout 或 stderr 復制完成。它在看到命令退出後關閉管道。

運行該程序:

$ go run stdoutpipe.go 
AN OLD FALCON

總結

os/exec 包運行外部命令。它包裝瞭 os.StartProcess 以便更輕松地重新映射標準輸入和標準輸出、將 I/O 與管道連接以及進行其他調整。

參考鏈接:

https://zetcode.com/golang/exec-command/

以上就是Go語言入門exec的基本使用示例的詳細內容,更多關於Go語言exec用例的資料請關註WalkonNet其它相關文章!

推薦閱讀: