在同一个 goroutine 中使用 Lock 和 Rlock
我正在使用 Go 中的 RWMutex 进行实验,我意识到可以使用以下代码实现这种行为:
- goroutine 1 - RLock
- goroutine 1 - 运行解锁
- goroutine 2 - RLock
- goroutine 2 - 运行解锁
- goroutine 2 - 锁
- goroutine 2 - 解锁
- goroutine 1 - 锁
- goroutine 1 - 解锁
package main
import (
"fmt"
"sync"
"time"
)
type RLockAndLockStruct struct {
mu sync.RWMutex
mapEx map[string]string
}
func main() {
r := &RLockAndLockStruct{}
r.mapEx = make(map[string]string)
go r.RLockAndLockTest("test", "goroutine 1 - ")
go r.RLockAndLockTest("test", "goroutine 2 - ")
time.Sleep(4000 * time.Millisecond)
}
func (r *RLockAndLockStruct) RLockAndLockTest(value string, goroutine string) string {
r.mu.RLock()
fmt.Printf("%sRLock\n", goroutine)
t := r.mapEx[value]
r.mu.RUnlock()
fmt.Printf("%sRUnlock\n", goroutine)
if len(t) <= 0 {
time.Sleep(500 * time.Millisecond)
r.mu.Lock()
fmt.Printf("%sLock\n", goroutine)
r.mapEx[value] = value
r.mu.Unlock()
fmt.Printf("%sUnlock\n", goroutine)
return r.mapEx[value]
}
return t
}
我在一些文章中读到过,在 Goroutines 中使用 map 的正确方法是使用 RWMutex 和 RLock 来读取和使用 Lock 来写入。但是,正如你在上面的代码中看到的,如果两个 Goroutines 几乎同时启动,则有可能在同一个 map 中有两次写入,而不是一次写入和一次读取。
由此,我在这里有一些问题:
- 有没有一种方法可以保证只有一个 goroutine 在映射上写入(输入上面的 if 代码块),并且所有其他例程都使用新值读取该映射(避免输入 if 代码块)?
- 它是 goroutine 和 map 的正确实现吗?
回答
这是因为您的代码中存在竞争条件。您对地图进行读锁定以读取它,做出决定,然后对其进行写锁定。无法保证当您获得写锁时,您做出决定的条件仍然成立。
正确的方法是在锁定后重新测试条件:
if len(t) <= 0 {
time.Sleep(500 * time.Millisecond)
r.mu.Lock()
if len(r.mapEx[value])<=0 {
fmt.Printf("%sLock\n", goroutine)
r.mapEx[value] = value
}
r.mu.Unlock()
return value
请注意return value上面的用法,否则它将不得不再次访问地图。