如何在Golang中将函数标准输出通过管道传递给fzf命令？

原捕获函数的问题分析 #

你写的capture函数存在几个关键错误：

• 提前关闭了管道写端，导致目标函数执行时没有有效输出通道
• 未使用goroutine异步执行目标函数，会引发阻塞死锁
• 缺少错误处理逻辑，且标准输出恢复时机不合理

直接传递输出到fzf的高效实现 #

不需要先将输出转为字符串，直接通过管道将Go函数的标准输出传递给fzf的标准输入，更节省内存且高效：

package main

import (
	"os"
	"os/exec"
)

func example() {
	println("January")
	println("February")
	println("March")
}

func runExampleWithFzf() error {
	// 创建管道，用于衔接example输出与fzf输入
	r, w, err := os.Pipe()
	if err != nil {
		return err
	}
	defer r.Close()

	// 保存原标准输出，后续恢复
	originalStdout := os.Stdout
	os.Stdout = w

	// 异步执行目标函数，避免阻塞主流程
	go func() {
		defer func() {
			// 恢复原标准输出并关闭管道写端，让fzf收到EOF
			os.Stdout = originalStdout
			w.Close()
		}()
		example()
	}()

	// 初始化fzf命令
	fzfCmd := exec.Command("fzf")
	// 将管道读端绑定为fzf的标准输入
	fzfCmd.Stdin = r
	// 让fzf的交互输出和错误信息直接显示在当前终端
	fzfCmd.Stdout = os.Stdout
	fzfCmd.Stderr = os.Stderr

	// 执行fzf并等待其退出
	return fzfCmd.Run()
}

func main() {
	if err := runExampleWithFzf(); err != nil {
		println("执行出错:", err.Error())
		os.Exit(1)
	}
}

先捕获输出再传递给fzf的实现（适用于需要预处理输出的场景） #

如果需要先把Go函数的输出捕获为字符串再处理，可修复capture函数后配合fzf使用：

修复后的捕获函数

import (
	"bytes"
	"io"
	"os"
)

func capture(f func()) (string, error) {
	r, w, err := os.Pipe()
	if err != nil {
		return "", err
	}
	defer r.Close()

	originalStdout := os.Stdout
	os.Stdout = w

	var buf bytes.Buffer
	done := make(chan error)

	// 异步执行目标函数
	go func() {
		defer func() {
			os.Stdout = originalStdout
			w.Close()
		}()
		f()
		done <- nil
	}()

	// 读取管道数据到缓冲区
	_, err = io.Copy(&buf, r)
	if err != nil {
		return "", err
	}
	// 等待目标函数执行完成
	<-done

	return buf.String(), nil
}

配合fzf调用的函数

func runCapturedExampleWithFzf() error {
	output, err := capture(example)
	if err != nil {
		return err
	}

	fzfCmd := exec.Command("fzf")
	fzfCmd.Stdin = bytes.NewBufferString(output)
	fzfCmd.Stdout = os.Stdout
	fzfCmd.Stderr = os.Stderr

	return fzfCmd.Run()
}

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