並發讀取檔案：建立 goroutine 並發讀取資料區塊，使用管道通訊協調；並發寫入檔案：使用 goroutine 寫入數據，用互斥鎖保護寫入操作。
如何在 Golang 中並發讀取和寫入檔案？
並發處理檔案 I/O 是提升 Golang 程式效能的常用技術。透過並發訪問，我們可以同時讀取和寫入數據，而不是等待一個操作完成再開始另一個操作。本文將展示如何在 Golang 中使用 goroutine 並發讀取和寫入檔案。
並發讀取
為了並發讀取文件，我們可以建立多個 goroutine，每個 goroutine 從文件讀取資料區塊。為了協調這些 goroutine，我們將使用管道進行通訊。
package main
import (
"bufio"
"os"
"fmt"
"sync"
"time"
)
func readFile(filePath string, wg *sync.WaitGroup, ch chan<- string) {
defer wg.Done()
file, err := os.Open(filePath)
if err != nil {
fmt.Println(err)
return
}
defer file.Close()
scanner := bufio.NewScanner(file)
for scanner.Scan() {
ch <- scanner.Text()
}
}
func main() {
// 创建一个容量为 10 的管道
ch := make(chan string, 10)
// 等待组用于跟踪 goroutine
var wg sync.WaitGroup
filePath := "filename.txt"
for i := 0; i < 10; i++ {
wg.Add(1)
go readFile(filePath, &wg, ch)
}
// 从管道中读取数据，并停止管道以等待所有 goroutine 完成
go func() {
wg.Wait()
close(ch)
}()
for data := range ch {
fmt.Println(data)
}
// 等待管道数据完全输出
time.Sleep(1 * time.Second)
}並發寫入
對於並發寫入文件，我們使用類似的方法，但使用goroutine 寫入數據，並使用互斥鎖確保獨佔存取寫入操作。
package main
import (
"os"
"fmt"
"sync"
)
func writeFile(filePath string, wg *sync.WaitGroup, data <-chan string) {
defer wg.Done()
// 使用互斥锁保护文件写入
lock := new(sync.Mutex)
file, err := os.OpenFile(filePath, os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0666)
if err != nil {
fmt.Println(err)
return
}
defer file.Close()
for dataString := range data {
lock.Lock()
_, err = file.WriteString(dataString + "\n")
if err != nil {
fmt.Println(err)
continue
}
lock.Unlock()
}
}
func main() {
// 创建一个容量为 10 的管道
ch := make(chan string, 10)
// 等待组用于跟踪 goroutine
var wg sync.WaitGroup
filePath := "filename.txt"
for i := 0; i < 10; i++ {
wg.Add(1)
go writeFile(filePath, &wg, ch)
}
// 向管道写入数据，并停止管道以等待所有 goroutine 完成
go func() {
for i := 0; i < 100; i++ {
ch <- fmt.Sprintf("Data point %d\n", i)
}
close(ch)
}()
wg.Wait()
}以上是如何在 Golang 中並發讀取和寫入檔案？的詳細內容。更多資訊請關注PHP中文網其他相關文章！
