sync.Map 似乎不安全的并发读/写

Question

测试 golang 标准包中的 sync.Map。并发读写似乎不安全。 怎么了？

测试代码：

package main

import (
    "log"
    "sync"
)

func main() {
    var m sync.Map
    m.Store("count", 0)

    var wg sync.WaitGroup

    for numOfThread := 0; numOfThread < 10; numOfThread++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            for i := 0; i < 1000; i++ {
                value, ok := m.Load("count")
                if !ok {
                    log.Println("load count error")
                } else {
                    v, _ := value.(int)
                    m.Store("count", v+1)
                }
            }
        }()
    }

    log.Println("threads starts")
    wg.Wait()

    value, ok := m.Load("count")
    if ok {
        v, _ := value.(int)
        log.Printf("final count: %d", v)
    }

    log.Println("all done")
}

https://play.golang.org/p/E-pw4iZUceB

结果应该是10000，但是得到的是随机数而不是10000：

2009/11/10 23:00:00 threads starts
2009/11/10 23:00:00 final count: 6696
2009/11/10 23:00:00 all done

Answer 1

你有一个竞争条件：

 value, ok := m.Load("count")
 ...
 v, _ := value.(int)
 m.Store("count", v+1)

上面的 read-modify-store 不保护其他 goroutine 做同样的事情，因此其他 goroutine 执行的一些增量会被遗漏。

sync.Map 保护对其成员的并发访问。这意味着，对 map 的写入不会导致其他 goroutine 读取不一致的 map。如果你读-修改-写，没有什么可以保护其他 goroutine 同时更新值。当您读取-修改-更新时，您需要一个互斥锁来保护对地图的访问。

Answer 2

package main

import (
    "log"
    "sync"
)

func main() {
    var m sync.Map
    m.Store("count", 0)

    var wg sync.WaitGroup
    var mu *sync.Mutex

    for numOfThread := 0; numOfThread < 10; numOfThread++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            for i := 0; i < 1000; i++ {
                mu.Lock()
                value, ok := m.Load("count")
                if !ok {
                    log.Println("load count error")
                } else {
                    v, _ := value.(int)
                    m.Store("count", v+1)
                }
               mu.Unlock()
            }
        }()
    }

    log.Println("threads starts")
    wg.Wait()

    value, ok := m.Load("count")
    if ok {
        v, _ := value.(int)
        log.Printf("final count: %d", v)
    }

    log.Println("all done")
}

读取和写入操作都是线程安全的，但您正在尝试更新插入或读取+写入操作。它不是线程安全的。修改代码使其线程安全。