如何在Go中运行前台/后台Shell命令并获取输出

问题描述 #

需要在Go中实现类似Bash的功能：执行用户输入的Shell命令，前台命令（不带&）会阻塞Go程序，后台命令（带&）不阻塞，同时要捕获命令输出并添加前缀line of output: 。

原Bash实现：

#!/bin/bash
while read -p 'enter a command: ' cmd; do
    /bin/sh -c "$cmd" >(sed 's/^/line of output: /')
done

自行尝试的Go代码在执行后台命令（如{ sleep 5; echo hi; } &）时，会报错panic: read |0: file already closed，但前台命令可正常运行。

错误原因 #

当执行带&的后台命令时，/bin/sh -c会启动后台进程后立刻退出，此时Go通过StdoutPipe创建的管道写端会被关闭，而后台goroutine仍在尝试读取该管道，从而触发文件已关闭的错误。此外，后台进程的stdout会随父Shell进程的退出而失效，导致输出无法被捕获。

解决方案 #

方案1：包装Shell命令，模拟进程替换 #

模仿Bash的进程替换机制，将用户命令的输出重定向到cat进程，确保后台进程的输出能持续被转发，即使父Shell退出。同时改进错误处理，忽略管道关闭的正常错误，并判断命令是否为后台命令以决定是否阻塞。

修改后的Go代码：

package main

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

func main() {
	input := bufio.NewScanner(os.Stdin)
	for {
		print("enter a command: ")
		input.Scan()
		cmdStr := input.Text()

		// 包装命令，将输出重定向到cat进程，确保后台输出能被捕获
		// 若需要捕获stderr，添加2>&1：cmdStr + " > >(cat) 2>&1"
		wrappedCmd := cmdStr + " > >(cat)"
		cmd := exec.Command("/bin/sh", "-c", wrappedCmd)

		stdout, err := cmd.StdoutPipe()
		if err != nil {
			panic(err)
		}

		// 后台goroutine处理输出
		go func() {
			output := bufio.NewScanner(stdout)
			for output.Scan() {
				println("line of output: " + output.Text())
			}
			// 仅在非管道关闭的错误时panic
			if err := output.Err(); err != nil {
				errMsg := err.Error()
				if !strings.Contains(errMsg, "file already closed") && !strings.Contains(errMsg, "EOF") {
					panic(err)
				}
			}
		}()

		if err := cmd.Start(); err != nil {
			panic(err)
		}

		// 判断是否为后台命令（简单处理：去掉末尾空格后以&结尾）
		isBackground := false
		trimmedCmd := strings.TrimSpace(cmdStr)
		if len(trimmedCmd) > 0 && trimmedCmd[len(trimmedCmd)-1] == '&' {
			isBackground = true
		}

		// 前台命令阻塞等待，后台命令直接继续
		if !isBackground {
			if err := cmd.Wait(); err != nil {
				println("command error:", err.Error())
			}
		}
	}
}

方案说明 #

1. 命令包装：通过> >(cat)将命令输出重定向到cat进程，cat会持续读取后台进程的输出并写入Go的管道，直到后台进程结束，避免父Shell退出导致管道关闭。
2. 错误处理：忽略管道关闭、EOF等正常错误，仅在遇到其他异常时panic。
3. 后台命令判断：通过简单的字符串判断（命令末尾是否为&）决定是否阻塞，若需要更准确的Shell命令解析，可引入第三方Shell解析库。

注意事项 #

• 若需要捕获标准错误输出，将包装命令修改为cmdStr + " > >(cat) 2>&1"。
• 简单的字符串判断可能误判包含&的非后台命令（如echo "hello & world"），若需更精准解析，可使用如github.com/mvdan/sh等Shell语法解析库。

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