如何检测输出写入端是否为tty，在Go中实现类Bash的彩色输出自动适配

核心逻辑 #

你要实现的效果本质是检测程序的标准输出是否连接到交互式终端（TTY）：

• 输出目标是终端时，携带颜色控制码输出彩色内容
• 输出目标是文件/管道等非终端时，去掉颜色控制码输出纯文本

Go语言实现步骤 #

1. 使用官方扩展包golang.org/x/term提供的IsTerminal方法，检测标准输出的文件描述符是否为终端
2. 封装通用的彩色输出函数，根据检测结果决定是否拼接颜色控制码

完整示例代码 #

package main

import (
	"fmt"
	"os"
	"golang.org/x/term"
)

// 定义颜色控制码常量
const (
	colorBlueBold = "\033[1;34m"
	colorReset    = "\033[0m"
)

// 全局变量，初始化时判断是否为终端输出
var isTTY = term.IsTerminal(int(os.Stdout.Fd()))

// 封装蓝色加粗输出函数
func bluePrintln(s string) {
	if isTTY {
		fmt.Println(colorBlueBold + s + colorReset)
		return
	}
	fmt.Println(s)
}

func main() {
	bluePrintln("Hello World!")
}

如果需要对标准错误输出做同样的判断，将检测的文件描述符替换为os.Stderr.Fd()即可。

可选优化：支持NO_COLOR规范 #

可以额外检测环境变量NO_COLOR，只要该变量存在（无论值是什么）就强制关闭颜色输出，符合主流命令行工具的通用约定：

var isTTY = term.IsTerminal(int(os.Stdout.Fd())) && os.Getenv("NO_COLOR") == ""

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