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

Z时代
2024-01-10
分类：IT

最佳方式：根据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)
}