如何在Golang CLI程序中无需按Enter接收扫码器输入的字符串？

问题描述 #

需要实现扫码器读取的字符串直接传入函数处理，无需按Enter键，终端不能像bufio.NewScanner.Scan()那样等待回车，需立即处理读取到的字符串。尝试了两种方案均无效，终端仍等待回车，代码如下：

尝试的方案一代码 #

package main

import (
    "bufio"
    "fmt"
    "log"
    "os"

    "golang.org/x/term"
)

func main() {
    state, err := term.MakeRaw(int(os.Stdin.Fd()))
    if err != nil {
        log.Fatalln("setting stdin to raw:", err)
    }
    defer func() {
        if err := term.Restore(int(os.Stdin.Fd()), state); err != nil {
            log.Println("warning, failed to restore terminal:", err)
        }
    }()

    in := bufio.NewReader(os.Stdin)

    l, _, err := in.ReadLine()
    if err != nil {
        fmt.Printf("error: %v\r\n", err)
    }
    fmt.Printf("read line %v\r\n", l)

}

尝试的方案二代码 #

package main

import (
    "fmt"
    "log"
    "os"

    "golang.org/x/term"
)

func main() {
    var (
        line string
        quit bool
        err  error
    )

    // setup terminal
    oldState, err := term.MakeRaw(int(os.Stdin.Fd()))
    if err != nil {
        log.Fatalln("setting stdin to raw:", err)
    }
    defer func() {
        if err := term.Restore(int(os.Stdin.Fd()), oldState); err != nil {
            log.Println("warning, failed to restore terminal:", err)
        }
    }()

    term := term.NewTerminal(os.Stdin, "> ")

    for err == nil && !quit {
        line, err = term.ReadLine()
        if err != nil {
            break
        }

        if string(line) == "q" {
            quit = true
        }
    }

    if err != nil {
        fmt.Println(err)
    }

    if quit {
        fmt.Println("Quitting...")
    }
}

解决方案 #

你之前的代码使用bufio.Reader.ReadLine()和term.ReadLine()，这两个方法本质上都是等待换行符触发读取，所以必须按回车才会返回结果。要实现即时读取，需要直接从标准输入逐字节读取——扫码器本质是模拟键盘连续输出字符，我们可以在raw模式下循环捕获每个字节，直到识别到扫码结束的标记（多数扫码器默认会在结尾输出换行符，可根据实际调整）。

可行代码实现 #

package main

import (
	"fmt"
	"log"
	"os"

	"golang.org/x/term"
)

func main() {
	// 保存终端原始状态，退出时恢复
	oldState, err := term.MakeRaw(int(os.Stdin.Fd()))
	if err != nil {
		log.Fatalln("无法设置终端为raw模式:", err)
	}
	defer func() {
		if err := term.Restore(int(os.Stdin.Fd()), oldState); err != nil {
			log.Println("警告：无法恢复终端状态:", err)
		}
	}()

	fmt.Println("等待扫码输入... (按Ctrl+C退出)")

	var scanBuffer []byte
	for {
		var buf [1]byte
		// 直接读取单个字节，无缓冲，输入即读取
		n, err := os.Stdin.Read(buf[:])
		if err != nil {
			log.Fatalln("读取输入失败:", err)
		}
		if n == 0 {
			continue
		}

		char := buf[0]
		// 以换行/回车作为扫码结束标记，适配多数扫码器默认配置
		if char == '\n' || char == '\r' {
			if len(scanBuffer) > 0 {
				scanContent := string(scanBuffer)
				fmt.Printf("捕获扫码内容: %s\n", scanContent)
				// 调用自定义处理函数
				processScan(scanContent)
				scanBuffer = nil // 清空缓冲区，准备下一次扫码
			}
		} else {
			scanBuffer = append(scanBuffer, char)
		}
	}
}

// 自定义扫码结果处理函数，替换为你的业务逻辑
func processScan(content string) {
	fmt.Printf("正在处理扫码内容: %s\n", content)
	// 示例：可以在这里调用其他业务函数、存储数据等
}

关键说明 #

• 直接字节读取：使用os.Stdin.Read()读取单个字节，绕过了行缓冲机制，确保输入的每个字符都能被立即捕获
• Raw模式终端：终端设置为raw模式后，取消了终端的预处理（如回显、换行转换），保证扫码器的所有输入字符都能被完整接收
• 结束标记判断：多数扫码器默认会在扫码完成后输出换行/回车，以此作为完整内容的结束信号；如果你的扫码器没有这个配置，可以改为判断固定长度、特定后缀等
• 终端状态恢复：通过defer语句确保程序退出时恢复终端原始状态，避免终端出现异常

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