在go语言开发中，有时我们需要获取当前运行环境的终端（tty）尺寸，即其宽度和高度。初学者可能会尝试通过执行外部命令，例如stty size，来获取这些信息。然而，这种方法往往会遇到问题，导致无法获得预期的结果。
为什么stty size命令在Go中可能失败？
当我们使用os/exec包来执行stty size命令时，Go会在一个新的、独立的进程中运行该命令。这个新进程默认情况下不会继承或关联到当前程序所连接的终端。因此，当stty size在一个没有关联到任何交互式终端的进程中执行时，它无法获取到有效的终端尺寸信息，通常会返回空输出或错误。
考虑以下示例代码，它展示了这种尝试及其失败的输出：
package main
import (
"fmt"
"log"
"os/exec"
)
func main() {
out, err := exec.Command("stty", "size").Output()
fmt.Printf("输出: %#v\n", out)
fmt.Printf("错误: %#v\n", err)
if err != nil {
log.Fatal(err)
}
}运行上述代码，你可能会得到类似如下的输出：
输出: []byte{}
错误: &exec.ExitError{ProcessState:(*os.ProcessState)(0xc0000a6000)}
2023/10/27 10:00:00 exit status 1这明确表明stty size命令未能成功执行或返回有效数据，通常伴随着非零的退出状态码。
立即学习“go语言免费学习笔记（深入）”；
正确的解决方案：使用golang.org/x/crypto/ssh/terminal包
Go语言生态系统为我们提供了更优雅、更可靠的解决方案，即使用golang.org/x/crypto/ssh/terminal包。尽管这个包最初是为SSH客户端/服务器功能设计的，但它包含了一个非常实用的GetSize方法，可以用来获取任何文件描述符所关联终端的尺寸。
terminal.GetSize方法介绍
terminal.GetSize方法的签名如下：
func GetSize(fd int) (width, height int, err error)
它接受一个整数类型的文件描述符（fd），并返回终端的宽度、高度以及可能发生的错误。
如何获取当前终端的文件描述符？
在Go中，我们可以通过os.Stdin.Fd()来获取标准输入的文件描述符。os.Stdin代表了程序的标准输入流，通常它会连接到当前运行程序的终端。Fd()方法返回一个uintptr类型的值，我们需要将其转换为int类型以符合GetSize的参数要求。
import (
"os"
"golang.org/x/crypto/ssh/terminal"
)
// ...
fd := int(os.Stdin.Fd())
width, height, err := terminal.GetSize(fd)
// ...完整代码示例
下面是一个完整的Go程序，演示了如何使用golang.org/x/crypto/ssh/terminal包来正确获取终端尺寸：
package main
import (
"fmt"
"log"
"os"
"golang.org/x/crypto/ssh/terminal" // 导入terminal包
)
func main() {
// 获取标准输入的文件描述符
fd := int(os.Stdin.Fd())
// 检查标准输入是否连接到终端
if !terminal.IsTerminal(fd) {
log.Fatal("标准输入不是一个终端，无法获取尺寸。")
}
// 使用terminal.GetSize获取终端的宽度和高度
width, height, err := terminal.GetSize(fd)
if err != nil {
log.Fatalf("获取终端尺寸失败: %v", err)
}
fmt.Printf("当前终端宽度: %d\n", width)
fmt.Printf("当前终端高度: %d\n", height)
}在运行此代码之前，请确保你已经安装了golang.org/x/crypto模块：
go get golang.org/x/crypto/ssh/terminal
工作原理简析
golang.org/x/crypto/ssh/terminal包的GetSize方法在底层使用了操作系统提供的系统调用（syscall）来查询指定文件描述符的终端属性。例如，在类Unix系统上，它可能调用ioctl函数，并传入TIOCGWINSZ等命令来获取窗口大小信息。这种直接与操作系统交互的方式，避免了执行外部命令所带来的进程隔离问题，从而能够准确地获取当前终端的尺寸。
注意事项
- 依赖管理: 确保你的项目中已正确引入golang.org/x/crypto/ssh/terminal包。如果你的go.mod文件中没有，go get命令会自动添加。
- 错误处理: GetSize方法可能会返回错误，例如当文件描述符不关联到终端时。因此，始终检查返回的err是良好的编程习惯。
- 非终端环境: 如果你的程序运行在一个非交互式环境中（例如作为后台服务、通过管道输入），terminal.IsTerminal(fd)可能会返回false，或者GetSize会返回错误。在这些情况下，获取终端尺寸通常没有意义，或需要特殊的处理逻辑。
- 跨平台兼容性: golang.org/x/crypto/ssh/terminal包提供了良好的跨平台兼容性，它会根据不同的操作系统（如Linux, macOS, Windows）调用相应的底层API来获取终端尺寸。
总结
在Go语言中获取终端尺寸，避免使用exec.Command("stty", "size")这类依赖外部命令的方式，因为它们容易受到进程环境和TTY关联性的限制。推荐的做法是利用golang.org/x/crypto/ssh/terminal包提供的GetSize方法，结合os.Stdin.Fd()获取当前终端的文件描述符。这种方法不仅专业、高效，而且具有良好的跨平台兼容性，是Go语言中获取终端尺寸的标准和推荐实践。
