如何使用 ReadLine() 获取字符串输入？

Golang小白一枚，正在不断学习积累知识，现将学习到的知识记录一下，也是将我的所得分享给大家！而今天这篇文章《如何使用 ReadLine() 获取字符串输入？》带大家来了解一下##content_title##，希望对大家的知识积累有所帮助，从而弥补自己的不足，助力实战开发！

我想使用 func (b *reader) readline() (line []byte, isprefix bool, err error) 从 stdio 获取字符串，但我的代码无法正常工作。

我倾向于 golang。我想知道如何使用 readline() 从标准输入获取字符串

我知道，fmt.scan 或 scanner 可以帮助我，但我想使用 readline()

package main

import (
        "bufio"
        "fmt"
        "io"
        "os"
        "strconv"
)

var sc = bufio.newscanner(os.stdin)
var rd = bufio.newreadersize(os.stdin, 1000000)

func nextint() int {
        sc.scan()
        i, e := strconv.atoi(sc.text())
        if e != nil {
                panic(e)
        }
        return i
}

func nextline() string {
        buf := make([]byte, 0, 1000000)
        for {
                line, isprefix, err := rd.readline()

                if err == io.eof {
                        break
                } else if err != nil {
                        panic(err)
                }

                buf = append(buf, line...)

                if !isprefix {
                        break
                }
        }
        return string(buf)
}

func main() {
        var s string
        var a int

        s = nextline()
        a = nextint()

        fmt.println(s)
        fmt.println(a)
}

• 结果

$ ./a.out
test # input
334  # input
test
334

$ cat in.txt
test
334
$ ./a.out < in.txt
panic: strconv.Atoi: parsing "": invalid syntax

goroutine 1 [running]:
main.nextInt(0xc042056088)

我希望两个输出应该是相同的， 但是当我使用重定向时，它不起作用并得到不同的输出。

解决方案

摆脱扫描仪（您已经说过您更喜欢 readline()）并更改您的 nextint() 函数以调用 nextline()，如下所示：

func nextint() int {
    i, e := strconv.atoi(nextline())
    if e != nil {
        panic(e)
    }
    return i
}

（顺便说一句，对错误的用户输入感到恐慌不是一个好主意，但我认为这只是一个测试，您不会对生产代码这样做:)

您可以尝试不使用扫描仪，也许像这样 readline 将为您获取数字，只需转换即可

package main

import (
    "bufio"
    "fmt"
    "io"
    "os"
    "strconv"
)

var rd = bufio.NewReaderSize(os.Stdin, 1000000)

func nextLine() string {
    buf := make([]byte, 0, 1000000)
    for {
        line, isPrefix, err := rd.ReadLine()

        if err == io.EOF {
            break
        } else if err != nil {
            panic(err)
        }

        buf = append(buf, line...)

        if !isPrefix {
            break
        }
    }
    return string(buf)
}

func main() {
    var s string
    var a int

    s = nextLine()
    fmt.Println(s)
    s = nextLine()
    a, e := strconv.Atoi(s)
    if e != nil {
        panic(e)
    }
    fmt.Println(a)
}

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《如何使用 ReadLine() 获取字符串输入？》文章吧，也可关注golang学习网公众号了解相关技术文章。