Question

所以我有一个Go程序从STDIN读取如下。我想从键盘或设备输入用户名和密码，但字符串切片可以使用管道传递。如果我运行如下命令：

回声“Hello World”|去运行main.go

os.Stdin将设置为从管道读取而不是键盘。有没有办法我可以更改os.Stdin FileMode，它将从设备读取，即用户名和密码的键盘？

我尝试使用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")
    }

Answer 1

可能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)
}

稍微详细一些，以检查是否有从stdin读取的内容，如果没有提示用户：

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 {
            if 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)
    }
}

Answer 2

您可以从/dev/tty读取，因为它始终是终端（如果程序在终端上运行）。这只适用于类Unix系统（Linux，BSD，macOS等），不适用于Windows。

// +build !windows

tty, err := os.Open("/dev/tty")
if err != nil {
    log.Fatalf("can't open /dev/tty: %s", err)
}
scanner := bufio.NewScanner(tty)
// as you were ...

当stdin是管道时如何从设备读取

2 个答案: