Golang 使用map需要注意的几个点

时间:2022-11-23 08:36:01

1.简介

mapGolang 中的方便而强大的内建数据结构,是一个同种类型元素的无序组,元素通过另一类型唯一的键进行索引。其键可以是任何相等性操作符支持的类型, 如整数、浮点数、复数、字符串、指针、接口(只要其动态类型支持相等性判断)、结构以及数组。 切片不能用作映射键,因为它们的相等性还未定义。与切片一样,映射也是引用类型。 若将映射传入函数中,并更改了该映射的内容,则此修改对调用者同样可见。未初始化的映射值为 nil。

使用示例如下:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
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的元素进行取址操作。

?
1
2
3
4
5
6
7
8
var m = map[int]int {
    0 : 0,
    1: 1,
}
 
func main() {
    fmt.Println(&m[0])
}

运行报错:

cannot take the address of m[0]

因此,当 map 的元素为结构体类型的值,那么无法直接修改结构体中的字段值。考察如下示例:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
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的指针。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
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,
    p2.name: p2,
    p3.name: p3,
  
  whoIsDead(personMap)
  
    for _, v :=range personMap {
        if v.isDead {
            fmt.Printf("%s is dead\n", v.name)
        
    
}

输出结果:

px is dead

(2)使用临时变量覆盖原来的元素。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
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: 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)
        
    
}

输出结果:

px is dead

2.2 map并发读写问题

共享 map 在并发读写时需要加锁。先看错误示例:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
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)实现互斥访问。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
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)
}

正常运行输出:

0
1
...
9999

以上就是Golang 使用map需要注意的几个点的详细内容,更多关于golang map的资料请关注服务器之家其它相关文章!

原文链接:https://cloud.tencent.com/developer/article/1446306