golang os/exec 执行终端命令

1. 只执行命令，不获取结果

直接调用 Cmd 对象的 Run 函数，返回的只有成功和失败，获取不到任何输出的结果

package main

import (
    "log"
    "os/exec"
)

func main() {
    cmd := exec.Command("ls", "-l", "/root")
    err := cmd.Run()
    if err != nil {
        log.Fatalf("cmd.Run() failed with %s\n", err)
    }
}

2. 执行命令，并获取结果

  Cmd 的 CombinedOutput 函数执行终端命令并返回执行结果

package main

import (
"log"
"os/exec"
)

func main() {
    cmd := exec.Command("ls", "-l", "/root")
    out, err := cmd.CombinedOutput()
    if err != nil {
        log.Fatalf("cmd.Run() failed with %s\n", err)
    }
    log.Printf("combined out:\n%s\n", string(out))
}

CombinedOutput 函数，只返回 out，并不区分 stdout 和 stderr

3. 执行命令，并区分stdout 和 stderr

package main

import (
  "bytes"
  "log"
  "os/exec"
)

func main() {
  cmd := exec.Command("/bin/bash", "-c", "nginx -t")
  var stdout, stderr bytes.Buffer
  cmd.Stdout = &stdout // 标准输出
  cmd.Stderr = &stderr // 标准错误
  err := cmd.Run()
  if len(stderr.Bytes()) > 0 {
    log.Printf("stderr:%s\n", string(stderr.Bytes()))
  }
  if len(stdout.Bytes()) > 0 {
    log.Printf("stdout:%s\n", string(stdout.Bytes()))
  }

  if err != nil {
    log.Fatalf("cmd.Run() failed with %s\n", err)
  }
}

4. 多条命令组合，可以使用管道

在终端中经常使用管道

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

5. 设置命令级别的环境变量

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

感谢：https://zhuanlan.zhihu.com/p/296409942