一道Go练习题引发的思考
我像麋鹿一样在林荫中走着,为着自己的香气而发狂;夜晚是五月正中的夜晚,清风是南国的清风; 我迷路了,我游荡着,我寻求那得不到的东西,我得到了我所没有寻求的东西。 —— 泰戈尔
前言
当我在看 《Go程序设计语言》这本书第13页时,练习题1.7提到,用Copy
函数替换ReadAll
,还接了一句这样就不需要装下整个数据流的缓冲区
,这让我不得不思考,这两的差异是啥,当时想着难道一个是全部一次性读完,一个是一行一行或者一点一点的读?
,既然有疑问,那还是搞懂吧。
在搜索一番后,也没有搜索到啥有用的东西,或者可能是我的方式不对,这就很烦躁了,然后又想着学Go不就是看源码的意思嘛,随机硬着头皮打开源码,开始梳理,硬着头皮写下这篇文章,免得明天就忘记啦!
ReadAll
func ReadAll(r Reader) ([]byte, error) {
b := make([]byte, 0, 512)
for {
n, err := r.Read(b[len(b):cap(b)])
b = b[:len(b)+n]
if err != nil {
if err == EOF {
err = nil
}
return b, err
}
if len(b) == cap(b) {
// Add more capacity (let append pick how much).
b = append(b, 0)[:len(b)]
}
}
}
首先创建一个 []byte
类型,长度为0,容量为512的切片b,再直接进入一个无限循环
r.Read(b[len(b):cap(b))
,第一眼看不懂,那就查吧,查看文档,解释是Read reads data into p. It returns the number of bytes read into p.
也就是将r
读入b[len(b):cap(b)
,返回读入的字节数
一直扩展b的长度用来包含读入的数据,接着我们看到了这个函数的 return 语句,这个return的条件是,报错,或者是读完了才会return这个切片b
还可以看到,如果没有读完,但是b容量不够,还会进行扩容,直到读取完成
所以可以很清晰的知道了ReadAll
就是将一个Reader中的所有内容读取完毕后一次性返回
Copy
func Copy(dst Writer, src Reader) (written int64, err error) {
return copyBuffer(dst, src, nil)
}
func CopyBuffer(dst Writer, src Reader, buf []byte) (written int64, err error) {
if buf != nil && len(buf) == 0 {
panic("empty buffer in CopyBuffer")
}
return copyBuffer(dst, src, buf)
}
func copyBuffer(dst Writer, src Reader, buf []byte) (written int64, err error) {
// 删减代码......
for {
nr, er := src.Read(buf)
if nr > 0 {
nw, ew := dst.Write(buf[0:nr])
if nw < 0 || nr < nw {
nw = 0
if ew == nil {
ew = errInvalidWrite
}
}
}
if er != nil {
if er != EOF {
err = er
}
break
}
}
return written, err
}
起初一行一行看,没看懂,后来删了点代码,然后加上点猜,好像能理解一点了,同样有个无限循环,也有缓存区,不同的是,每次读入缓存区buf后,都会直接将缓存区的内容写入目标写入器dst.Write(buf[0:nr)
,nr同样是读入的多少就是多少(缓冲区大小是:32*1024)
结束同样是读完或者报错(其实还有其它的结束条件,删除了)
总结
- ReadAll 和 Copy 都使用了新的内存区域来存储从目标文件中读取的数据,ReadAll是存储了读到的全部数据,相当于内存直接复制了一遍,Copy是一个缓冲区,每次会读一部分,就会将读取到的字写入到目标写入器中,但是这个缓冲区大小
32*1024
- EOF用来表示对一个文件的读取成功,也就是读取到了目标文件的最后一行
- 当数据量大的时候千万别用 ReadAll,数据量小,影响不大
转载自:https://juejin.cn/post/7300812626278563859