golang 如何获取map所有key的方式

最佳方式：根据map的长度，新建一个数组，遍历map逐个压入

方法1（效率很高）：

func getKeys1(m map[int]int) []int {
// 数组默认长度为map长度,后面append时,不需要重新申请内存和拷贝,效率很高
j := 0
keys := make([]int, len(m))
for k := range m {
 keys[j] = k
 j++
}
return keys
}

方法2（效率很高）：

func getKeys2(m map[int]int) []int {
// 数组默认长度为map长度,后面append时,不需要重新申请内存和拷贝,效率很高
keys := make([]int, 0, len(m))
for k := range m {
 keys = append(keys, k)
}
return keys
}

其他方式：

方法3（效率较低）：

func getKeys3(m map[int]int) []int {
// 注意：由于数组默认长度为0，后面append时，需要重新申请内存和拷贝，所以效率较低
keys := []int{}
for k := range m {
 keys = append(keys, k)
}
return keys
}

方法4（效率极低）：

func getKeys4(m map[int]int) int {
// 注意:虽然此写法简洁,但MapKeys函数内部操作复杂,效率极低
keys := reflect.ValueOf(m).MapKeys()
return len(keys)
}

实验结果如图（可以看到方法1和方法2效率最高，内存操作也最少）：

完整代码如下：

package test

import (
"reflect"
"testing"
)

// 初始化map
func initMap() map[int]int {
m := map[int]int{}
for i := 0; i < 10000; i++ {
 m[i] = i
}
return m
}

func getKeys1(m map[int]int) []int {
// 数组默认长度为map长度,后面append时,不需要重新申请内存和拷贝,效率较高
j := 0
keys := make([]int, len(m))
for k := range m {
 keys[j] = k
 j++
}
return keys
}

func getKeys2(m map[int]int) []int {
// 数组默认长度为map长度,后面append时,不需要重新申请内存和拷贝,效率较高
keys := make([]int, 0, len(m))
for k := range m {
 keys = append(keys, k)
}
return keys
}

// 初始化默认
func getKeys3(m map[int]int) []int {
// 注意：由于数组默认长度为0，后面append时，需要重新申请内存和拷贝，所以效率较低
keys := []int{}
for k := range m {
 keys = append(keys, k)
}
return keys
}

// 使用反射
func getKeys4(m map[int]int) int {
// 注意:虽然此写法简洁,但MapKeys函数内部操作复杂,效率极低
keys := reflect.ValueOf(m).MapKeys()
return len(keys)
}

func BenchmarkMapkeys1(b *testing.B) {
// 初始化map
m := initMap()

b.ResetTimer()
for i := 0; i < b.N; i++ {
 getKeys1(m)
}
}
func BenchmarkMapkeys2(b *testing.B) {
// 初始化map
m := initMap()

b.ResetTimer()
for i := 0; i < b.N; i++ {
 getKeys2(m)
}
}

func BenchmarkMapkeys3(b *testing.B) {
// 初始化map
m := initMap()

b.ResetTimer()
for i := 0; i < b.N; i++ {
 getKeys3(m)
}
}

func BenchmarkMapkeys4(b *testing.B) {
// 初始化map
m := initMap()

b.ResetTimer()
for i := 0; i < b.N; i++ {
 getKeys4(m)
}
}

补充：Golang踩坑——判断map中是否有key

最近在实习，下班回去十点多了，再加上比较懒，快有两个月没写东西了。

今天在开发一个模块的时候是接着上一个人的写的，好不容易各种配置写好了开始跑，发现他踩了一个很容易踩的坑。

把bug抽出来单独写了个文件

package main

import (
"log"
)

type agent struct {
id  int
str string
}

var m map[int]*agent
func main() {
m = make(map[int]*agent)
a := &agent{
id:  1,
str: "hello",
}
log.Println(a)
var ok bool
if a, ok = m[1]; ok {
log.Println("ok")
}
m[1] = a
log.Println(a)
log.Println(a.str)
}

运行一下

可以看到报了无效指针的问题（invalid memory address or nil pointer dereference）。

从上面的两个log可以看到，最开始指针是有值的，第二个却没了，原因就出现在那个if判断那里。

if判断是想通过查看map知道agent是否已经存在，这个地方的错误是a,ok = map; 我们想通过ok判断是否存在，但是在这里还有个a，这会导致有一个对a赋值的操作，也就是说如果map里有这个值，那没事，如果没有的话a就被赋值成了nil了。

在下面打印的时候也就成了nil，取它的属性的时候会告诉我们找了个空指针的属性，这肯定找不到。

我们把a换成_，再来看结果：

ok了。

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。如有错误或未考虑完全的地方，望不吝赐教。

来源：https://blog.csdn.net/yzf279533105/article/details/94010954