温馨提示×

Go语言读写锁如何管理资源

小樊
83
2024-11-02 16:22:32
栏目: 编程语言

Go语言的sync.RWMutex是一个读写锁,用于在多个goroutine之间同步对共享资源的访问。它允许多个读操作同时进行,但在写操作进行时,只允许一个goroutine进行写操作。这样可以提高并发性能,因为读操作通常比写操作更频繁。

下面是一个简单的示例,展示了如何使用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() // 读锁结束时释放读锁
	return d.value
}

func (d *Data) Write(value int) {
	d.mu.Lock() // 加写锁
	defer d.mu.Unlock() // 写锁结束时释放写锁
	d.value = value
}

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

	// 启动10个读操作
	for i := 0; i < 10; i++ {
		wg.Add(1)
		go func() {
			defer wg.Done()
			fmt.Println("Read:", data.Read())
		}()
	}

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

	// 启动1个写操作
	wg.Add(1)
	go func() {
		defer wg.Done()
		data.Write(42)
	}()

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

	// 再次启动10个读操作
	for i := 0; i < 10; i++ {
		wg.Add(1)
		go func() {
			defer wg.Done()
			fmt.Println("Read:", data.Read())
		}()
	}

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

在这个示例中,我们定义了一个Data结构体,它包含一个整数值和一个sync.RWMutex类型的字段。我们还定义了ReadWrite方法,分别用于读取和写入数据。在读取数据时,我们使用RLock方法加读锁,并在读取完成后使用RUnlock方法释放读锁。在写入数据时,我们使用Lock方法加写锁,并在写入完成后使用Unlock方法释放写锁。

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

0