使用文件作为 Stdin (Golang) 时在第二个提示时获取 EOF

欢迎各位小伙伴来到golang学习网，相聚于此都是缘哈哈哈！今天我给大家带来《使用文件作为 Stdin (Golang) 时在第二个提示时获取 EOF》，这篇文章主要讲到等等知识，如果你对Golang相关的知识非常感兴趣或者正在自学，都可以关注我，我会持续更新相关文章！当然，有什么建议也欢迎在评论留言提出！一起学习！

我正在尝试以类似的方式对 cli 应用程序进行功能测试。

由于命令在命令提示符下要求一些输入，我将它们放入一个文件中并将其设置为 os.stdin。

cmd := exec.command(path.join(dir, binaryname), "myarg")
tmpfile := setstdin("themasterpassword\nsecondanswer\n12121212\n")
cmd.stdin = tmpfile
output, err := cmd.combinedoutput()

setstdin 只是创建一个 tmpfile，将字符串写入文件并返回 *os.file。

现在，我期望 themasterpassword 成为第一个输入，并且它正在工作。但对于第二个输入，总是出现 严重错误：eof。

我用来询问和获取用户输入的函数：

func ask(question string, minlen int) string {
    reader := bufio.newreader(os.stdin)

    for {
        fmt.printf("%s: ", question)

        response, err := reader.readstring('\n')
        exitiferror(err)

        if len(response) >= minlen {
            return strings.trimspace(response)
        } else {
            fmt.printf("provide at least %d character.\n", minlen)
        }
    }
}

你能帮我看看出了什么问题吗？ 非常感谢！

根据要求添加 setstdin

func setStdin(userInput string) *os.File {
    tmpfile, err := ioutil.TempFile("", "test_stdin_")
    util.ExitIfError(err)

    _, err = tmpfile.Write([]byte(userInput))
    util.ExitIfError(err)
    _, err = tmpfile.Seek(0, 0)
    util.ExitIfError(err)

    return tmpfile
}

解决方案

在您的应用程序中，每当您需要单个输入行时，您都可以调用 ask() 。

在 ask() 中，您创建一个 bufio.Reader 以从 os.stdin 读取。要知道 bufio.reader（顾名思义）使用缓冲读取，这意味着它可能从其源读取比其方法返回的数据更多的数据（在本例中为 Reader.ReadString()）。这意味着如果您只用它来读取一行（或几行）并且扔掉读取器，您将扔掉缓冲的未读数据。

因此，下次您再次调用 ask() 并尝试从 os.stdin 读取时，您将不会从上次中断的位置继续...

要解决此问题，只需从 os.stdin 创建单个 bufio.reader，将其存储在全局变量中，例如，在 ask() 内，始终使用此单个读取器。因此，在 ask() 调用之间缓冲和未读的数据不会丢失。当然，这个解决方案对于从多个 goroutine 调用是无效的，但从单个 os.stdin 读取也不是。

例如：

var reader = bufio.NewReader(os.Stdin)

func Ask(question string, minLen int) string {
    // use the global reader here...
}

另请注意，在您的情况下使用 bufio.Scanner 会更容易。但同样，bufio.scanner 也可能从其源读取超出所需的数据，因此您也必须在此处使用共享 bufio.scanner。另请注意， reader.readstring() 返回一个包含定界符的字符串（在您的情况下以 \n 结尾的行），您可能需要修剪该定界符，而 Scanner.Text() （使用默认的行分割功能）将在之前先删除该定界符返回线路。这也是您可以利用的一种简化。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。