Go 零基础编程入门教程

 主页   资讯   文章   代码   电子书 

读文件

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

全量读

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

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