GO语言常用的文件读取方式

本文实例讲述了GO语言常用的文件读取方式。分享给大家供大家参考。具体分析如下：

Golang 的文件读取方法很多，刚上手时不知道怎么选择，所以贴在此处便后速查。

一次性读取

小文件推荐一次性读取，这样程序更简单，而且速度最快。

func ReadAll(filePth string) ([]byte, error) {

 f, err := os.Open(filePth)

 if err != nil {

  return nil, err

 }

 return ioutil.ReadAll(f)

}

分块读取

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

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)

}

希望本文所述对大家的GO语言程序设计有所帮助。