Go -- 讀取文件內容

Golang 的文件讀取方法很多，剛上手時不知道怎么選擇，所以貼在此處便后速查。

一次性讀取

小文件推薦一次性讀取，這樣程序更簡單，而且速度最快。

代碼如下:

func ReadAll(filePth string) ([]byte, error) {
 f, err := os.Open(filePth)
 if err != nil {
  return nil, err
 }
 

 return ioutil.ReadAll(f)
}

還有更加簡單的方法，我經常使用ioutil.ReadFile(filePth)

 

分塊讀取

可在速度和內存占用之間取得很好的平衡。

代碼如下:

package main
 import (

 "bufio"
 "io"
 "os"
)

func processBlock(line []byte) {
 os.Stdout.Write(line)
}

func ReadBlock(filePth string, bufSize int, hookfn func([]byte)) error {
 f, err := os.Open(filePth)
 if err != nil {
  return err
 }
 defer f.Close()

 buf := make([]byte, bufSize) //一次讀取多少個字節
 bfRd := bufio.NewReader(f)
 for {
  n, err := bfRd.Read(buf)
  hookfn(buf[:n]) // n 是成功讀取字節數

  if err != nil { //遇到任何錯誤立即返回，並忽略 EOF 錯誤信息
   if err == io.EOF {
    return nil
   }
   return err
  }
 }

 return nil
}

func main() {
 ReadBlock("test.txt", 10000, processBlock)
}

 

逐行讀取

逐行讀取有的時候真的很方便，性能可能慢一些，但是僅占用極少的內存空間。

代碼如下:

package main
 

import (
 "bufio"
 "io"
 "os"
)

func processLine(line []byte) {
 os.Stdout.Write(line)
}

func ReadLine(filePth string, hookfn func([]byte)) error {
 f, err := os.Open(filePth)
 if err != nil {
  return err
 }
 defer f.Close()

 bfRd := bufio.NewReader(f)
 for {
  line, err := bfRd.ReadBytes('\n')
  hookfn(line) //放在錯誤處理前面，即使發生錯誤，也會處理已經讀取到的數據。
  if err != nil { //遇到任何錯誤立即返回，並忽略 EOF 錯誤信息
   if err == io.EOF {
    return nil
   }
   return err
  }
 }
 return nil
}

func main() {
 ReadLine("test.txt", processLine)
}