golang执行命令 && 实时获取输出结果

背景

• golang可以获取命令执行的输出结果，但要执行完才能够获取。
  
• 如果执行的命令是ssh，我们要实时获取，并执行相应的操作呢？
  

示例

1. func main() {
2.         user := "root"
3.         host := "172.16.116.133"
5.         //获取执行命令
6.         cmd := exec.Command("ssh", fmt.Sprintf("%s@%s", user, host))
7.         cmd.Stdin = os.Stdin
9.         var wg sync.WaitGroup
10.         wg.Add(2)
11.         //捕获标准输出
12.         stdout, err := cmd.StdoutPipe()
13.         if err != nil {
14.                 fmt.Println("ERROR:", err)
15.                 os.Exit(1)
16.         }
17.         readout := bufio.NewReader(stdout)
18.         go func() {
19.                 defer wg.Done()
20.                 GetOutput(readout)
21.         }()
23.         //捕获标准错误
24.         stderr, err := cmd.StderrPipe()
25.         if err != nil {
26.                 fmt.Println("ERROR:", err)
27.                 os.Exit(1)
28.         }
29.         readerr := bufio.NewReader(stderr)
30.         go func() {
31.                 defer wg.Done()
32.                 GetOutput(readerr)
33.         }()
35.         //执行命令
36.         cmd.Run()
37.         wg.Wait()
38.         return
39. }
40. func GetOutput(reader *bufio.Reader) {
41.         var sumOutput string                                //统计屏幕的全部输出内容
42.         outputBytes := make([]byte, 200)
43.         for {
44.                 n, err := reader.Read(outputBytes)                //获取屏幕的实时输出(并不是按照回车分割，所以要结合sumOutput)
45.                 if err != nil {
46.                         if err == io.EOF {
47.                                 break
48.                         }
49.                         fmt.Println(err)
50.                         sumOutput += err.Error()
51.                 }
52.                 output := string(outputBytes[:n])
53.                 fmt.Print(output) //输出屏幕内容
54.                 sumOutput += output
55.         }
56.         return
57. }

复制代码

应用场景

ssh是交互式命令，本示例实现了实时获取输出结果，并判断输出结果中有没有报错，报错则重试(再次登陆)。
场景：本Demo只是把"错误"二字视为异常，然后重试，实际上比这复杂的多，比如ssh连接超时重试等，这个逻辑请自行补充。
[code]package mainimport (        "bufio"        "fmt"        "io"        "os"        "os/exec"        "strings"        "sync"        "time")func main(){        retryTimes := 3        var retryInterval time.Duration = 3        user := "root"        host := "172.16.116.133"        //部分场景下重试登录        shouldRetry := true        for i:=1;i