Golang reader.ReadLine并不能一下读一行
在秋招笔试的时候，我们如果用牛客答题，那很有可能会让你从终端读取传给你的数据，那么如何接收才是准确的呢？今天给大家避个坑。不要使用bufio.Reader.ReadLine，因为他默认最多读4096个byte。
通常我们要读取终端的输入一般会这样写：
reader := bufio.NewReader(os.Stdin) line, _, _ := reader.ReadLine() fmt.Println(string(line))
这样看似很完美，但是如果你输入的长度超过了4096那他只能接收前4096个字符。
下面的代码中，reader1函数完美呈现NewReader的缺陷，reader2，reader3函数，则是对这一缺陷的两种解决方式，他们都完美的解决了只能读4096 的问题。
代码较长，但重复内容已标注，希望读者稍花时间阅读一下~
package main
import (
"bufio"
"fmt"
"os"
)
func main() {
// todo 先向一个文件中写入一行内容，这行内容包含10240个字符。
write()
// todo 然后分别用三种方法读取这一行内容，并输出读取到的数据的长度
read1()
read2()
read3()
}
//在该文件夹下新建一个名字为1.txt的文件，并向其中写入10240个字符1。
func write() {
file, _ := os.OpenFile("1.txt", os.O_CREATE, 0644)
writer := bufio.NewWriter(file)
str := ""
for i := 0; i < 10240; i++ {
str += "1"
}
writer.WriteString(str)
writer.Flush()
file.Close()
}
func read1() {
file, _ := os.OpenFile("1.txt", os.O_CREATE, 0644)
reader := bufio.NewReader(file) //使用NewReader创建读取器
content, _, _ := reader.ReadLine()
fmt.Println("reader1 读取长度：", len(content))
file.Close()
}
func read2() {
file, _ := os.OpenFile("1.txt", os.O_CREATE, 0644)
reader := bufio.NewReaderSize(file, 10240) //相比reader1仅改变这一行，10240指一次读取的最大长度，如不满足需求可以设置更大。
content, _, _ := reader.ReadLine()
fmt.Println("reader2 读取长度：", len(content))
file.Close()
}
func read3() {
file, _ := os.OpenFile("1.txt", os.O_CREATE, 0644)
reader := bufio.NewReader(file)
content, _ := reader.ReadString('\n') //相比reader1仅改变这一行，直到读取到换行符为止
fmt.Println("reader3 读取长度：", len(content))
file.Close()
} 输出： reader1 读取长度： 4096reader2 读取长度： 10240reader3 读取长度： 10240
可见，reader1确实读不全。其实细翻代码会发现，bufio.NewReader(rd io.Reader) 函数内部调用了 NewReaderSize(rd, defaultBufSize)，而这个defaultBufSize的值就是4096。懂我意思吧~~
欢迎有疑问的小伙伴留言，我们一起讨论~
#秋招##go语言#
华为工作强度 838人发布
投递腾讯等公司10个岗位 >