快捷導(dǎo)航

如何避免go的map競態(tài)問題的方法

更新時間：2023年02月09日 09:21:03 作者：有想法的工程師

本文主要介紹了如何避免go的map競態(tài)問題的方法，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

背景

在使用go語言開發(fā)的過程中，我碰到過這樣一種情況，就是代碼自測沒問題，代碼檢查沒問題，上線跑了一段時間時間了也沒問題，就是突然偶爾會抽風(fēng)panic，導(dǎo)致程序所在的pod(k8s的運(yùn)行docker鏡像的最小單位)重啟了，而程序里拋出來的異常如下

，意思是多個協(xié)程正在同時對同一個map變量進(jìn)行讀寫，這個就涉及到go程序的競態(tài)問題，而競態(tài)問題也是我們?nèi)粘ｉ_發(fā)中遇到比較多的情況

為什么會出現(xiàn)競態(tài)問題

出現(xiàn)這個問題的主要原因是有多個協(xié)程在對同一個map變量進(jìn)行修改，這樣就可能會出現(xiàn)map被一個協(xié)程修改到一半的時候，然后另外一個協(xié)程就來讀取了，導(dǎo)致讀到一個“半成品”的map變量。而這個就說明一個問題，就是map類型并不是并發(fā)安全的

而并發(fā)安全的定義就是：在高并發(fā)下，進(jìn)程、線程(協(xié)程)出現(xiàn)資源競爭，導(dǎo)致出現(xiàn)臟讀，臟寫，死鎖等情況。

那么go語言有如下幾種類型不具備并發(fā)安全：map，slice，struct，channel，string

不過奇怪的是，只有map類型發(fā)生并發(fā)競爭的時候，才會拋出fatal error，這個是無法被recover的，一定會中斷程序，而這也導(dǎo)致程序運(yùn)行的pod會被檢測出異常從而重啟

查了資料，有一種說法是，map大部分會被用來存配置文件，而配置文件出錯可能會導(dǎo)致一些嚴(yán)重的業(yè)務(wù)問題，所以寧愿程序崩潰也要保全業(yè)務(wù)數(shù)據(jù)不會出現(xiàn)臟數(shù)據(jù)（只是一種說法，不用太過在意）

如何解決競態(tài)問題

1、使用go的一些并發(fā)原語

如果需要修改的變量是程序啟動之后就不需要修改的配置，那么可以使用sync.Once包來處理，這個包的作用就是限制一件事情只做一次，示例代碼如下

type User struct {
   Name string
   Other map[string]interface{}
   ConfigOnce sync.Once
}
 
// InitConfigOnce
// @description "初始化配置信息，只執(zhí)行一次"
// @auth yezibin 2023-01-21 15:38:09
// @param name string "description"
// @param other map[string]interface{} "description"
// @return *User "description"
func (u *User)InitConfigOnce(name string, other map[string]interface{}) *User {
   //Do包起來的方法，只會執(zhí)行一次，但是必須是同一個sync.Once變量
   u.ConfigOnce.Do(func() {
      fmt.Println("ok")
      u.Name = name
      u.Other = other
   })
   return u
}
 
// GetUserConfig
// @description "打印配置文件"
// @auth yezibin 2023-01-21 15:38:36
func (u *User) GetUserConfig()  {
   fmt.Println(u)
}

2、加讀寫鎖（RWMutex map）

出現(xiàn)競態(tài)的本質(zhì)是因?yàn)槎鄠€協(xié)程對同一個變量同時進(jìn)行讀與寫，通過用鎖來防止這個情況，因?yàn)槲遗e得案例是讀多寫少的情況，用上讀寫鎖性能會更好，示例代碼如下

type Mmap struct {
   Data map[string]interface{}
   Mu sync.RWMutex //因?yàn)橹饕桥渲?，屬于讀多寫少情況，所以使用讀寫鎖提高鎖的性能
}
 
 
// InitMmap
// @description "初始化讀寫鎖的map結(jié)構(gòu)體"
// @auth yezibin 2023-01-21 00:09:30
// @return *Config "description"
func InitMmap() *Mmap {
   return &Mmap{
      Data: make(map[string]interface{}),
   }
}
// Get
// @description "獲取配置map數(shù)據(jù)"
// @auth yezibin 2023-01-21 00:10:09
// @param name string "description"
// @return interface{} "description"
func (m *Mmap) Get(name string) interface{} {
   m.Mu.RLock()
   defer m.Mu.RUnlock()
   return m.Data[name]
}
 
// Set
// @description "批量設(shè)置map的值"
// @auth yezibin 2023-02-05 13:08:17
// @param data map[string]interface{} "description"
func (m *Mmap) Set(data map[string]interface{})  {
   m.Mu.Lock()
   defer m.Mu.Unlock()
   for k, v := range data {
      m.Data[k] = v
   }
 
}
// SetOne
// @description "設(shè)置配置map數(shù)據(jù)"
// @auth yezibin 2023-01-21 00:10:23
// @param key string "description"
// @param val string "description"
func (m *Mmap) SetOne(key, val string)  {
   m.Mu.Lock()
   defer m.Mu.Unlock()
   m.Data[key] = val
}

