Golang文件写入的四种方式

Golang 中关于文件写入的方法很多

  1. 简单覆盖式文件写入
  2. 常规文件写入
  3. 带有缓冲区的文件写入
  4. 复制操作的文件写入

1. 简单覆盖式文件写入

特点 :

  1. 操作简单一个函数完成数据写入
  2. 新内容覆盖旧的内容
  3. 操作的文件不存在的时候会自动创建

使用Golang的标准包 io/ioutil

函数参数说明 :

filename 操作的文件名

data 写入的内容

perm 文件不存在时创建文件并赋予的权限,例如 : 0666

func WriteFile(filename string, data []byte, perm os.FileMode) error

WriteFile 内部实现

func WriteFile(filename string, data []byte, perm os.FileMode) error {
    f, err := os.OpenFile(filename, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, perm)
    if err != nil {
        return err
    }
    n, err := f.Write(data)
    if err == nil && n < len(data) {
        err = io.ErrShortWrite
    }
    if err1 := f.Close(); err == nil {
        err = err1
    }
    return err
}

看得出来 WriteFile 的本质是对 os 包中 File 的封装

2. 常规文件写入

特点 :

  1. 文件写入灵活 ,对文件的操作更强

操作流程 : 打开文件(或者创建文件) ,写入内容 ,关闭文件

使用Golang标准包 os 的 type File 部分 , 其中常用到的方法和函数如下

// 可读可写模式创建文件
func Create(name string) (file *File, err error)
// 只读模式打文件
func Open(name string) (file *File, err error)
// 通用的文件打开函数(综合和 Create 和 Open的作用)
// OpenFile第二个参数 flag 有如下可选项
//    O_RDONLY  文件以只读模式打开
//    O_WRONLY  文件以只写模式打开
//    O_RDWR   文件以读写模式打开
//    O_APPEND 追加写入
//    O_CREATE 文件不存在时创建
//    O_EXCL   和 O_CREATE 配合使用,创建的文件必须不存在
//    O_SYNC   开启同步 I/O
//    O_TRUNC  打开时截断常规可写文件
func OpenFile(name string, flag int, perm FileMode) (file *File, err error)
// 向文件写入字节数据
func (f *File) Write(b []byte) (n int, err error)
// 向文件写入字符串
func (f *File) WriteString(s string) (ret int, err error)
// 关闭文件
func (f *File) Close() error

示例代码

func generalWrite(param string) {
    f, err := os.OpenFile(w2, os.O_RDONLY|os.O_CREATE|os.O_APPEND, 0666)
    if err != nil {
        log.Println("open file error :", err)
        return
    }
    // 关闭文件
    defer f.Close()
    // 字节方式写入
    _, err = f.Write([]byte("write : " + param))
    if err != nil {
        log.Println(err)
        return
    }
    // 字符串写入
    _, err = f.WriteString("writeString : " + param)
    if err != nil {
        log.Println(err)
        return
    }
}

3. 带有缓冲区的文件写入

特点：

  1. 先将数据写入缓存区,再由缓冲区写入文件中
  2. 根据设置缓存的大小,可以存储更多数据然后一次写入文件

数据写入的速度更快一点

使用Golang标准包 bufio 的 Writer 部分 , 其中常用到的方法和函数如下

// 新建一个Writer 采用默认缓冲大小 4096
func NewWriter(w io.Writer) *Writer
// 新建一个Writer 采用自定义缓冲大小
func NewWriterSize(w io.Writer, size int) *Writer
// 字节写入
func (b *Writer) Write(p []byte) (nn int, err error)
// 字符串写入
func (b *Writer) WriteString(s string) (int, error)
// 单字节写入
func (b *Writer) WriteByte(c byte) error
// 写入一个unicode码值（的utf-8编码）
func (b *Writer) WriteRune(r rune) (size int, err error)
// 将缓冲中的数据写入下层的io.Writer接口
func (b *Writer) Flush() error

示例代码

func bufferWrite(param string) {
    fileHandle, err := os.OpenFile(w3, os.O_RDONLY|os.O_CREATE|os.O_APPEND, 0666)
    if err != nil {
        log.Println("open file error :", err)
        return
    }
    defer fileHandle.Close()
    // NewWriter 默认缓冲区大小是 4096
    // 需要使用自定义缓冲区的writer 使用 NewWriterSize()方法
    buf := bufio.NewWriter(fileHandle)
    // 字节写入
    buf.Write([]byte("buffer Write : " + param))
    // 字符串写入
    buf.WriteString("buffer WriteString : " + param)
    // 将缓冲中的数据写入
    err = buf.Flush()
    if err != nil {
        log.Println("flush error :", err)
    }
}

4. 复制操作的文件写入

