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

map
这篇文章主要介绍了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了。

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