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 读取长度： 4096

reader2 读取长度： 10240

reader3 读取长度： 10240

可见，reader1确实读不全。其实细翻代码会发现，bufio.NewReader(rd io.Reader) 函数内部调用了 NewReaderSize(rd, defaultBufSize)，而这个defaultBufSize的值就是4096。懂我意思吧~~

Golang reader.ReadLine并不能一下读一行

全站热榜