在同一个 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 中有两次写入,而不是一次写入和一次读取。

由此,我在这里有一些问题:

  1. 有没有一种方法可以保证只有一个 goroutine 在映射上写入(输入上面的 if 代码块),并且所有其他例程都使用新值读取该映射(避免输入 if 代码块)?
  2. 它是 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上面的用法,否则它将不得不再次访问地图。


以上是在同一个 goroutine 中使用 Lock 和 Rlock的全部内容。
THE END
分享
二维码
< <上一篇
下一篇>>