使用 Go 写入文件
在本教程中，我们将学习如何使用 Go 将数据写入文件。我们还将学习如何同时写入文件。
本教程有以下部分
- 将字符串写入文件
- 将字节写入文件
- 逐行将数据写入文件
- 附加到文件
- 同时写入文件
由于 Playground 不支持文件操作，请在本地系统中运行本教程的所有程序。
将字符串写入文件
最常见的文件写入操作之一是将字符串写入文件。这很简单。它由以下步骤组成。
- 创建文件
- 将字符串写入文件
让我们立即开始讨论代码。
package main
import (
"fmt"
"os"
)
func main() {
f, err := os.Create("test.txt")
if err != nil {
fmt.Println(err)
return
}
l, err := f.WriteString("Hello World")
if err != nil {
fmt.Println(err)
f.Close()
return
}
fmt.Println(l, "bytes written successfully")
err = f.Close()
if err != nil {
fmt.Println(err)
return
}
}
上面程序第 9 行中的函数创建了一个名为 test.txt 的文件。如果具有该名称的文件已存在，则 create 函数将截断该文件。此函数返回 File 描述符。https://pkg.go.dev/os#File)。
在第 14 行，我们使用 方法将字符串Hello World写入文件WriteString。此方法返回写入的字节数和错误（如果有）。
最后，我们关闭第 1 行的文件。21.
上面的程序会打印
11 bytes written successfully
您可以找到在执行该程序的目录中创建的名为test.txt的文件。如果使用任何文本编辑器打开该文件，您可以发现它包含文本Hello World。
将字节写入文件
将字节写入文件与将字符串写入文件非常相似。我们将使用 Write 方法将字节写入文件。以下程序将字节切片写入文件。
package main
import (
"fmt"
"os"
)
func main() {
f, err := os.Create("/home/naveen/bytes")
if err != nil {
fmt.Println(err)
return
}
d2 := []byte{104, 101, 108, 108, 111, 32, 98, 121, 116, 101, 115}
n2, err := f.Write(d2)
if err != nil {
fmt.Println(err)
f.Close()
return
}
fmt.Println(n2, "bytes written successfully")
err = f.Close()
if err != nil {
fmt.Println(err)
return
}
}
在上面的程序中，在第 15 行中，我们使用 Write 方法将字节切片写入目录中命名的文件。您可以将此目录更改为其他目录。其余程序是不言自明的。该程序将打印并创建一个名为 .打开文件，您可以看到它包含文本
逐行将字符串写入文件
另一种常见的文件操作是需要将字符串逐行写入文件。在本节中，我们将编写一个程序来创建包含以下内容的文件。
Welcome to the world of Go.
Go is a compiled language.
It is easy to learn Go.
让我们立即开始讨论代码。
package main
import (
"fmt"
"os"
)
func main() {
f, err := os.Create("lines")
if err != nil {
fmt.Println(err)
f.Close()
return
}
d := []string{"Welcome to the world of Go1.", "Go is a compiled language.", "It is easy to learn Go."}
for _, v := range d {
fmt.Fprintln(f, v)
if err != nil {
fmt.Println(err)
return
}
}
err = f.Close()
if err != nil {
fmt.Println(err)
return
}
fmt.Println("file written successfully")
}
在上面程序的第 9 行中，我们创建了一个名为 lines 的新文件。在第 17 行中，我们使用 for range 循环遍历数组，并使用 Fprintln 函数将行写入文件。Fprintln 函数将 a 作为参数并附加一个新行，这正是我们想要的。运行此程序将打印，并在当前目录中创建一个文件。该文件的内容如下
Welcome to the world of Go1.
Go is a compiled language.
It is easy to learn Go.
追加到文件
在本节中，我们将在上一节中创建的文件后再附加一行。我们将在文件后面附加一行文件处理很容易。
文件必须以追加和只写模式打开。这些标志作为参数传递给 Open 函数。在追加模式下打开文件后，我们将新行添加到文件中。
package main
import (
"fmt"
"os"
)
func main() {
f, err := os.OpenFile("lines", os.O_APPEND|os.O_WRONLY, 0644)
if err != nil {
fmt.Println(err)
return
}
newLine := "File handling is easy."
_, err = fmt.Fprintln(f, newLine)
if err != nil {
fmt.Println(err)
f.Close()
return
}
err = f.Close()
if err != nil {
fmt.Println(err)
return
}
fmt.Println("file appended successfully")
}
在上面程序的第 9 行中，我们以追加和只写模式打开文件。文件打开成功后，我们在第 15 行的文件中添加一个新行。该程序将打印 .运行此程序后，文件的内容将是
Welcome to the world of Go1.
Go is a compiled language.
It is easy to learn Go.
File handling is easy.
并发写入文件
当多个goroutine同时写入文件时，我们最终会遇到竞争条件。因此，必须使用通道来协调对文件的并发写入。
我们将编写一个创建 100 个 goroutine 的程序。每个 goroutine 都会同时生成一个随机数，总共生成一百个随机数。这些随机数将被写入文件中。我们将使用以下方法解决竞争条件问题。
- 创建一个用于读取和写入生成的随机数的通道。
- 创建 100 个生产者 goroutine。每个 goroutine 都会生成一个随机数，并将该随机数写入通道。
- 创建一个消费者 goroutine，将从通道读取数据并将生成的随机数写入文件。因此，我们只有一个 goroutine 同时写入文件，从而避免竞争条件
- 完成后关闭文件。
让我们produce首先编写生成随机数的函数。
func produce(data chan int, wg *sync.WaitGroup) {
n := rand.Intn(999)
data <- n
wg.Done()
}
上面的函数生成一个随机数并将其写入通道data，然后调用Done等待组以通知其任务已完成。
现在让我们转到写入文件的函数。
func consume(data chan int, done chan bool) {
f, err := os.Create("concurrent")
if err != nil {
fmt.Println(err)
return
}
for d := range data {
_, err = fmt.Fprintln(f, d)
if err != nil {
fmt.Println(err)
f.Close()
done <- false
return
}
}
err = f.Close()
if err != nil {
fmt.Println(err)
done <- false
return
}
done <- true
}
该consume函数创建一个名为concurrent. 然后它从通道读取随机数data并写入文件。一旦它读取并写入了所有随机数，它就会写入true通道done以通知它已完成其任务。
让我们编写main函数并完成这个程序。我在下面提供了整个程序。
package main
import (
"fmt"
"math/rand"
"os"
"sync"
)
func produce(data chan int, wg *sync.WaitGroup) {
n := rand.Intn(999)
data <- n
wg.Done()
}
func consume(data chan int, done chan bool) {
f, err := os.Create("concurrent")
if err != nil {
fmt.Println(err)
return
}
for d := range data {
_, err = fmt.Fprintln(f, d)
if err != nil {
fmt.Println(err)
f.Close()
done <- false
return
}
}
err = f.Close()
if err != nil {
fmt.Println(err)
done <- false
return
}
done <- true
}
func main() {
data := make(chan int)
done := make(chan bool)
wg := sync.WaitGroup{}
for i := 0; i < 100; i++ {
wg.Add(1)
go produce(data, &wg)
}
go consume(data, done)
go func() {
wg.Wait()
close(data)
}()
d := <-done
if d {
fmt.Println("File written successfully")
} else {
fmt.Println("File writing failed")
}
}
main 函数在第 41 行创建通道，从中读取和写入随机数。第 42 行中的通道被 goroutine 用来通知它已完成其任务。第 43 行中的等待组用于等待所有 100 个 goroutines 完成随机数的生成。
第 44 行中的循环创建了 100 个 goroutine。第 49 行中的 goroutine 调用要求等待组等待所有 100 个 goroutine 完成创建随机数。之后，它关闭通道。一旦通道关闭并且 goroutine 将所有生成的随机数写入文件，它就会写入第 37 行中的通道，并且主 goroutine 被解锁并打印。
现在您可以在任何文本编辑器中并发打开文件并查看生成的 100 个随机数:)
本教程到此结束。希望您喜欢阅读。祝你有美好的一天。