readfile.md

读文件

在日常开发中我们少不了对文件读取，今天我们就从三部分来讲解：全量读，带缓冲区读，任意位置读。

全量读

我们先来看看一个简单的例子：

package main

import (
	"fmt"
	"io/ioutil"
)

func main() {
	dat, err := ioutil.ReadFile("./main.go")
	fmt.Println(err)
	fmt.Println(string(dat))
}

运行程序可以打印整个 main.go 文件内容，如果我们将 ./main.go 修改为 ./main.go1，程序将出现 no such file or directory 的错误， 所以在文件读取的时候一定要注意检查 err。

带缓冲区读

package main

import (
	"fmt"
	"os"
)

func main() {
	f, _ := os.Open("./main.go")
	defer f.Close()

	buf := make([]byte, 16)
	f.Read(buf)

	fmt.Println(string(buf))
}

运行程序会输出 main.go 的前 16 个字节内容，具体为：

package main

im

任意位置读

有些时候我们想在一个文件特定地方读取特定长度的内容，那我们有什么方法可以使用呢？

第一种： f.Seek + f.Read

package main

import (
	"fmt"
	"os"
)

func main() {
	f, _ := os.Open("./main.go")
	defer f.Close()

	b1 := make([]byte, 2)

	f.Seek(5, 0)
	f.Read(b1)
	fmt.Println(string(b1))
}

运行代码输出结果为：

ge

第二种：使用 f.ReadAt

package main

import (
	"fmt"
	"os"
)

func main() {
	f, _ := os.Open("./main.go")
	defer f.Close()

	b1 := make([]byte, 2)

	f.ReadAt(b1, 5)
	fmt.Println(string(b1))
}

运行结果同样为：

ge

但注意：

第一种方式是非并发安全的，例如：

package main

import (
	"fmt"
	"os"
	"time"
)

func main() {
	f, _ := os.Open("./main.go")
	defer f.Close()

	for i := 0; i < 5; i++ {
		go func() {
			b1 := make([]byte, 2)

			f.Seek(5, 0)
			f.Read(b1)
			fmt.Println(string(b1))

			f.Seek(2, 0)
			f.Read(b1)
			fmt.Println(string(b1))
		}()
	}

	time.Sleep(time.Second)
}

输出结果为：

ge
ge
ge
ck
ck
ai
ck
 m
ck
ck

第二种 f.ReadAt 是并发安全的，例如：

package main

import (
	"fmt"
	"os"
	"time"
)

func main() {
	f, _ := os.Open("./main.go")
	defer f.Close()

	for i := 0; i < 5; i++ {
		go func() {
			b1 := make([]byte, 2)
			f.ReadAt(b1, 5)
			fmt.Println(string(b1))

			f.ReadAt(b1, 2)
			fmt.Println(string(b1))
		}()
	}

	time.Sleep(time.Second)
}

输出结果为：

ge
ge
ck
ck
ge
ge
ck
ck
ge
ck

项目实战

实战项目一： 如何使用 buf 实现 ioutil.ReadFile 类似效果：

package main

import (
	"fmt"
	"io"
	"os"
)

func main() {
	f, _ := os.Open("./main.go")
	defer f.Close()

	content := make([]byte, 0)
	buf := make([]byte, 16)

	for {
		n, err := f.Read(buf)
		if err == io.EOF {
			break
		}

		if n == 16 {
			content = append(content, buf...)
		} else {
			content = append(content, buf[0:n]...)
		}
	}

	fmt.Println(string(content))
}

实战项目二： 使用 bufio 实现行统计：

package main

import (
	"bufio"
	"fmt"
	"io"
	"os"
)

func main() {
	f, _ := os.Open("./main.go")
	defer f.Close()

	br := bufio.NewReader(f)
	totalLine := 0

	for {
		_, isPrefix, err := br.ReadLine()

		if !isPrefix {
			totalLine += 1
		}

		if err == io.EOF {
			break
		}
	}

	fmt.Println("total lines is:", totalLine)
}

运行结果为：

total lines is: 31