1. 程式人生 > 其它 >GO語言常用的檔案讀取方式

GO語言常用的檔案讀取方式

本文例項講述了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)
}