在Go语言开发终端交互类程序时，捕获方向键输入是实现菜单选择、光标移动等功能的基础。方向键的输入不属于普通字符，而是以转义序列的形式发送到终端，因此需要通过特定的方式读取和解析。
终端输入的基本原理
常规情况下，终端处于规范模式，输入会以行缓冲的形式处理，用户按下回车后才会将输入内容传递给程序。而方向键、功能键等输入会产生以ESC（转义字符，对应ASCII码27）开头的多字节序列，规范模式下无法正确读取这些序列，因此需要先将终端切换到原始模式。
设置终端原始模式
不同操作系统的终端设置方式不同，这里以Linux/macOS系统为例，通过syscall包调用系统函数设置终端属性。首先需要定义终端属性结构体，然后修改终端的模式标志。
package main
import (
"fmt"
"syscall"
"unsafe"
)
// 设置终端为原始模式
func enableRawMode() (*syscall.Termios, error) {
var oldState syscall.Termios
// 获取当前终端属性
if _, _, err := syscall.Syscall6(syscall.SYS_IOCTL, uintptr(syscall.Stdin), uintptr(syscall.TCGETS), uintptr(unsafe.Pointer(&oldState)), 0, 0, 0); err != 0 {
return nil, err
}
newState := oldState
// 关闭回显和规范模式
newState.Lflag &= ^uint32(syscall.ECHO | syscall.ICANON)
// 设置最小读取字符数为1，超时时间为0
newState.Cc[syscall.VMIN] = 1
newState.Cc[syscall.VTIME] = 0
// 设置新的终端属性
if _, _, err := syscall.Syscall6(syscall.SYS_IOCTL, uintptr(syscall.Stdin), uintptr(syscall.TCSETS), uintptr(unsafe.Pointer(&newState)), 0, 0, 0); err != 0 {
return nil, err
}
return &oldState, nil
}
// 恢复终端原有属性
func disableRawMode(oldState *syscall.Termios) error {
if _, _, err := syscall.Syscall6(syscall.SYS_IOCTL, uintptr(syscall.Stdin), uintptr(syscall.TCSETS), uintptr(unsafe.Pointer(oldState)), 0, 0, 0); err != 0 {
return err
}
return nil
}
读取并解析方向键输入
方向键的转义序列通常为ESC [ A（上）、ESC [ B（下）、ESC [ C（右）、ESC [ D（左），其中ESC对应字节0x1b，[对应0x5b，后续字母对应不同方向。我们可以逐个字节读取输入，识别转义序列来判断方向键。
// 读取单个字节
func readByte() (byte, error) {
buf := make([]byte, 1)
n, err := syscall.Read(syscall.Stdin, buf)
if err != nil || n != 1 {
return 0, err
}
return buf[0], nil
}
// 方向键类型定义
type Direction int
const (
DirUp Direction = iota
DirDown
DirLeft
DirRight
DirUnknown
)
// 读取方向键输入
func readDirection() Direction {
b, err := readByte()
if err != nil {
return DirUnknown
}
// 如果不是转义字符，直接返回未知
if b != 0x1b {
return DirUnknown
}
// 读取下一个字节，应该是 [
b, err = readByte()
if err != nil || b != 0x5b {
return DirUnknown
}
// 读取方向标识字节
b, err = readByte()
if err != nil {
return DirUnknown
}
switch b {
case 0x41: // A 对应上
return DirUp
case 0x42: // B 对应下
return DirDown
case 0x43: // C 对应右
return DirRight
case 0x44: // D 对应左
return DirLeft
default:
return DirUnknown
}
}
完整示例程序
将上面的函数组合起来，实现一个可以捕获方向键输入并输出的示例程序，程序运行后按下方向键会打印对应的方向，按下q键退出。
func main() {
oldState, err := enableRawMode()
if err != nil {
fmt.Println("设置终端模式失败:", err)
return
}
defer disableRawMode(oldState)
fmt.Println("按下方向键测试，按q退出")
for {
dir := readDirection()
if dir != DirUnknown {
switch dir {
case DirUp:
fmt.Println("检测到上方向键")
case DirDown:
fmt.Println("检测到下方向键")
case DirLeft:
fmt.Println("检测到左方向键")
case DirRight:
fmt.Println("检测到右方向键")
}
continue
}
// 读取普通字符，判断是否为q
b, err := readByte()
if err != nil {
break
}
if b == 'q' || b == 'Q' {
break
}
}
fmt.Println("程序退出")
}
注意事项
- 上述代码仅适用于Linux/macOS系统，Windows系统需要使用
golang.org/x/crypto/ssh/terminal或者调用Windows API实现终端模式切换。 - 程序退出前一定要恢复终端原有属性，否则终端会出现输入无回显等异常情况。
- 部分终端的方向键转义序列可能存在差异，实际开发中可以根据需要补充更多序列的解析逻辑。