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

 更新时间:2021年04月25日 09:22:33   作者:YZF_Kevin  
这篇文章主要介绍了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了。

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

相关文章

  • Golang Gorm 更新字段save、update、updates

    Golang Gorm 更新字段save、update、updates

    在gorm中,批量更新操作可以通过使用Update方法来实现,本文主要介绍了Golang Gorm 更新字段save、update、updates,具有一定的参考价值,感兴趣的可以了解一下
    2023-12-12
  • 使用docker构建golang线上部署环境的步骤详解

    使用docker构建golang线上部署环境的步骤详解

    这篇文章主要介绍了使用docker构建golang线上部署环境的步骤,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧。
    2017-11-11
  • Go 1.18新特性之泛型的全面讲解

    Go 1.18新特性之泛型的全面讲解

    本文力求能让未接触过泛型编程的人也能较好理解Go的泛型,所以行文可能略显啰嗦。但是请相信我,看完这篇文章你能获得对Go泛型非常全面的了解
    2023-03-03
  • Go语言sync.Cond使用方法详解

    Go语言sync.Cond使用方法详解

    Go语言标准库中还包含条件变量 sync.Cond,它可以让一组 Goroutine 都在满足特定条件时被唤醒,每一个sync.Cond结构体在初始化时都需要传入一个互斥锁,接下来我们将通过文中例子了解它的使用方法,感兴趣的同学跟着小编一起来看看吧
    2023-07-07
  • Go习惯用法(多值赋值短变量声明赋值简写模式)基础实例

    Go习惯用法(多值赋值短变量声明赋值简写模式)基础实例

    本文为大家介绍了Go习惯用法(多值赋值,短变量声明和赋值,简写模式、多值返回函数、comma,ok 表达式、传值规则)的基础实例,帮大家巩固扎实Go语言基础
    2024-01-01
  • Golang设计模式之外观模式讲解和代码示例

    Golang设计模式之外观模式讲解和代码示例

    外观是一种结构型设计模式, 能为复杂系统、 程序库或框架提供一个简单 (但有限) 的接口,这篇文章就给大家详细介绍一下Golang的外观模式,文中有详细的代码示例,具有一定的参考价值,需要的朋友可以参考下
    2023-06-06
  • vim配置go语言语法高亮问题的解决方法

    vim配置go语言语法高亮问题的解决方法

    vim配置go语言语法高亮的问题已经遇到过好几次了,每次都是找不到答案,今天小编给大家带来了vim配置go语言语法高亮问题的解决方法,感兴趣的朋友一起看看吧
    2018-01-01
  • 详解golang中接口使用的最佳时机

    详解golang中接口使用的最佳时机

    接口在系统设计中,以及代码重构优化中,是一个不可或缺的工具,能够帮助我们写出可扩展,可维护性更强的程序,本文主要为大家介绍一下golang中接口使用的最佳时机,有兴趣的可以了解下
    2023-09-09
  • golang中json和struct的使用说明

    golang中json和struct的使用说明

    这篇文章主要介绍了golang中json和struct的使用说明,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2021-05-05
  • Go中的fuzz模糊测试使用实战详解

    Go中的fuzz模糊测试使用实战详解

    这篇文章主要为大家介绍了Go中的fuzz模糊测试使用实战详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-12-12

最新评论