当前位置 博文首页 > GO语言常用的文件读取方式

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

    作者:admin 时间:2021-02-09 15:04

    本文实例讲述了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)
    }

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

    js
    下一篇:没有了