我正在尝试编写一个函数,该函数可以一次从键盘读取输入或从管道文件中读取一行。我已经有一个类似于prompt()
此测试代码中的键盘输入功能:
package main
import (
"fmt"
"bufio"
"os"
)
func print(format string, a ...interface{}) {
fmt.Printf(format+"\n", a...)
}
func prompt(format string) string {
fmt.Print(format)
in := bufio.NewScanner(os.Stdin)
in.Scan()
return in.Text()
}
func greet() {
name := prompt("enter name: ")
print(`Hello %s!`, name)
}
func humor() {
color := prompt("enter favorite color: ")
print(`I like %s too!`, color)
}
func main() {
greet()
humor()
}
在这里,greet()
两者humor()
都用于prompt()
获取输入,如果我运行程序并输入响应,它将按预期工作。但是,如果我有一个文件a.txt
:
bobby bill
soft, blue-ish turquoise
然后运行: .\test< a.txt
,程序会输出:
enter name: Hello bobby bill!
enter favorite color: I like too!
代替:
enter name: Hello bobby bill!
enter favorite color: I like soft, blue-ish turquoise too!
据我了解,这是因为bufio.Scanner
那是在greet()
read all of中制作的a.txt
。我可以通过创建bufio.Scanner
一个全局变量并prompt()
使用它而不是每次都创建一个新变量来轻松解决这个问题bufio.Scanner
,但我想知道是否有更好的方法来做到这一点而不必求助于全局变量。