欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2022-12-07 20:50:02
本文实例讲述了go语言常用的文件读取方式。分享给大家供大家参考。具体分析如下: golang 的文件读取方法很多,刚上手时不知道怎么选择,所以贴在此处便后速查。 一次性...

本文实例讲述了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语言程序设计有所帮助。