快捷導(dǎo)航

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

更新時(shí)間：2021年04月25日 09:22:33 作者：YZF_Kevin

這篇文章主要介紹了golang 獲取map所有key的方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧

最佳方式：根據(jù)map的長(zhǎng)度，新建一個(gè)數(shù)組，遍歷map逐個(gè)壓入

方法1（效率很高）：

func getKeys1(m map[int]int) []int {
 // 數(shù)組默認(rèn)長(zhǎng)度為map長(zhǎng)度,后面append時(shí),不需要重新申請(qǐng)內(nèi)存和拷貝,效率很高
 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 {
 // 數(shù)組默認(rèn)長(zhǎng)度為map長(zhǎng)度,后面append時(shí),不需要重新申請(qǐng)內(nèi)存和拷貝,效率很高
 keys := make([]int, 0, len(m))
 for k := range m {
  keys = append(keys, k)
 }
 return keys
}

其他方式：

方法3（效率較低）：

func getKeys3(m map[int]int) []int {
 // 注意：由于數(shù)組默認(rèn)長(zhǎng)度為0，后面append時(shí)，需要重新申請(qǐng)內(nèi)存和拷貝，所以效率較低
 keys := []int{}
 for k := range m {
  keys = append(keys, k)
 }
 return keys
}

方法4（效率極低）：

func getKeys4(m map[int]int) int {
 // 注意:雖然此寫法簡(jiǎn)潔,但MapKeys函數(shù)內(nèi)部操作復(fù)雜,效率極低
 keys := reflect.ValueOf(m).MapKeys()
 return len(keys)
}

實(shí)驗(yàn)結(jié)果如圖（可以看到方法1和方法2效率最高，內(nèi)存操作也最少）：

完整代碼如下：

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 {
 // 數(shù)組默認(rèn)長(zhǎng)度為map長(zhǎng)度,后面append時(shí),不需要重新申請(qǐng)內(nèi)存和拷貝,效率較高
 j := 0
 keys := make([]int, len(m))
 for k := range m {
  keys[j] = k
  j++
 }
 return keys
}
 
func getKeys2(m map[int]int) []int {
 // 數(shù)組默認(rèn)長(zhǎng)度為map長(zhǎng)度,后面append時(shí),不需要重新申請(qǐng)內(nèi)存和拷貝,效率較高
 keys := make([]int, 0, len(m))
 for k := range m {
  keys = append(keys, k)
 }
 return keys
}
 
// 初始化默認(rèn)
func getKeys3(m map[int]int) []int {
 // 注意：由于數(shù)組默認(rèn)長(zhǎng)度為0，后面append時(shí)，需要重新申請(qǐng)內(nèi)存和拷貝，所以效率較低
 keys := []int{}
 for k := range m {
  keys = append(keys, k)
 }
 return keys
}
 
// 使用反射
func getKeys4(m map[int]int) int {
 // 注意:雖然此寫法簡(jiǎn)潔,但MapKeys函數(shù)內(nèi)部操作復(fù)雜,效率極低
 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)
 }
}

補(bǔ)充：Golang踩坑——判斷map中是否有key

最近在實(shí)習(xí)，下班回去十點(diǎn)多了，再加上比較懶，快有兩個(gè)月沒寫東西了。

今天在開發(fā)一個(gè)模塊的時(shí)候是接著上一個(gè)人的寫的，好不容易各種配置寫好了開始跑，發(fā)現(xiàn)他踩了一個(gè)很容易踩的坑。

把bug抽出來單獨(dú)寫了個(gè)文件

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)
}

運(yùn)行一下

可以看到報(bào)了無效指針的問題（invalid memory address or nil pointer dereference）。

從上面的兩個(gè)log可以看到，最開始指針是有值的，第二個(gè)卻沒了，原因就出現(xiàn)在那個(gè)if判斷那里。

if判斷是想通過查看map知道agent是否已經(jīng)存在，這個(gè)地方的錯(cuò)誤是a,ok = map; 我們想通過ok判斷是否存在，但是在這里還有個(gè)a，這會(huì)導(dǎo)致有一個(gè)對(duì)a賦值的操作，也就是說如果map里有這個(gè)值，那沒事，如果沒有的話a就被賦值成了nil了。

在下面打印的時(shí)候也就成了nil，取它的屬性的時(shí)候會(huì)告訴我們找了個(gè)空指針的屬性，這肯定找不到。

我們把a(bǔ)換成_，再來看結(jié)果：