温馨提示×

Go语言读写锁怎样实现同步

小樊
81
2024-11-02 16:15:31
栏目: 编程语言

在Go语言中,可以使用sync.RWMutex来实现读写锁。sync.RWMutex是一个读写互斥锁,它允许多个读操作同时进行,但在写操作进行时,会阻止其他读写操作。下面是一个简单的示例,展示了如何使用sync.RWMutex实现同步:

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结构体,它包含一个整数值和一个sync.RWMutex类型的字段。我们为Data结构体定义了ReadWrite方法,分别用于读取和写入数据。在读取数据时,我们使用RLock方法加读锁,并在操作完成后使用RUnlock方法读解锁。在写入数据时,我们使用Lock方法加写锁,并在操作完成后使用Unlock方法写解锁。

main函数中,我们创建了一个Data实例,并启动了多个读操作和一个写操作。通过使用sync.WaitGroup,我们可以确保所有操作都完成后再继续执行。

0