Golang如何快速删除map所有元素

目录

• 1. 所有Go版本通用方法
• 2. Go 1.11版本以上用法
• mapclear源码实现
• 实现思路
• 总结

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

把上述源php代码直接编译成汇编（默认编译是会优化的）：

go tool compile -S map_clear.go

可以看到编译器把源码9行的for循环直接优化成了mapclear去删除所有元素。

如下：

再来看看关闭优化后的结果：

go tool compile http://www.devze.com-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.extrajs = 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时，做的工作就是初始化和释放申请内存块，效率很高。

以上为个人经验，希望能给大家一个参考，也希php望大家多多编程客栈支持编程客栈(www.devze.com)。