快捷導(dǎo)航

Go語言調(diào)用Shell與可執(zhí)行文件的實(shí)現(xiàn)

更新時(shí)間：2020年10月16日 15:07:09 作者：莫問君心

這篇文章主要介紹了Go語言調(diào)用Shell與可執(zhí)行文件的實(shí)現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

os/exec包可用于調(diào)用外部命令，可以使用管道連接輸入輸出，并支持阻塞與非阻塞方式執(zhí)行命令。

os/exec包中關(guān)鍵的類型為Cmd，以下介紹的所有方法皆服務(wù)于該類型：

func Command(name string, arg ...string) *Cmd
方法返回一個(gè)*Cmd，用于執(zhí)行name指定的程序(攜帶arg參數(shù))

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

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

func (c *Cmd) Wait() error
該方法會阻塞直到Cmd中的命令執(zhí)行完成，但該命令必須是被Start方法開始執(zhí)行的

func (c *Cmd) Output() ([]byte, error)
執(zhí)行Cmd中包含的命令，并返回標(biāo)準(zhǔn)輸出的切片

func (c *Cmd) CombinedOutput() ([]byte, error)
執(zhí)行Cmd中包含的命令，并返回標(biāo)準(zhǔn)輸出與標(biāo)準(zhǔn)錯(cuò)誤合并后的切片

func (c *Cmd) StdinPipe() (io.WriteCloser, error)
返回一個(gè)管道，該管道會在Cmd中的命令被啟動(dòng)后連接到其標(biāo)準(zhǔn)輸入

func (c *Cmd) StdoutPipe() (io.ReadCloser, error)
返回一個(gè)管道，該管道會在Cmd中的命令被啟動(dòng)后連接到其標(biāo)準(zhǔn)輸出

func (c *Cmd) StderrPipe() (io.ReadCloser, error)
返回一個(gè)管道，該管道會在Cmd中的命令被啟動(dòng)后連接到其標(biāo)準(zhǔn)錯(cuò)誤

普通調(diào)用示例：

調(diào)用Shell命令或可執(zhí)行文件

演示在當(dāng)前目錄創(chuàng)建一個(gè)空文件

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

一般不建議使用這種默認(rèn)方式調(diào)用Shell腳本：

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

因?yàn)檫@種方式實(shí)際的執(zhí)行結(jié)果和命令行執(zhí)行#sh my_shell.sh一樣，如果你的Shell腳本不滿足sh的規(guī)范，就會調(diào)用失敗。

調(diào)用Shell腳本

設(shè)置bash來調(diào)用指定Shell腳本，dir_size.sh為我們測試用的Shell腳本。調(diào)用完成后打印Shell腳本的標(biāo)準(zhǔn)輸出到控制臺。

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", command, string(output))
}

dir_size.sh示例文件內(nèi)容如下，用于輸出當(dāng)前目錄的大?。?/p>

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

Go程序運(yùn)行結(jié)果：

[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命令的標(biāo)準(zhǔn)輸入，過濾包含test的字符串，并使用管道連接標(biāo)準(zhǔn)輸出，打印運(yùn)行結(jié)果：

package main

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

func main(){
  cmd := exec.Command("/bin/bash", "-c", "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程序運(yùn)行結(jié)果：