所以我有一个从 STDIN 读取的 Go 程序,如下所示。我希望从键盘或设备输入用户名和密码,但可以使用管道传递字符串切片。如果我运行如下命令:
echo "Hello World" | go run main.go
Run Code Online (Sandbox Code Playgroud)
os.Stdin将被设置为从管道读取,而不是从键盘读取。有没有一种方法可以更改os.StdinFileMode,使其从设备(即键盘)读取用户名和密码?
我尝试使用os.Stdin.Chmod(FileMode)但收到此错误:
chmod /dev/stdin:参数无效
func main() {
var n = []string{}
scanner := bufio.NewScanner(os.Stdin)
fmt.Println("Please type anything with Newline Separated, empty line signals termination")
for scanner.Scan() {
h := scanner.Text()
if h == "" {
break
}
n = append(n, h)
}
if err := scanner.Err(); err != nil {
fmt.Printf("Error in reading from STDIN: %v\n", err)
}
reader := bufio.NewReader(os.Stdin)
os.Stdout.WriteString("Username: ")
username, err := reader.ReadString('\n')
if err != nil {
fmt.Printf("Unable to read username: %v\n", err)
}
username = strings.TrimSpace(username)
os.Stdout.WriteString("Password: ")
bytePassword, _ := terminal.ReadPassword(int(os.Stdin.Fd()))
password := string(bytePassword)
os.Stdout.WriteString("\n")
}
Run Code Online (Sandbox Code Playgroud)
也许scanf可以提供帮助,请检查这个示例:
https://play.golang.org/p/tteQNl0trJp
package main
import (
"fmt"
)
func main() {
fmt.Println("Enter your name")
var name string
fmt.Scanf("%s", &name)
fmt.Printf("name = %s\n", name)
}
Run Code Online (Sandbox Code Playgroud)
更详细的内容是检查是否有东西可以从标准输入读取,如果没有提示用户:
https://play.golang.org/p/7qeAQ5UNhdQ
package main
import (
"bufio"
"fmt"
"log"
"os"
)
func main() {
// check if there is somethinig to read on STDIN
stat, _ := os.Stdin.Stat()
if (stat.Mode() & os.ModeCharDevice) == 0 {
var stdin []byte
scanner := bufio.NewScanner(os.Stdin)
for scanner.Scan() {
stdin = append(stdin, scanner.Bytes()...)
}
if err := scanner.Err(); err != nil {
log.Fatal(err)
}
fmt.Printf("stdin = %s\n", stdin)
} else {
fmt.Println("Enter your name")
var name string
fmt.Scanf("%s", &name)
fmt.Printf("name = %s\n", name)
}
}
Run Code Online (Sandbox Code Playgroud)