Давайте создадим Go 1-совместимый список всех способов чтения и записи файлов в Go.
Поскольку файловый API недавно изменился и большинство других ответов не работают с Go 1. Они также пропускают bufio, что важно, ИМХО.
В следующих примерах я копирую файл, читая его и записывая в файл назначения.
Начните с основ
package main
import (
"io"
"os"
)
func main() {
// open input file
fi, err := os.Open("input.txt")
if err != nil {
panic(err)
}
// close fi on exit and check for its returned error
defer func() {
if err := fi.Close(); err != nil {
panic(err)
}
}()
// open output file
fo, err := os.Create("output.txt")
if err != nil {
panic(err)
}
// close fo on exit and check for its returned error
defer func() {
if err := fo.Close(); err != nil {
panic(err)
}
}()
// make a buffer to keep chunks that are read
buf := make([]byte, 1024)
for {
// read a chunk
n, err := fi.Read(buf)
if err != nil && err != io.EOF {
panic(err)
}
if n == 0 {
break
}
// write a chunk
if _, err := fo.Write(buf[:n]); err != nil {
panic(err)
}
}
}
Здесь я использовал os.Open и os.Create, которые являются удобными обертками для os.OpenFile. Нам обычно не нужно звонить OpenFile напрямую.
Обратите внимание на лечение EOF. Read пытается заполнить buf при каждом вызове и возвращает io.EOF как ошибку, если при этом он достигает конца файла. В этом случае buf все еще будет хранить данные. Последующие вызовы Read возвращают ноль в качестве количества прочитанных байтов и то же io.EOF в качестве ошибки. Любая другая ошибка приведет к панике.
Использование bufio
package main
import (
"bufio"
"io"
"os"
)
func main() {
// open input file
fi, err := os.Open("input.txt")
if err != nil {
panic(err)
}
// close fi on exit and check for its returned error
defer func() {
if err := fi.Close(); err != nil {
panic(err)
}
}()
// make a read buffer
r := bufio.NewReader(fi)
// open output file
fo, err := os.Create("output.txt")
if err != nil {
panic(err)
}
// close fo on exit and check for its returned error
defer func() {
if err := fo.Close(); err != nil {
panic(err)
}
}()
// make a write buffer
w := bufio.NewWriter(fo)
// make a buffer to keep chunks that are read
buf := make([]byte, 1024)
for {
// read a chunk
n, err := r.Read(buf)
if err != nil && err != io.EOF {
panic(err)
}
if n == 0 {
break
}
// write a chunk
if _, err := w.Write(buf[:n]); err != nil {
panic(err)
}
}
if err = w.Flush(); err != nil {
panic(err)
}
}
bufio просто выступает здесь в качестве буфера, потому что мы не имеем ничего общего с данными. В большинстве других ситуаций (особенно с текстовыми файлами) bufio очень полезен, предоставляя нам прекрасный API для простого и гибкого чтения и записи, а также для буферизации за сценой.
Использование ioutil
package main
import (
"io/ioutil"
)
func main() {
// read the whole file at once
b, err := ioutil.ReadFile("input.txt")
if err != nil {
panic(err)
}
// write the whole body at once
err = ioutil.WriteFile("output.txt", b, 0644)
if err != nil {
panic(err)
}
}
Легко, как пирог! Но используйте его, только если вы уверены, что не имеете дело с большими файлами.