Как защититься от ошибок при конкурентной записи в map?

Для защиты от ошибок при конкурентной записи в map в Golang, которые не являются потокобезопасными по умолчанию, используются следующие механизмы:

• sync.Mutex: Блокировка map перед записью и разблокировка после.

  import "sync"
  
  type SafeMap struct {
      mu sync.Mutex
      data map[string]int
  }
  
  func NewSafeMap() *SafeMap {
      return &SafeMap{
          data: make(map[string]int),
      }
  }
  
  func (sm *SafeMap) Set(key string, value int) {
      sm.mu.Lock()
      defer sm.mu.Unlock()
      sm.data[key] = value
  }
  
  func (sm *SafeMap) Get(key string) (int, bool) {
      sm.mu.Lock()
      defer sm.mu.Unlock()
      val, ok := sm.data[key]
      return val, ok
  }
  

• sync.RWMutex: Позволяет одновременно читать из map нескольким горутинам, но блокирует запись при concurrent read or write.

  import "sync"
  
  type SafeMapRWMutex struct {
      mu sync.RWMutex
      data map[string]int
  }
  
  func NewSafeMapRWMutex() *SafeMapRWMutex {
      return &SafeMapRWMutex{
          data: make(map[string]int),
      }
  }
  
  func (sm *SafeMapRWMutex) Set(key string, value int) {
      sm.mu.Lock() // Write lock
      defer sm.mu.Unlock()
      sm.data[key] = value
  }
  
  func (sm *SafeMapRWMutex) Get(key string) (int, bool) {
      sm.mu.RLock() // Read lock
      defer sm.mu.RUnlock()
      val, ok := sm.data[key]
      return val, ok
  }
  

• sync.Map: Специализированный потокобезопасный тип, оптимизированный для сценариев, где набор ключей относительно стабилен, а записи редки, либо когда несколько горутин читают и пишут для дизъюнктивных наборов ключей.

  import "sync"
  
  var safeMap sync.Map // Declaration
  
  func UseSyncMap() {
      safeMap.Store("key1", 10) // Write
      if val, ok := safeMap.Load("key1"); ok { // Read
          // Use val
      }
      safeMap.Delete("key1") // Delete
  }
  

Сравнение подходов:

Выбор конкретного подхода зависит от характера конкурентного доступа к map. Для общих случаев часто достаточно sync.Mutex или sync.RWMutex. Для специфических сценариев sync.Map может обеспечить лучшую производительность.