Golang 使用map需要注意的幾個點

阿新 • • 發佈：2020-09-07

1.簡介

map 是 Golang 中的方便而強大的內建資料結構，是一個同種型別元素的無序組，元素通過另一型別唯一的鍵進行索引。其鍵可以是任何相等性操作符支援的型別，如整數、浮點數、複數、字串、指標、介面（只要其動態型別支援相等性判斷）、結構以及陣列。切片不能用作對映鍵，因為它們的相等性還未定義。與切片一樣，對映也是引用型別。若將對映傳入函式中，並更改了該對映的內容，則此修改對呼叫者同樣可見。未初始化的對映值為 nil。

使用示例如下：

package main

import "fmt"

func main() {
    nameAge := make(map[string]int)
    nameAge["bob"] = 18           //增
    nameAge["tom"] = 16           //增
    delete(nameAge,"bob")         //刪
    nameAge["tom"] = 19           //改
    v := nameAge["tom"]           //查
    fmt.Println("v=",v)
    v,ok := nameAge["tom"]         //查，推薦用法
    if ok { 
      fmt.Println("v=",v,"ok=",ok)
    }  
    for k,v :=range nameAge {   		//遍歷
        fmt.Println(k,v)
    }  
}

輸出結果：

v= 19
v= 19 ok= true
tom 19

2.注意事項

2.1 map的元素不可取址

map中的元素並不是一個變數，而是一個值。因此，我們不能對map的元素進行取址操作。

var m = map[int]int {
	0 : 0,1: 1,}

func main() {
    fmt.Println(&m[0])
}

執行報錯：

cannot take the address of m[0]

因此，當 map 的元素為結構體型別的值，那麼無法直接修改結構體中的欄位值。考察如下示例：

package main

import (
    "fmt"
)

type person struct {
  name  string
  age  byte
  isDead bool
}

func whoIsDead(personMap map[string]person) {
  for name,_ := range personMap {
    if personMap[name].age < 50 {
      personMap[name].isDead = true
    }  
  }  
}

func main() {
  p1 := person{name: "zzy",age: 100}
  p2 := person{name: "dj",age: 99} 
  p3 := person{name: "px",age: 20} 
  personMap := map[string]person{
    p1.name: p1,p2.name: p2,p3.name: p3,}  
  whoIsDead(personMap)
  
  for _,v :=range personMap {
    if v.isDead {
      fmt.Printf("%s is dead\n",v.name)
    }  
  }  
}

編譯報錯：

cannot assign to struct field personMap[name].isDead in map

原因是 map 元素是無法取址的，也就說可以得到 personMap[name]，但是無法對其進行修改。解決辦法有二，一是 map 的 value用 strct 的指標型別，二是使用臨時變數，每次取出來後再設定回去。

（1）將map中的元素改為struct的指標。

package main

import (
    "fmt"
)

type person struct {
  name  string
  age  byte
  isDead bool
}

func whoIsDead(people map[string]*person) {
  for name,_ := range people {
    if people[name].age < 50 {
      people[name].isDead = true
    }  
  }  
}

func main() {
  p1 := &person{name: "zzy",age: 100}
  p2 := &person{name: "dj",age: 99} 
  p3 := &person{name: "px",age: 20} 
  personMap := map[string]*person {
    p1.name: p1,}  
  whoIsDead(personMap)
  
    for _,v :=range personMap {
        if v.isDead {
            fmt.Printf("%s is dead\n",v.name)
        }  
    }  
}

輸出結果：

px is dead

（2）使用臨時變數覆蓋原來的元素。

package main

import (
    "fmt"
)

type person struct {
  name  string
  age  byte
  isDead bool
}

func whoIsDead(people map[string]person) {
  for name,_ := range people {
    if people[name].age < 50 {
      tmp := people[name]
      tmp.isDead = true
      people[name] = tmp 
    }  
  }  
}

func main() {
  p1 := person{name: "zzy",age: 20} 
  personMap := map[string]person {
    p1.name: p1,v.name)
        }  
    }  
}

輸出結果：

px is dead

2.2 map併發讀寫問題

共享 map 在併發讀寫時需要加鎖。先看錯誤示例：

package main

import (
    "fmt"
    "time"
)

var m = make(map[int]int)

func main() {
    //一個go程寫map 
    go func(){
        for i := 0; i < 10000; i++ {
            m[i] = i  
        }  
    }() 

    //一個go程讀map 
    go func(){
        for i := 0; i < 10000; i++ { 
            fmt.Println(m[i])  
        }  
    }() 
    time.Sleep(time.Second*20)
}

執行報錯：

fatal error: concurrent map read and map write

可以使用讀寫鎖（sync.RWMutex）實現互斥訪問。

package main

import (
    "fmt"
    "time"
    "sync"
)

var m = make(map[int]int)
var rwMutex sync.RWMutex

func main() {
    //一個go程寫map 
    go func(){
        rwMutex.Lock()
        for i := 0; i < 10000; i++ {
            m[i] = i  
        }  
        rwMutex.Unlock()
    }() 

    //一個go程讀map
    go func(){
        rwMutex.RLock()
        for i := 0; i < 10000; i++ { 
            fmt.Println(m[i])  
        }  
        rwMutex.RUnlock()
    }() 
    time.Sleep(time.Second*20)
}

正常執行輸出：