确保正确捕获用户输入

对于一个Golang开发者来说，牢固扎实的基础是十分重要的，golang学习网就来带大家一点点的掌握基础知识点。今天本篇文章带大家了解《确保正确捕获用户输入》，主要介绍了，希望对大家的知识积累有所帮助，快点收藏起来吧，否则需要时就找不到了！

我对 golang 还很陌生，我正在尝试读取用户的输入，这是要以编程方式打开的文件的名称。

我最初尝试过这个：

package main

import (
    "fmt"
    "os"
    "log"
    "io/ioutil"
)

func takeinput() {
    r, _ := os.getwd()
    fmt.println("your working dir:", r)
    fmt.print("enter a file: ")
    f, err := bufio.newreader(os.stdin).readstring('\n')
    if err != nil {
        log.fatal(err)
    }
    s, err := ioutil.readfile(string(f))
    if err != nil {
        log.fatal(err)
    }
    fmt.println(string(s))
}

func main() {
    takeinput()
}

但不知怎的，它总是会导致：

$ go run getin.go

your working dir: /media/data/src
enter a file: data.txt
2020/07/23 22:21:03 open data.txt
: no such file or directory
exit status 1

我在当前工作目录中有文件 data.txt。

我用谷歌搜索了一下，发现我也可以使用 fmt.scanln 来获取用户输入，效果很好：

func takeinput() {
    r, _ := os.getwd()
    fmt.println("your working dir:", r)
    var f string
    fmt.print("enter a file: ")
    fmt.scanln(&f)  // what i changed is how input is taken
    s, err := ioutil.readfile(string(f))
    if err != nil {
        log.fatal(err)
    }
    fmt.println(string(s))
}

$ go run getin.go

Your working dir: /media/DATA/Src
Enter a file: data.txt
100,200,300,445
...

有人可以帮助我了解使用 bufio.newreader 获取用户输入时出了什么问题吗？

解决方案

根据 readstring 的文档：

readstring 读取直到输入中第一次出现 delim，返回包含分隔符之前的数据的字符串。

所以第一种方法的问题是文件名末尾包含换行符。使用 scanln，它会读取直到看到换行符，但它不会在输出中包含换行符。

因此，在获取用户输入后，您可以通过更新代码轻松删除空格：

func takeInput() {
    r, _ := os.Getwd()
    fmt.Println("Your working dir:", r)
    fmt.Print("Enter a file: ")
    f, err := bufio.NewReader(os.Stdin).ReadString('\n')
    if err != nil {
        log.Fatal(err)
    }
    s, err := ioutil.ReadFile(strings.TrimSpace(f)))  // what needs to be changed
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println(string(s))
}

另外不要忘记导入 strings 包。

今天关于《确保正确捕获用户输入》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！