golang bufio包中Write方法的深入讲解

前言

bufio包实现了带缓冲的i/o，它封装了io.reader和io.writer对象，然后创建了另外一种对象（reader或writer）实现了相同的接口，但是增加了缓冲功能。

首先来看没有缓冲功能的write(os包中)方法，它会将数据直接写到文件中。

package main

import (
 "os"
 "fmt"
)

func main() {
 file, err := os.openfile("a.txt", os.o_create|os.o_rdwr, 0666)
 if err != nil {
  fmt.println(err)
 }
 defer file.close()

 content := []byte("hello world!")
 if _, err = file.write(content); err != nil {
  fmt.println(err)
 }
 fmt.println("write file successful")
}

接着看一个错误的使用带缓冲的write方法例子，当下面的程序执行后是看不到写入的数据的。

package main

import (
  "os"
  "fmt"
  "bufio"
)

func main() {
  file, err := os.openfile("a.txt", os.o_create|os.o_rdwr, 0666)
  if err != nil {
    fmt.println(err)
  }
  defer file.close()

  content := []byte("hello world!")
  newwriter := bufio.newwriter(file)
  if _, err = newwriter.write(content); err != nil {
    fmt.println(err)
  }
  fmt.println("write file successful")
}

为什么会在文件中看不到写入的数据呢，我们来看看bufio中的write方法。

func (b *writer) write(p []byte) (nn int, err error){
  for len(p) > b.available() && b.err == nil {
    var n int
    if b.buffered() == 0{
      n,b.err =b.wr.write(p)
    }else {
      n = copy(b.buf[b.n:],p)
      b.n+=n
      b.flush()
    }
    nn+=n
    p=p[n:]
  }
  if b.err!=nil {
    return nn, b.err
  }
  n:= copy(b.buf[b.n:],p)
  b.n+= n
  nn+=n
  return nn,nil
}

write方法首先会判断写入的数据长度是否大于设置的缓冲长度，如果小于，则会将数据copy到缓冲中；当数据长度大于缓冲长度时，如果数据特别大，则会跳过copy环节，直接写入文件。其他情况依然先会将数据拷贝到缓冲队列中，然后再将缓冲中的数据写入到文件中。

所以上面的错误示例，只要给其添加flush()方法，将缓存的数据写入到文件中。

package main

import (
  "os"
  "fmt"
  "bufio"
)

func main() {
  file, err := os.openfile("./a.txt", os.o_create|os.o_rdwr, 0666)
  if err != nil {
    fmt.println(err)
  }
  defer file.close()

  content := []byte("hello world!")
  newwriter := bufio.newwritersize(file, 1024)
  if _, err = newwriter.write(content); err != nil {
    fmt.println(err)
  }
  if err = newwriter.flush(); err != nil {
    fmt.println(err)
  }
  fmt.println("write file successful")
}

总结

以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，如果有疑问大家可以留言交流，谢谢大家对的支持。