让我们列出在 Go 中读取和写入文件的所有方式的 Go 1 兼容列表。
因为文件 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
非常有用,因为它为我们提供了 a nice API 以方便灵活地读写,同时它在后台处理缓冲。
注意:以下代码适用于较旧的 Go 版本(Go 1.15 及之前版本)。事情变了。对于新方法,请查看 this answer。
使用 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)
}
}
非常简单!但只有在您确定不处理大文件时才使用它。
这是一个很好的版本:
package main
import (
"io/ioutil";
)
func main() {
contents,_ := ioutil.ReadFile("plikTekstowy.txt")
println(string(contents))
ioutil.WriteFile("filename", contents, 0644)
}
0x777
是假的。无论如何,它应该更像 0644
或 0755
(八进制,而不是十六进制)。
使用 io.Copy
package main
import (
"io"
"log"
"os"
)
func main () {
// open files r and w
r, err := os.Open("input.txt")
if err != nil {
panic(err)
}
defer r.Close()
w, err := os.Create("output.txt")
if err != nil {
panic(err)
}
defer w.Close()
// do the actual work
n, err := io.Copy(w, r)
if err != nil {
panic(err)
}
log.Printf("Copied %v bytes\n", n)
}
如果您不想重新发明轮子,io.Copy
和 io.CopyN
可能会为您服务。如果您使用 io.Copy 函数check the source,那么它只不过是封装在 Go 库中的 Mostafa 解决方案之一(实际上是“基本”解决方案)。不过,他们使用的缓冲区比他大得多。
io.Copy(w, r)
之后使用 w.Sync()
io.Copy()
只会写入您提供给它的数据,因此如果现有文件有更多内容,它不会被删除,这可能会导致文件损坏。
w, err := os.Create("output.txt")
,您所描述的就不会发生,因为“创建创建或截断命名文件。如果文件已存在,则将其截断。” golang.org/pkg/os/#Create。
新的方法
从 Go 1.16 开始,使用 os.ReadFile 将文件加载到内存中,并使用 os.WriteFile
从内存中写入文件(ioutil.ReadFile 现在调用 os.ReadFile
)。
小心使用 os.ReadFile
,因为它将整个文件读入内存。
package main
import "os"
func main() {
b, err := os.ReadFile("input.txt")
if err != nil {
log.Fatal(err)
}
// `b` contains everything your file has.
// This writes it to the Standard Out.
os.Stdout.Write(b)
// You can also write it to a file as a whole.
err = os.WriteFile("destination.txt", b, 0644)
if err != nil {
log.Fatal(err)
}
}
使用较新的 Go 版本,读取/写入文件很容易。从文件中读取:
package main
import (
"fmt"
"io/ioutil"
)
func main() {
data, err := ioutil.ReadFile("text.txt")
if err != nil {
return
}
fmt.Println(string(data))
}
要写入文件:
package main
import "os"
func main() {
file, err := os.Create("text.txt")
if err != nil {
return
}
defer file.Close()
file.WriteString("test\nhello")
}
这将覆盖文件的内容(如果不存在则创建一个新文件)。
[]byte
是全部或部分字节数组的切片(类似于子字符串)。将切片视为具有隐藏指针字段的值结构,供系统定位和访问数组(切片)的全部或部分,以及切片长度和容量的字段,您可以使用 {2 } 和 cap()
函数。
这是一个适合您的入门工具包,它读取并打印二进制文件;您将需要更改 inName
文字值以引用系统上的一个小文件。
package main
import (
"fmt";
"os";
)
func main()
{
inName := "file-rw.bin";
inPerm := 0666;
inFile, inErr := os.Open(inName, os.O_RDONLY, inPerm);
if inErr == nil {
inBufLen := 16;
inBuf := make([]byte, inBufLen);
n, inErr := inFile.Read(inBuf);
for inErr == nil {
fmt.Println(n, inBuf[0:n]);
n, inErr = inFile.Read(inBuf);
}
}
inErr = inFile.Close();
}
if
块之外
尝试这个:
package main
import (
"io";
)
func main() {
contents,_ := io.ReadFile("filename");
println(string(contents));
io.WriteFile("filename", contents, 0644);
}
您也可以使用 fmt
包:
package main
import "fmt"
func main(){
file, err := os.Create("demo.txt")
if err != nil {
panic(err)
}
defer file.Close()
fmt.Fprint(file, name)
}
只需查看文档,您似乎应该只声明一个 []byte 类型的缓冲区并将其传递给 read,然后它将读取那么多字符并返回实际读取的字符数(以及一个错误)。
The docs 说
Read 从文件中读取最多 len(b) 个字节。它返回读取的字节数和错误(如果有)。 EOF 由 err 设置为 EOF 的零计数发出信号。
那不行吗?
编辑:另外,我认为您可能应该使用在 bufio 包中声明的 Reader/Writer 接口,而不是使用 os 包。
Read 方法接受一个字节参数,因为这是它将读取到的缓冲区。在某些圈子中,这是一个常见的成语,当你考虑它时,它是有道理的。
通过这种方式,您可以确定读取器将读取多少字节并检查返回以查看实际读取了多少字节并适当地处理任何错误。
正如其他人在他们的答案中指出的那样, bufio 可能是您想要从大多数文件中读取的内容。
我将添加另一个提示,因为它真的很有用。从文件中读取一行最好不要使用 ReadLine 方法,而是使用 ReadBytes 或 ReadString 方法。
不定期副业成功案例分享
panic("error in writing")
) 中的额外检查不是必需的。