特点 :

  1. 复制出一份相同内容的数据
  2. 默认带有缓冲并且可以自己设置缓冲区大小
  3. 操作相对简单

使用Golang标准包 io 的两个函数如下

// 复制 采用系统默认缓冲区大小
func Copy(dst Writer, src Reader) (written int64, err error){}
// 复制 自定义缓冲区大小
func CopyBuffer(dst Writer, src Reader, buf []byte) (written int64, err error) {}

io 包中的 Copy 和 CopyBuffer 函数的核心都是调用了一个内部函数 copyBuffer , 我们看一下它的源代码

io.go

// copyBuffer is the actual implementation of Copy and CopyBuffer.
// if buf is nil, one is allocated.
func copyBuffer(dst Writer, src Reader, buf []byte) (written int64, err error) {
    // If the reader has a WriteTo method, use it to do the copy.
    // Avoids an allocation and a copy.
    if wt, ok := src.(WriterTo); ok {
        return wt.WriteTo(dst)
    }
    // Similarly, if the writer has a ReadFrom method, use it to do the copy.
    if rt, ok := dst.(ReaderFrom); ok {
        return rt.ReadFrom(src)
    }
    if buf == nil {
        size := 32 * 1024
        if l, ok := src.(*LimitedReader); ok && int64(size) > l.N {
            if l.N < 1 {
                size = 1
            } else {
                size = int(l.N)
            }
        }
        buf = make([]byte, size)
    }
    for {
        nr, er := src.Read(buf)
        if nr > 0 {
            nw, ew := dst.Write(buf[0:nr])
            if nw > 0 {
                written += int64(nw)
            }
            if ew != nil {
                err = ew
                break
            }
            if nr != nw {
                err = ErrShortWrite
                break
            }
        }
        if er != nil {
            if er != EOF {
                err = er
            }
            break
        }
    }
    return written, err
}

使用代码示例

func fileCopy(dst, src string) {
    start := time.Now()
    dstWriter, err := os.Create(dst)
    if err != nil {
        log.Println("create file error :", err)
        return
    }
    srcReader, err := os.Open(src)
    if err != nil {
        log.Println("open file error :", err)
        return
    }
    // Copy()函数其实是调用了
    // io包中私有函数copyBuffer() 默认缓冲区是32K
    // 与Copy()函数功能一致的是CopyBuffer()可以设置缓冲区
    c, err := io.Copy(dstWriter, srcReader)
    if err != nil {
        log.Println("copy file error :", err)
        return
    }
    fmt.Println("copy spend :", time.Now().Sub(start), c)
}

参考资料 :

- [1] Golang文档

©著作权归作者所有,转载或内容合作请联系作者
  • 人面猴
    序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
    沈念sama阅读 159,716评论 4赞 364
  • 死咒
    序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
    沈念sama阅读 67,558评论 1赞 294
  • 救了他两次的神仙让他今天三更去死
    文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
    开封第一讲书人阅读 109,431评论 0赞 244
  • 道士缉凶录：失踪的卖姜人
    文/不坏的土叔 我叫张陵，是天一观的道长。 经常有香客问我，道长，这世上最难降的妖魔是什么？ 我笑而不...
    开封第一讲书人阅读 44,127评论 0赞 209
  • 港岛之恋（遗憾婚礼）
    正文 为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
    茶点故事阅读 52,511评论 3赞 287
  • 恶毒庶女顶嫁案：这布局不是一般人想出来的
    文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
    开封第一讲书人阅读 40,692评论 1赞 222
  • 城市分裂传说
    那天，我揣着相机与录音，去河边找鬼。 笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。 我是一名探鬼主播，决...
    沈念sama阅读 31,915评论 2赞 313
  • 双鸳鸯连环套：你想象不到人心有多黑
    文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
    开封第一讲书人阅读 30,664评论 0赞 202
  • 万荣杀人案实录
    序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
    沈念sama阅读 34,412评论 1赞 246
  • 护林员之死
    正文 独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,616评论 2赞 245
  • 白月光启示录
    正文 我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,105评论 1赞 260
  • 活死人
    序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
    沈念sama阅读 28,424评论 2赞 254
  • 日本核电站爆炸内幕
    正文 年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
    茶点故事阅读 33,098评论 3赞 238
  • 男人毒药：我在死后第九天来索命
    文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,096评论 0赞 8
  • 一桩弑父案，背后竟有这般阴谋
    文/苍兰香墨 我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,869评论 0赞 197
  • 情欲美人皮
    我被黑心中介骗来泰国打工， 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。 一个月前我还...
    沈念sama阅读 35,748评论 2赞 276
  • 代替公主和亲
    正文 我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
    茶点故事阅读 35,641评论 2赞 271