读写锁概述

  • 除了上一篇提到的互斥锁以外,Go语言还给我们提供了另一种资源锁——读写锁(sync.RWMutex);
  • 读写锁可以锁定和解锁两种模式:只读模式和只写模式:
  • 只读模式:多路只读不可写;
  • 只写模式:单路只写不可读;

只读模式示例

//定义读写锁
var rwMutex sync.RWMutex

/*只读模式:多路只读不可写*/
func main() {
	for i := 0; i < 3; i++ {
		go read(i)
	}
	time.Sleep(500 * time.Millisecond)
	for i := 0; i < 3; i++ {
		go write(i)
	}
	time.Sleep(10*time.Second)
}

func read(i int) {
	rwMutex.RLock()
	fmt.Println(i, "reading...")
	time.Sleep(time.Second)
	fmt.Println(i, "read over")
	rwMutex.RUnlock()
}

func write(i int) {
	rwMutex.Lock()
	fmt.Println(i, "writing...")
	time.Sleep(time.Second)
	fmt.Println(i, "write over")
	rwMutex.Unlock()
}

其输出效果为:
1 reading…
2 reading…
0 reading…
0 read over
2 read over
1 read over
2 writing…
2 write over
0 writing…
0 write over
1 writing…
1 write over

不难看出,在只读模式下,三条协程同时在进行读取;而只有全部读取协程结束并释放只读锁后,写模式才得以执行——此之谓“多路只读不可写”

只写模式示例

//定义读写锁
var rwMutex sync.RWMutex

/*只写模式:单路只写不可读*/
func main() {
	for i := 0; i < 3; i++ {
		go write(i)
	}
	time.Sleep(500 * time.Millisecond)
	for i := 0; i < 3; i++ {
		go read(i)
	}

	time.Sleep(10*time.Second)
}

func read(i int) {
	rwMutex.RLock()
	fmt.Println(i, "reading...")
	time.Sleep(time.Second)
	fmt.Println(i, "read over")
	rwMutex.RUnlock()
}

func write(i int) {
	rwMutex.Lock()
	fmt.Println(i, "writing...")
	time.Sleep(time.Second)
	fmt.Println(i, "write over")
	rwMutex.Unlock()
}

其输出为:
1 writing…
1 write over
0 reading…
1 reading…
2 reading…
2 read over
0 read over
1 read over
2 writing…
2 write over
0 writing…
0 write over

不难看出,在只写锁释放之前,只有一条协程可以执行写操作,其余协程无论读写都被阻塞——此之谓“单路只写不可读”;
当只写锁释放时,首先被只读锁抢到资源,又是一个三路齐读没有写;
最后再次锁定为只写时,也是逐条写入直到释放只写锁,其间其余的写协程全部被阻塞;