Golang 使用map需要注意的几个点
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, 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)使用临时变量覆盖原来的元素。
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 在并发读写时需要加锁。先看错误示例:
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) }
正常运行输出:
0
1
...
9999
以上就是Golang 使用map需要注意的几个点的详细内容,更多关于golang map的资料请关注我们其它相关文章!