片的并发安全映射

Question

我有一个类型，它包含一个sync.Map，其中映射中的键是一个字符串，值是一个切片。我将项目插入到地图中的代码如下：

newList := []*Item{item}
if result, ok := map.LoadOrStore(key, newList); ok {
    resultList := result.([]*Item)
    resultList = append(resultList, item)
    map.Store(key, resultList)
}

这不是并发安全的，因为片可以由多个调用并发加载和修改。此代码非常脆弱，因此我尝试将其修改为：

newList := []*Item{item}
if result, ok := map.LoadOrStore(key, &newList); ok {
    resultList := result.(*[]*Item)
    *resultList = append(*resultList, item)
}

所有这些所做的就是使问题确定性地发生。因此，我正在尝试找到一种方法来创建一个可以同时添加到的切片映射。我的直觉是在添加列表时使用sync.Mutex锁定列表，但为了保持对sync.Map的并发访问，我还需要创建一个sync.Mutex对象的映射，如下所示：

newLock := sync.Mutex{}
raw, _ := lockMap.LoadOrStore(key, &newLock)
lock := raw.(*sync.Mutex)

newList := []*Item{item}
if result, ok := map.LoadOrStore(key, &newList); ok {
    lock.Lock()
    resultList := result.(*[]*Item)
    *resultList = append(*resultList, item)
    lock.Unlock()
}

有没有更简单的方法来解决这个问题？

Answer 1

这与您当前的计划没有太大不同，但是您可以省去处理两个map的麻烦，方法是使用一个结构，其中嵌入了一个用于map的值的互斥锁。

这个结构看起来像这样：

type SafeItems struct {
    sync.Mutex
    Items []*Item
}

它可以像这样使用：

newMapEntry := SafeItems{Items: itemPtrList}
if result, ok := map.LoadOrStore(key, &newMapEntry); ok {
    mapEntry := result.(*SafeItems)
    mapEntry.Lock()
    mapEntry.Items = append(mapEntry.Items, item)
    mapEntry.Unlock()
}

这不是一个巨大的变化，但它确实提供了一些语法糖。