Golang执行shell命令并通过channel返回输出及错误问题求助

原有代码存在的核心问题 #

• 调用cmd.Output()会阻塞等待命令执行完成，一次性返回所有标准输出，无法实现运行过程中实时返回输出，且该方法只会返回执行成功的stdout，错误场景下不会返回stderr内容。
• 错误分支里读取的是当前程序的os.Stdout，而不是执行shell命令的子进程的stdout、stderr，完全拿不到目标命令的输出内容。
• CommandResponse里的StdOut、StdErr通道没有在正常执行场景初始化，错误场景也没有初始化StdErr通道，调用方读取会阻塞甚至panic。
• read函数返回的是只读通道，原有代码里试图把只读通道赋值给output双向通道会报类型不匹配错误，同时错误分支里的goroutine写法错误，read已经返回通道了，不需要再套一层goroutine往另一个通道写。
• CommandRequest结构体定义了但完全没有用到，入参写死了ping命令，不符合通用执行shell命令的需求。

修复后的代码 #

package main

import (
	"bufio"
	"io"
	"os/exec"
)

type CommandRequest struct {
	Command string
	Args    []string // 调整为字符串切片，支持多参数场景
}

type CommandResponse struct {
	StdOut <-chan string
	StdErr <-chan string
}

func main() {
	// 示例调用
	req := CommandRequest{
		Command: "ping",
		Args:    []string{"192.168.0.1", "-c", "4"}, // 加-c 4避免无限ping
	}
	resp := ExecuteCommand(req)
	// 输出打印示例
	go func() {
		for line := range resp.StdOut {
			println("stdout:", line)
		}
	}()
	for line := range resp.StdErr {
		println("stderr:", line)
	}
}

func ExecuteCommand(req CommandRequest) CommandResponse {
	cmd := exec.Command(req.Command, req.Args...)
	// 分别获取子进程的stdout、stderr管道
	stdoutPipe, _ := cmd.StdoutPipe()
	stderrPipe, _ := cmd.StderrPipe()

	// 启动命令，无需等待执行完成即可读取输出
	_ = cmd.Start()

	// 分别读取两个管道的输出
	stdoutChan := read(stdoutPipe)
	stderrChan := read(stderrPipe)

	// 后台等待命令执行完成，回收子进程资源避免僵尸进程
	go func() {
		_ = cmd.Wait()
	}()

	return CommandResponse{
		StdOut: stdoutChan,
		StdErr: stderrChan,
	}
}

func read(r io.Reader) <-chan string {
	lines := make(chan string)
	go func() {
		defer close(lines)
		scan := bufio.NewScanner(r)
		for scan.Scan() {
			lines <- scan.Text()
		}
	}()
	return lines
}

核心修改点说明 #

• 用cmd.StdoutPipe()、cmd.StderrPipe()分别获取子进程的输出管道，调用cmd.Start()启动命令后就可以实时读取管道内容，不需要等待命令执行结束。
• 新增后台goroutine调用cmd.Wait()回收子进程资源，避免产生僵尸进程。
• 调整CommandRequest.Args类型为字符串切片，支持多参数传入的场景。
• 调整CommandResponse的通道类型为只读通道，符合通道使用的最佳实践，避免调用方误写入。

内容的提问来源于stack exchange，提问作者techipank