机构,里面有个online字段是一个map,里面保存了在线的用户信息
我们有一个添加在线用户的方法AddOnline,方法里面使用了锁,是因为map是并发不安全的
我们还有一个统计所有在线用户的方法AllOnline
在AllOnline中,我们访问u.online的map,我们加上了读锁。这里的想法是访问当前在线用户的map,并赋值给online,然后释放读锁
遍历赋值的online查出在线用户的数量
可能我们觉得这个是没问题的,但是当我们运行程序的时候会发现这里存在数据竞争:
go run -race 10.go
==================
WARNING: DATA RACE
Write at 0x00c0000a0060 by goroutine 6:
runtime.mapassign_faststr()
...
==================
fatal error: concurrent map iteration and map write
这是因为,在map内部,是hmap结构,主要包含元数据(例如,计数器)和引用数据桶的指针。 因此,online := u.online
不会复制实际数据,而是复制的指针,实际操作的还是同一片内存。
解决这个问题也不难:
-
我们可以把锁的范围扩大,像下面这样:
func (u *User) AllOnline() int {
u.mu.RLock()
defer u.mu.RUnlock()
online := u.online
sum := 0
for _, o := range online {
if o {
sum++
}
}
return sum
}
-
另一种方法就是复制一个副本出来,像上面我们说的切片一样:
func (u *User) AllOnline() int {
u.mu.RLock()
online := make(map[string]bool, len(u.online))
for s, b := range u.online {
online[s] = b
}
u.mu.RUnlock()
sum := 0
for _, o := range online {
if o {
sum++
}
}
return sum
}
上面的例子中我们使用了*User定义了2个方法:
func (u *User) AddOnline(id string) {
u.mu.Lock()
u.online[id] = true
u.mu.Unlock()
}
func (u *User) AllOnline() int {
u.mu.RLock()
online := make(map[string]bool, len(u.online))
for s, b := range u.online {
online[s] = b
}
u.mu.RUnlock()
sum := 0
for _, o := range online {
if o {
sum++
}
}
return sum
}
我现在我们稍微修改下上面的列子:
package main
import (
"strconv"
"sync"
"time"
)
type User struct {
mu sync.RWMutex
online map[string]bool
}
func (u User) AddOnline(id string) {
u.mu.Lock()
u.online[id] = true
u.mu.Unlock()
}
func (u User) AllOnline() int {
u.mu.RLock()
online := make(map[string]bool, len(u.online))
for s, b := range u.online {
online[s] = b
}
u.mu.RUnlock()
sum := 0
for _, o := range online {
if o {
sum++
}
}
return sum
}
func main() {
u := User{}
u.online = make(map[string]bool)
go func() {
for i := 0; i < 10000; i++ {
u.AddOnline("userid" + strconv.Itoa(i))
}
}()
go func() {
for i := 0; i < 10000; i++ {
u.AllOnline()
}
}()
time.Sleep(time.Second)
}
现在我们直接使用User结构体定义这两个方法,但是当我们执行程序的时候,报了数据竞争的错误:
go run -race 10.go
==================
WARNING: DATA RACE
Read at 0x00c00011e060 by goroutine 7:
main.User.AllOnline()
这个又是什么原因造成的呢?这是因为,当我门使用User作为参数时,直接复制了User的副本,因此sync.RWMutex也会被复制。
因为锁被复制了,所以对于同一个临界资源,处于不同锁的读写操作可以同时访问。