建議

1、如果屬于讀多寫少的情況，盡量選擇讀寫鎖來減少鎖住的范圍，從而提高讀寫性能

2、這里推薦將需要用來讀寫的map變量和鎖共同組建一個struct，這樣能保證讀和寫上的是同一把讀寫鎖，同時也方便整合對map變量的操作

3、分片加鎖

方案2中雖然加了讀寫鎖，比加一把普通的鎖要性能高些，不過鎖的粒度還是大了些，當(dāng)高并發(fā)來襲時，寫的操作必然會阻塞讀的動作，那么有沒有辦法將鎖住的范圍縮小一些呢

思路：如果給map里的每個元素加鎖，每次修改只是單個元素的鎖生效，其他沒改到的元素就正常讀，這樣鎖的粒度會更細(xì)，這就是分片加鎖的原理

這種就是將一把“大”鎖拆成一把把小鎖，是一種空間換時間的方法

實(shí)現(xiàn)上，已經(jīng)有人實(shí)現(xiàn)了好用的具有分片鎖的map，庫地址：https://github.com/orcaman/concurrent-map

import (
   cmap "github.com/orcaman/concurrent-map"
   "sync"
)
// InitCmap
// @description "初始化分片鎖的map"
// @auth yezibin 2023-02-05 14:08:17
// @return *cmapConfig "description"
func InitCmap() *cmapConfig {
   return &cmapConfig{
      cmap.New(),
   }
}
 
// Set
// @description "批量往map寫入元素"
// @auth yezibin 2023-02-05 14:10:02
// @param config map[string]interface{} "description"
func (c *cmapConfig) Set(config map[string]interface{})  {
   for k, v := range config{
      c.Cmap.Set(k, v)
   }
}
 
// Get
// @description "從map獲取元素"
// @auth yezibin 2023-02-05 14:10:22
// @param k string "description"
// @return interface{} "description"
func (c *cmapConfig) Get(k string) interface{} {
   v, ok := c.Cmap.Get(k)
   if ok {
      return v
   } else {
      return nil
   }
}

4、go的原生可并發(fā)map

最后還會跟大家介紹一個go原生庫里就有一個可并發(fā)讀寫的map，這個放在sync庫

官方的文檔中指出，在以下兩個場景中使用 sync.Map，會比使用 map+RWMutex 的方式，性能要好得多：

1、只會增長的緩存系統(tǒng)中，一個 key 只寫入一次而被讀很多次；

2、多個 goroutine 為不相交的鍵集讀、寫和重寫鍵值對。

原理：sync.Map結(jié)構(gòu)里有兩個字段，一個read，一個dirty。dirty包含read的所有字段，新增字段是寫在dirty上，有個miss變量用戶訪問到read沒有，但是dirty有的數(shù)據(jù)次數(shù)

空間換時間。通過冗余的兩個數(shù)據(jù)結(jié)構(gòu)（只讀的 read 字段、可寫的 dirty），來減少加鎖對性能的影響。對只讀字段（read）的操作不需要加鎖。優(yōu)先從 read 字段讀取、更新、刪除，因?yàn)閷?read 字段的讀取不需要鎖。
動態(tài)調(diào)整。miss 次數(shù)多了之后，將 dirty 數(shù)據(jù)提升為 read，避免總是從 dirty 中加鎖讀取。double-checking。加鎖之后先還要再檢查 read 字段，確定真的不存在才操作 dirty 字段。
延遲刪除。刪除一個鍵值只是打標(biāo)記，只有在提升 dirty 字段為 read 字段的時候才清理刪除的數(shù)據(jù)。

示例代碼

type syncMapConfig struct {
   Smap sync.Map
}
// InitSmap
// @description "初始化sync.map"
// @auth yezibin 2023-02-05 15:43:08
// @return *syncMapConfig "description"
func InitSmap() *syncMapConfig {
   return &syncMapConfig{
      sync.Map{},
   }
}
// Set
// @description "批量寫入map"
// @auth yezibin 2023-02-05 15:43:57
// @param config map[string]interface{} "description"
func (s *syncMapConfig) Set(config map[string]interface{})  {
   for k, v := range config {
      s.Smap.Store(k, v)
   }
}
// Get
// @description "從map里獲取數(shù)據(jù)"
// @auth yezibin 2023-02-05 15:44:09
// @param k string "description"
// @return interface{} "description"
func (s *syncMapConfig) Get(k string) interface{} {
   c, ok := s.Smap.Load(k)
   if ok {
      return c
   } else {
      return nil
   }
}