golang bufio包中Write方法的示例分析

小編給大家分享一下golang bufio包中Write方法的示例分析，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！

前言

bufio包實(shí)現(xiàn)了帶緩沖的I/O，它封裝了io.Reader和io.Writer對(duì)象，然后創(chuàng)建了另外一種對(duì)象（Reader或Writer）實(shí)現(xiàn)了相同的接口，但是增加了緩沖功能。

首先來(lái)看沒(méi)有緩沖功能的Write(os包中)方法，它會(huì)將數(shù)據(jù)直接寫(xiě)到文件中。

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")
}

接著看一個(gè)錯(cuò)誤的使用帶緩沖的Write方法例子，當(dāng)下面的程序執(zhí)行后是看不到寫(xiě)入的數(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")
}

為什么會(huì)在文件中看不到寫(xiě)入的數(shù)據(jù)呢，我們來(lái)看看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方法首先會(huì)判斷寫(xiě)入的數(shù)據(jù)長(zhǎng)度是否大于設(shè)置的緩沖長(zhǎng)度，如果小于，則會(huì)將數(shù)據(jù)copy到緩沖中；當(dāng)數(shù)據(jù)長(zhǎng)度大于緩沖長(zhǎng)度時(shí)，如果數(shù)據(jù)特別大，則會(huì)跳過(guò)copy環(huán)節(jié)，直接寫(xiě)入文件。其他情況依然先會(huì)將數(shù)據(jù)拷貝到緩沖隊(duì)列中，然后再將緩沖中的數(shù)據(jù)寫(xiě)入到文件中。

所以上面的錯(cuò)誤示例，只要給其添加Flush()方法，將緩存的數(shù)據(jù)寫(xiě)入到文件中。

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")
}

以上是“golang bufio包中Write方法的示例分析”這篇文章的所有內(nèi)容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內(nèi)容對(duì)大家有所幫助，如果還想學(xué)習(xí)更多知識(shí)，歡迎關(guān)注億速云行業(yè)資訊頻道！