Golang如何快速删除map所有元素
作者:风神韵
这篇文章主要介绍了Golang如何快速删除map所有元素问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
1. 所有Go版本通用方法
重新申请一个新的map,旧的map交给GC去回收。
a := make(map[string]int) a["a"] = 1 a["b"] = 2 // clear all a = make(map[string]int)
2. Go 1.11版本以上用法
通过Go的内部函数mapclear方法删除。
这个函数并没有显示的调用方法,当你使用for循环遍历删除所有元素时,Go的编译器会优化成Go内部函数mapclear。
package main func main() { m := make(map[byte]int) m[1] = 1 m[2] = 2 for k := range m { delete(m, k) } }
把上述源代码直接编译成汇编(默认编译是会优化的):
go tool compile -S map_clear.go
可以看到编译器把源码9行的for循环直接优化成了mapclear去删除所有元素。
如下:
再来看看关闭优化后的结果:
go tool compile -l -N -S map_clear.go
关闭优化选项后,Go编译器直接通过循环遍历来删除map里面的元素。
由上可知,遍历删除在经过编译器优化后会调用mapclear一次性删除map所有元素,那这个mapclear函数是如何实现的,效率如何?
mapclear源码实现
这部分代码涉及到内存管理和GC,只能看懂个大概,后续再补充。
实现思路
- 清空统计数据,如元素个数、溢出数等。
- 重新申请一个新的extra,原有的extra交给GC。
- 释放桶内存块。
func mapclear(t *maptype, h *hmap) { ... // 把oldbuckets置nil,如果有oldbuckets就让GC处理 h.oldbuckets = nil // 初始化溢出数、元素个数 h.nevacuate = 0 h.noverflow = 0 h.count = 0 // 重新申请一个新的extra,旧的交给GC回收 if h.extra != nil { *h.extra = mapextra{} } // 清空bucket _, nextOverflow := makeBucketArray(t, h.B, h.buckets) ... } func makeBucketArray(t *maptype, b uint8, dirtyalloc unsafe.Pointer) (buckets unsafe.Pointer, nextOverflow *bmap) { base := bucketShift(b) nbuckets := base ... // 没有分配过内存,则申请一个新的 if dirtyalloc == nil { buckets = newarray(t.bucket, int(nbuckets)) } else { // 直接释放整个buckets buckets = dirtyalloc size := t.bucket.size * nbuckets if t.bucket.kind&kindNoPointers == 0 { memclrHasPointers(buckets, size) } else { memclrNoHeapPointers(buckets, size) } } ... }
总结
使用mapclear方法清空map时,做的工作就是初始化和释放申请内存块,效率很高。
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。