1. 程式人生 > 程式設計 >Go語言呼叫Shell與可執行檔案的實現

Go語言呼叫Shell與可執行檔案的實現

os/exec包可用於呼叫外部命令,可以使用管道連線輸入輸出,並支援阻塞與非阻塞方式執行命令。

os/exec包中關鍵的型別為Cmd,以下介紹的所有方法皆服務於該型別:

func Command(name string,arg ...string) *Cmd
方法返回一個*Cmd, 用於執行name指定的程式(攜帶arg引數)

func (c *Cmd) Run() error
執行Cmd中包含的命令,阻塞直到命令執行完成

func (c *Cmd) Start() error
執行Cmd中包含的命令,該方法立即返回,並不等待命令執行完成

func (c *Cmd) Wait() error

該方法會阻塞直到Cmd中的命令執行完成,但該命令必須是被Start方法開始執行的

func (c *Cmd) Output() ([]byte,error)
執行Cmd中包含的命令,並返回標準輸出的切片

func (c *Cmd) CombinedOutput() ([]byte,error)
執行Cmd中包含的命令,並返回標準輸出與標準錯誤合併後的切片

func (c *Cmd) StdinPipe() (io.WriteCloser,error)
返回一個管道,該管道會在Cmd中的命令被啟動後連線到其標準輸入

func (c *Cmd) StdoutPipe() (io.ReadCloser,error)

返回一個管道,該管道會在Cmd中的命令被啟動後連線到其標準輸出

func (c *Cmd) StderrPipe() (io.ReadCloser,error)
返回一個管道,該管道會在Cmd中的命令被啟動後連線到其標準錯誤

普通呼叫示例:

呼叫Shell命令或可執行檔案

演示在當前目錄建立一個空檔案

package main

import (
  "fmt"
  "os/exec"
)

func main(){
  cmd := exec.Command("touch","test_file")

  err := cmd.Run()
  if err != nil {
    fmt.Println("Execute Command failed:" + err.Error())
    return
  }

  fmt.Println("Execute Command finished.")
}

一般不建議使用這種預設方式呼叫Shell指令碼:

cmd := exec.Command("my_shell.sh")

因為這種方式實際的執行結果和命令列執行#sh my_shell.sh一樣,如果你的Shell指令碼不滿足sh的規範,就會呼叫失敗。

呼叫Shell指令碼

設定bash來呼叫指定Shell指令碼,dir_size.sh為我們測試用的Shell指令碼。呼叫完成後列印Shell指令碼的標準輸出到控制檯。

package main

import (
  "fmt"
  "os/exec"
)

func main(){
  command := `./dir_size.sh .`
  cmd := exec.Command("/bin/bash","-c",command)

  output,err := cmd.Output()
  if err != nil {
    fmt.Printf("Execute Shell:%s failed with error:%s",command,err.Error())
    return
  }
  fmt.Printf("Execute Shell:%s finished with output:\n%s",string(output))
}

dir_size.sh示例檔案內容如下,用於輸出當前目錄的大小:

#!/bin/bash
du -h --max-depth=1 $1

Go程式執行結果:

[root@localhost opt]# ll
total 2120
-rwx------. 1 root root   36 Jan 22 16:37 dir_size.sh
-rwx------. 1 root root 2152467 Jan 22 16:39 execCommand
drwxrwxr-x. 11 1000 1000  4096 Jul 12 2017 kibana
drwx------. 2 root root  4096 Jan 16 10:45 sftpuser
drwx------. 3 root root  4096 Jan 22 16:41 upload
[root@localhost opt]# ./execCommand 
Execute Shell:./dir_size.sh . finished with output:
4.0K  ./sftpuser
181M  ./kibana
1.1G  ./upload
1.2G  .

使用輸入輸出Pipe

演示使用管道連線到grep命令的標準輸入,過濾包含test的字串,並使用管道連線標準輸出,列印執行結果:

package main

import (
  "fmt"
  "io/ioutil"
  "os/exec"
)

func main(){
  cmd := exec.Command("/bin/bash","grep test")

  stdin,_ := cmd.StdinPipe()
  stdout,_ := cmd.StdoutPipe()

  if err := cmd.Start(); err != nil{
    fmt.Println("Execute failed when Start:" + err.Error())
    return
  }

  stdin.Write([]byte("go text for grep\n"))
  stdin.Write([]byte("go test text for grep\n"))
  stdin.Close()

  out_bytes,_ := ioutil.ReadAll(stdout)
  stdout.Close()

  if err := cmd.Wait(); err != nil {
    fmt.Println("Execute failed when Wait:" + err.Error())
    return
  }

  fmt.Println("Execute finished:" + string(out_bytes))
}

Go程式執行結果:

[root@localhost ~]# ./execCommand
Execute finished:go test text for grep

阻塞/非阻塞方式呼叫

文章開頭方法介紹處已經介紹的很清楚,且前面示例都有涉及,就不另行說明了。

參考文件:
GoLang標準庫文件

到此這篇關於Go語言呼叫Shell與可執行檔案的實現的文章就介紹到這了,更多相關Go語言呼叫Shell內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們!