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

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

func main() {
	user := "root"
	host := "172.16.116.133"

	//获取执行命令
	cmd := exec.Command("ssh", fmt.Sprintf("%s@%s", user, host))
	cmd.Stdin = os.Stdin

	var wg sync.WaitGroup
	wg.Add(2)
	//捕获标准输出
	stdout, err := cmd.StdoutPipe()
	if err != nil {
		fmt.Println("ERROR:", err)
		os.Exit(1)
	}
	readout := bufio.NewReader(stdout)
	go func() {
		defer wg.Done()
		GetOutput(readout)
	}()

	//捕获标准错误
	stderr, err := cmd.StderrPipe()
	if err != nil {
		fmt.Println("ERROR:", err)
		os.Exit(1)
	}
	readerr := bufio.NewReader(stderr)
	go func() {
		defer wg.Done()
		GetOutput(readerr)
	}()

	//执行命令
	cmd.Run()
	wg.Wait()
	return
}
func GetOutput(reader *bufio.Reader) {
	var sumOutput string				//统计屏幕的全部输出内容
	outputBytes := make([]byte, 200)
	for {
		n, err := reader.Read(outputBytes)		//获取屏幕的实时输出(并不是按照回车分割，所以要结合sumOutput)
		if err != nil {
			if err == io.EOF {
				break
			}
			fmt.Println(err)
			sumOutput += err.Error()
		}
		output := string(outputBytes[:n])
		fmt.Print(output) //输出屏幕内容
		sumOutput += output
	}
	return
}

应用场景

ssh是交互式命令，本示例实现了实时获取输出结果，并判断输出结果中有没有报错，报错则重试(再次登陆)。
场景：本Demo只是把"错误"二字视为异常，然后重试，实际上比这复杂的多，比如ssh连接超时重试等，这个逻辑请自行补充。

package main

import (
	"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<=retryTimes && shouldRetry;i++{
		//执行命令
		shouldRetry = RunSSHCommand(user,host)
		if !shouldRetry{
			return
		}
		time.Sleep(retryInterval * time.Second)
	}
	if shouldRetry{
		fmt.Println("\n失败,请重试或检查")
	}
}
func shouldRetryByOutput(output string)bool{
	if strings.Contains(output,"错误"){		//匹配到"错误"就重试.这里只是Demo，请根据实际情况设置。
		return true
	}
	return false
}
func GetAndFilterOutput(reader *bufio.Reader)(shouldRetry bool){
	var sumOutput string
	outputBytes:= make([]byte,200)
	for {
		n,err := reader.Read(outputBytes)
		if err!=nil{
			if err == io.EOF{
				break
			}
			fmt.Println(err)
			sumOutput += err.Error()
		}
		output := string(outputBytes[:n])
		fmt.Print(output)		//输出屏幕内容
		sumOutput += output
		if shouldRetryByOutput(output){
			shouldRetry = true
		}
	}
	if shouldRetryByOutput(sumOutput){
		shouldRetry = true
	}
	return
}
func RunSSHCommand(user,host string)(shouldRetry bool){
	//获取执行命令
	cmd := exec.Command("ssh",fmt.Sprintf("%s@%s",user,host))
	cmd.Stdin = os.Stdin

	var wg sync.WaitGroup
	wg.Add(2)
	//捕获标准输出
	stdout, err := cmd.StdoutPipe()
	if err != nil {
		fmt.Println("ERROR:",err)
		os.Exit(1)
	}
	readout := bufio.NewReader(stdout)
	go func() {
		defer wg.Done()
		shouldRetryTemp := GetAndFilterOutput(readout)
		if shouldRetryTemp{
			shouldRetry = true
		}
	}()

	//捕获标准错误
	stderr, err := cmd.StderrPipe()
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
	readerr := bufio.NewReader(stderr)
	go func() {
		defer wg.Done()
		shouldRetryTemp := GetAndFilterOutput(readerr)
		if shouldRetryTemp{
			shouldRetry = true
		}
	}()

	//执行命令
	cmd.Run()
	wg.Wait()
	return
}

那年，郭少在京城。