溫馨提示×

Go語言讀寫鎖怎樣實現(xiàn)同步

小樊
81
2024-11-02 16:15:31
欄目: 編程語言

在Go語言中,可以使用sync.RWMutex來實現(xiàn)讀寫鎖。sync.RWMutex是一個讀寫互斥鎖,它允許多個讀操作同時進行,但在寫操作進行時,會阻止其他讀寫操作。下面是一個簡單的示例,展示了如何使用sync.RWMutex實現(xiàn)同步:

package main

import (
	"fmt"
	"sync"
	"time"
)

type Data struct {
	value int
	mu    sync.RWMutex
}

func (d *Data) Read() {
	d.mu.RLock() // 加讀鎖
	defer d.mu.RUnlock() // 讀解鎖
	fmt.Println("Reading:", d.value)
}

func (d *Data) Write(value int) {
	d.mu.Lock() // 加寫鎖
	defer d.mu.Unlock() // 寫解鎖
	d.value = value
	fmt.Println("Writing:", value)
}

func main() {
	data := Data{value: 0}
	var wg sync.WaitGroup

	// 啟動多個讀操作
	for i := 0; i < 5; i++ {
		wg.Add(1)
		go func() {
			defer wg.Done()
			data.Read()
		}()
	}

	// 啟動一個寫操作
	wg.Add(1)
	go func() {
		defer wg.Done()
		data.Write(42)
	}()

	// 等待所有操作完成
	wg.Wait()
}

在這個示例中,我們定義了一個Data結(jié)構(gòu)體,它包含一個整數(shù)值和一個sync.RWMutex類型的字段。我們?yōu)?code>Data結(jié)構(gòu)體定義了ReadWrite方法,分別用于讀取和寫入數(shù)據(jù)。在讀取數(shù)據(jù)時,我們使用RLock方法加讀鎖,并在操作完成后使用RUnlock方法讀解鎖。在寫入數(shù)據(jù)時,我們使用Lock方法加寫鎖,并在操作完成后使用Unlock方法寫解鎖。

main函數(shù)中,我們創(chuàng)建了一個Data實例,并啟動了多個讀操作和一個寫操作。通過使用sync.WaitGroup,我們可以確保所有操作都完成后再繼續(xù)執(zhí)行。

0