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

第golangbufio包中Write方法的深入讲解前言

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

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

packagemain

import(

"os"

"fmt"

funcmain(){

file,err:=os.OpenFile("a.txt",os.O_CREATE|os.O_RDWR,0666)

iferr!=nil{

fmt.Println(err)

deferfile.Close()

content:=[]byte("helloworld!")

if_,err=file.Write(content);err!=nil{

fmt.Println(err)

fmt.Println("writefilesuccessful")

}

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

packagemain

import(

"os"

"fmt"

"bufio"

funcmain(){

file,err:=os.OpenFile("a.txt",os.O_CREATE|os.O_RDWR,0666)

iferr!=nil{

fmt.Println(err)

deferfile.Close()

content:=[]byte("helloworld!")

newWriter:=bufio.NewWriter(file)

if_,err=newWriter.Write(content);err!=nil{

fmt.Println(err)

fmt.Println("writefilesuccessful")

}

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

func(b*Writer)Write(p[]byte)(nnint,errerror){

forlen(p)b.Available()b.err==nil{

varnint

ifb.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:]

ifb.err!=nil{

returnnn,b.err

n:=copy(b.buf[b.n:],p)

b.n+=n

nn+=n

returnnn,nil

}

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

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

packagemain

import(

"os"

"fmt"

"bufio"

funcmain(){

file,err:=os.OpenFile("./a.txt",os.O_CREATE|os.O_RDWR,0666)

iferr!=nil{

fmt.Println(err)

deferfile.Close()

content:=[]byte("helloworld!")

newWriter:=bufio.NewWriterSize(file,1024)

if_,err=newWriter.Write(content);err!=nil{

fmt.Println(err)

iferr=newWriter.Flush();err!=nil{

fmt.Println(err)

fmt.Println("wr