Go开发控制台游戏：如何实现无缓冲输入管理器？

作为经常折腾Go控制台工具和小游戏的开发者，我太懂你需要无缓冲输入的迫切性了——游戏里按键按下就得立刻响应，延迟半秒玩家体验直接拉胯！

先聊聊你遇到的两个方案的问题：

• 类Unix多函数方案：确实能实现无缓冲输入，但不仅代码繁琐，而且完全不支持Windows，跨平台游戏直接pass。
• cgo单函数方案：代码简洁是简洁，但cgo的调用开销在游戏这种需要高频读取输入的场景下简直灾难，每次调用都要跨Go和C的边界，延迟高到没法玩。

那咱们直接上无cgo、低延迟、跨平台的解决方案，分系统实现，完全适配游戏场景：

类Unix系统（Linux/macOS/FreeBSD） #

直接通过syscall修改终端属性，禁用规范模式（canonical mode）和回显，这样就能实现按下按键立刻读取：

//go:build linux || darwin || freebsd
package main

import (
	"syscall"
	"unsafe"
)

type Termios struct {
	Iflag  uint32
	Oflag  uint32
	Cflag  uint32
	Lflag  uint32
	Cc     [19]byte
	Ispeed uint32
	Ospeed uint32
}

const termiosGet = 0x5401 // TCGETS

// GetChar 读取单个无缓冲按键输入
func GetChar() byte {
	fd := int(syscall.Stdin)
	var termios Termios
	// 保存当前终端属性
	syscall.Syscall(syscall.SYS_IOCTL, uintptr(fd), termiosGet, uintptr(unsafe.Pointer(&termios)))
	oldTermios := termios

	// 禁用规范模式和回显：规范模式下会等到换行才返回输入，回显会把按键显示在控制台
	termios.Lflag &^= syscall.ICANON | syscall.ECHO
	syscall.Syscall(syscall.SYS_IOCTL, uintptr(fd), termiosGet, uintptr(unsafe.Pointer(&termios)))

	// 读取单个字节
	var b [1]byte
	syscall.Read(fd, b[:])

	// 恢复终端属性，避免影响后续程序
	syscall.Syscall(syscall.SYS_IOCTL, uintptr(fd), termiosGet, uintptr(unsafe.Pointer(&oldTermios)))
	return b[0]
}

Windows系统 #

Windows需要调用控制台API来修改输入模式，直接读取按键事件：

//go:build windows
package main

import (
	"syscall"
	"unicode/utf16"
	"unsafe"
)

var (
	kernel32        = syscall.NewLazyDLL("kernel32.dll")
	getStdHandle    = kernel32.NewProc("GetStdHandle")
	setConsoleMode  = kernel32.NewProc("SetConsoleMode")
	readConsoleInput = kernel32.NewProc("ReadConsoleInputW")
)

const (
	stdInputHandle = ^uint32(0) - 1
	enableLineInput = 0x0002
	enableEchoInput = 0x0004
)

type INPUT_RECORD struct {
	EventType uint16
	_         uint16
	Event     [24]byte
}

type KEY_EVENT_RECORD struct {
	KeyDown         int32
	RepeatCount     uint16
	VirtualKeyCode  uint16
	VirtualScanCode uint16
	UnicodeChar     [2]uint16
	ControlKeyState uint32
}

// GetChar 读取单个无缓冲按键输入
func GetChar() rune {
	// 获取标准输入句柄
	handle, _, _ := getStdHandle.Call(uintptr(stdInputHandle))
	// 保存原控制台模式
	var oldMode uint32
	setConsoleMode.Call(handle, uintptr(oldMode))
	// 禁用行输入和回显，实现无缓冲输入
	setConsoleMode.Call(handle, uintptr(oldMode & ^(enableLineInput | enableEchoInput)))

	var records [1]INPUT_RECORD
	var numRead uint32
	// 循环读取直到拿到有效按键事件
	for {
		readConsoleInput.Call(handle, uintptr(unsafe.Pointer(&records)), 1, uintptr(unsafe.Pointer(&numRead)))
		if records[0].EventType == 1 { // KEY_EVENT 类型
			keyEvent := (*KEY_EVENT_RECORD)(unsafe.Pointer(&records[0].Event))
			if keyEvent.KeyDown != 0 {
				// 恢复控制台模式
				setConsoleMode.Call(handle, uintptr(oldMode))
				return rune(utf16.Decode(keyEvent.UnicodeChar[:])[0])
			}
		}
	}
}

跨平台整合 #

把上面的代码分别存为input_unix.go和input_windows.go，再创建一个通用的input.go：

package main

import "fmt"

func main() {
	fmt.Println("游戏测试：按下任意键...")
	c := GetChar()
	fmt.Printf("你按下了：%c\n", c)
}

这样编译时Go会自动根据目标系统选择对应的实现，完全不用cgo，延迟极低，完美适配游戏的高频输入需求！

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