Go語言讀寫鎖如何管理資源

小樊
81
2024-11-02 16:22:32
欄目: 編程語言

Go語言的sync.RWMutex是一個(gè)讀寫鎖,用于在多個(gè)goroutine之間同步對(duì)共享資源的訪問。它允許多個(gè)讀操作同時(shí)進(jìn)行,但在寫操作進(jìn)行時(shí),只允許一個(gè)goroutine進(jìn)行寫操作。這樣可以提高并發(fā)性能,因?yàn)樽x操作通常比寫操作更頻繁。

下面是一個(gè)簡單的示例,展示了如何使用sync.RWMutex來管理資源:

package main

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

type Data struct {
	value int
	mu    sync.RWMutex
}

func (d *Data) Read() int {
	d.mu.RLock() // 加讀鎖
	defer d.mu.RUnlock() // 讀鎖結(jié)束時(shí)釋放讀鎖
	return d.value
}

func (d *Data) Write(value int) {
	d.mu.Lock() // 加寫鎖
	defer d.mu.Unlock() // 寫鎖結(jié)束時(shí)釋放寫鎖
	d.value = value
}

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

	// 啟動(dòng)10個(gè)讀操作
	for i := 0; i < 10; i++ {
		wg.Add(1)
		go func() {
			defer wg.Done()
			fmt.Println("Read:", data.Read())
		}()
	}

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

	// 啟動(dòng)1個(gè)寫操作
	wg.Add(1)
	go func() {
		defer wg.Done()
		data.Write(42)
	}()

	// 等待寫操作完成
	wg.Wait()

	// 再次啟動(dòng)10個(gè)讀操作
	for i := 0; i < 10; i++ {
		wg.Add(1)
		go func() {
			defer wg.Done()
			fmt.Println("Read:", data.Read())
		}()
	}

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

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

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

0