Go语言入门【四】:源码学习-io

介绍

package io主要包含以下文件:

  • io.go 基础的io操作,如读写,较底层
  • multi.go 多个Reader封装成一个
  • pipe.go
  • ioutil 常用操作
    • ioutil.go 封装了ioutil
    • tempfile.go 有关创建临时文件

Reader & Writer

这两个接口是用途最广的接口之一,所以重点学习一下,后续会用在os,net等各种常用包。

io.Reader

type Reader interface {
    Read(p []byte) (n int, err error)
}

io.Writer

type Writer interface {
    Write(p []byte) (n int, err error)
}

具体例子,以strings.Reader为例,代码:

package strings
...

type Reader struct {
    s        string
    i        int64 // current reading index
    prevRune int   // index of previous rune; or < 0
}

func (r *Reader) Read(b []byte) (n int, err error) {
    if r.i >= int64(len(r.s)) {
        return 0, io.EOF
    }
    r.prevRune = -1
    n = copy(b, r.s[r.i:])
    r.i += int64(n)
    return
}

可见该类满足io.Reader契约,所以使用时可以这样(清晰起见,以显式类型编码):

var strReader io.Reader = strings.NewReader("hello")
if _, err := io.Copy(os.Stdout, r); err != nil {
    log.Fatal(err)
}

从源码来看,
第一,实际上Reader就是封装了序列化数据的一个容器。这一点从命名上来讲可能不是那么明显,从名字上看,可能Reader只是一个调用者,需要从别的地方去『读取』数据,实际上大部分Reader本身包括了数据。
第二,Reader类似我们现在的优酷土豆播放器,能够记住上一次播放的位置,每次去观看都从上次看的接着看。

Copy

这个方法从前面也能看到十分常用
func Copy(dst Writer, src Reader) (written int64, err error)
作用是将src的内容写入到dst之中,

注意,在buildin中有一个copy方法与之类似,
func copy(dst, src []Type) int
只不过是处理[]byte类型。

查看io.Copy方法的源码,我们发现其中调用了
func copyBuffer(dst Writer, src Reader, buf []byte) (written int64, err error)
并默认创建了一个buf,代码如下

if buf == nil {
    buf = make([]byte, 32*1024)
}

然后,内部调用了src.Read(buf),并在返回不为EOF的情况下将buf内容逐渐写入dst。加上一些错误处理。

Copy方法还有一个兄弟:io.CopyN:
func CopyN(dst Writer, src Reader, n int64) (written int64, err error)
最多拷贝n个字节。
与CopyN相反,还有一个方法:
func ReadAtLeast(r Reader, buf []byte, min int) (n int, err error)
该方法至少读取min个字节直到全部读完,否则将抛异常。
这个方法主要用途是保证byte的内容被完整的读取,见ReadFull:

func ReadFull(r Reader, buf []byte) (n int, err error) {
    return ReadAtLeast(r, buf, len(buf))
}

io/ioutil

相比之下io中的偏底层实现,ioutil的方法更加实用,常用方法有:

func ReadAll(r io.Reader) ([]byte, error)
将reader中的所有内容读取出来,返回字节数组,不需要自己提供字节数组

func ReadFile(filename string) ([]byte, error)
将一个文件名传入,返回内容字节数组

func WriteFile(filename string, data []byte, perm os.FileMode)error
将data写入文件,如果文件不存在,则以perm的模式创建文件

func ReadDir(dirname string) ([]os.FileInfo, error)
返回目录下的所有文件信息(个人认为这个方法出现得有点奇怪,应该在os包更合理)

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 227,022评论 6 528
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 97,760评论 3 412
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 174,687评论 0 373
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 62,318评论 1 307
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 71,112评论 6 405
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 54,645评论 1 320
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 42,724评论 3 435
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 41,885评论 0 285
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 48,387评论 1 330
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 40,368评论 3 354
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 42,506评论 1 365
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 38,063评论 5 355
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 43,759评论 3 343
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 34,150评论 0 25
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 35,414评论 1 281
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 51,080评论 3 386
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 47,494评论 2 370

推荐阅读更多精彩内容