Go文件读取

go文件读取

一次性读取

  • 适用于小文件
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    package main

    import (
    	"os"
    	"io/ioutil"
    	"fmt"
    )

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

    func main()  {
    	r , _ := ReadAll("1.go")
    	fmt.Println(string(r))
    }

    分块读取

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    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("1.go", 10000, processBlock)
    }

逐行读取

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
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("1.go", processLine)
}