设为首页 加入收藏

TOP

golang中的锁竞争问题(二)
2023-07-23 13:30:35 】 浏览:44
Tags:golang
机构,里面有个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也会被复制。

    因为锁被复制了,所以对于同一个临界资源,处于不同锁的读写操作可以同时访问。

    首页 上一页 1 2 下一页 尾页 2/2/2
    】【打印繁体】【投稿】【收藏】 【推荐】【举报】【评论】 【关闭】 【返回顶部
    上一篇Go 语言数组和切片的区别 下一篇go:快速添加接口方法及其实现

    最新文章

    热门文章

    Hot 文章

    Python

    C 语言

    C++基础

    大数据基础

    linux编程基础

    C/C++面试题目