前言
bufio包實現(xiàn)了帶緩沖的I/O，它封裝了io.Reader和io.Writer對象，然后創(chuàng)建了另外一種對象（Reader或Writer）實現(xiàn)了相同的接口，但是增加了緩沖功能。
首先來看沒有緩沖功能的Write(os包中)方法，它會將數(shù)據(jù)直接寫到文件中。
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方法例子，當下面的程序執(zhí)行后是看不到寫入的數(shù)據(jù)的。
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")
}
為什么會在文件中看不到寫入的數(shù)據(jù)呢，我們來看看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方法首先會判斷寫入的數(shù)據(jù)長度是否大于設(shè)置的緩沖長度，如果小于，則會將數(shù)據(jù)copy到緩沖中；當數(shù)據(jù)長度大于緩沖長度時，如果數(shù)據(jù)特別大，則會跳過copy環(huán)節(jié)，直接寫入文件。其他情況依然先會將數(shù)據(jù)拷貝到緩沖隊列中，然后再將緩沖中的數(shù)據(jù)寫入到文件中。
所以上面的錯誤示例，只要給其添加Flush()方法，將緩存的數(shù)據(jù)寫入到文件中。
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")
}
總結(jié)
以上就是這篇文章的全部內(nèi)容了，希望本文的內(nèi)容對大家的學習或者工作具有一定的參考學習價值，如果有疑問大家可以留言交流，謝謝大家對腳本之家的支持。
您可能感興趣的文章:- 詳解golang中bufio包的實現(xiàn)原理
- golang中bufio.SplitFunc的深入理解
