限制地图的并发访问

知识点掌握了，还需要不断练习才能熟练运用。下面golang学习网给大家带来一个Golang开发实战，手把手教大家学习《限制地图的并发访问》，在实现功能的过程中也带大家重新温习相关知识点，温故而知新，回头看看说不定又有不一样的感悟！

我有一个地图：map[string]map[string]*Struct，我需要在多个 Go 例程中读取/写入它。

实现这一目标的最佳方法是什么？互斥体或 RWMutex ？以及放置在哪里？

如果我使用 RWMutex，那么在执行涉及读和写的操作之前我应该​​锁定还是 RLock？

我尝试在根映射中使用 rwmutex，但我不确定这是否是解决此问题的最佳方法。

我还尝试在读写之前“锁定”，但有时会收到“并发写入”恐慌。

正确答案

您可以使用RWLock。如果操作涉及到写（不管是读还是只写）就需要使用Lock，如果只涉及读则RLock/RUnlock。

Lock也可以被认为是独占锁。另一方面，RLock 是非排他性的。 即使 RWMutex 已被锁定以进行读取，也可以获取 RLock，但如果资源被 Lock 方法独占锁定，则会阻止 goroutine 执行：

a blocked Lock call excludes new readers from acquiring the lock

另一方面，Lock 方法会阻止 goroutine 执行，直到所有读取器和写入器解锁资源（使用 RUnlock/Unlock 方法）。 Lock 是独占的，因为在调用 Unlock 方法之前，只有一个 goroutine 可以访问资源（无论是读还是写）。

典型方法：

package main

import (
    "fmt"
    "sync"
)

type SomeStruct struct {
    someInfo string
}

type ConcurrentStruct struct {
    mu   sync.RWMutex
    data map[string]map[string]*SomeStruct
}

func New() *ConcurrentStruct {
    return &ConcurrentStruct{
        data: make(map[string]map[string]*SomeStruct),
    }
}

func (cs *ConcurrentStruct) Set(key1, key2 string, val SomeStruct) {
    cs.mu.Lock()
    defer cs.mu.Unlock()

    if _, ok := cs.data[key1]; !ok {
        cs.data[key1] = make(map[string]*SomeStruct)
    }
    cs.data[key1][key2] = &val
}

func (cs *ConcurrentStruct) Get(key1, key2 string) (val *SomeStruct, ok bool) {
    cs.mu.RLock()
    defer cs.mu.RUnlock()

    if _, ok := cs.data[key1]; ok {
        val, ok := cs.data[key1][key2]
        return val, ok
    }

    return nil, false
}

func main() {
    cs := New()
    cs.Set("a", "b", SomeStruct{"Hello, World!"})

    if _, ok := cs.Get("a", "c"); !ok {
        fmt.Printf("key1=a, key2=c, not found\n")
    }

    if s, ok := cs.Get("a", "b"); ok {
        fmt.Printf("key1=a, key2=b, found: %v\n", s)
    }
}

好了，本文到此结束，带大家了解了《限制地图的并发访问》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！