当前位置 博文首页 > 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)
}
正常运行输出：
js0
1
...
9999
