读取和写入文件是许多 Go 程序所需的常见任务。首先,我们将介绍一些读取文件的示例。
|
|
|
package main
|
|
import (
"bufio"
"fmt"
"io"
"os"
)
|
读取文件需要检查大多数调用的错误。以下辅助函数将简化我们的错误检查。
|
func check(e error) {
if e != nil {
panic(e)
}
}
|
|
func main() {
|
也许最基本的读取文件任务是将文件的全部内容读入内存。
|
dat, err := os.ReadFile("/tmp/dat")
check(err)
fmt.Print(string(dat))
|
您通常希望对如何读取文件以及读取文件的哪些部分有更多控制。对于这些任务,首先通过Open 打开文件以获取os.File 值。
|
f, err := os.Open("/tmp/dat")
check(err)
|
从文件开头读取一些字节。允许最多读取 5 个字节,但也要注意实际读取了多少个字节。
|
b1 := make([]byte, 5)
n1, err := f.Read(b1)
check(err)
fmt.Printf("%d bytes: %s\n", n1, string(b1[:n1]))
|
您也可以Seek 到文件中已知的位置并从那里Read 。
|
o2, err := f.Seek(6, io.SeekStart)
check(err)
b2 := make([]byte, 2)
n2, err := f.Read(b2)
check(err)
fmt.Printf("%d bytes @ %d: ", n2, o2)
fmt.Printf("%v\n", string(b2[:n2]))
|
其他寻址方法相对于当前光标位置,
|
_, err = f.Seek(4, io.SeekCurrent)
check(err)
|
以及相对于文件末尾。
|
_, err = f.Seek(-10, io.SeekEnd)
check(err)
|
io 包提供了一些可能对文件读取有用的函数。例如,上面类似的读取可以使用ReadAtLeast 更稳健地实现。
|
o3, err := f.Seek(6, io.SeekStart)
check(err)
b3 := make([]byte, 2)
n3, err := io.ReadAtLeast(f, b3, 2)
check(err)
fmt.Printf("%d bytes @ %d: %s\n", n3, o3, string(b3))
|
没有内置的倒带功能,但Seek(0, io.SeekStart) 可以实现此功能。
|
_, err = f.Seek(0, io.SeekStart)
check(err)
|
bufio 包实现了一个缓冲读取器,它可能对提高许多小读取的效率以及它提供的其他读取方法很有用。
|
r4 := bufio.NewReader(f)
b4, err := r4.Peek(5)
check(err)
fmt.Printf("5 bytes: %s\n", string(b4))
|
完成后关闭文件(通常,这将在Open 之后立即使用defer 进行调度)。
|
f.Close()
}
|