听说过对 Go map 做 GC 吗？

最新推荐文章于 2023-03-25 18:19:26 发布

kevwan

最新推荐文章于 2023-03-25 18:19:26 发布

阅读量175

点赞数 1

分类专栏： go-zero 文章标签： go

本文链接：https://blog.csdn.net/jfwan/article/details/118725840

版权

go-zero 专栏收录该内容

27 篇文章 9 订阅

订阅专栏

在 Golang 中的 map 结构，在删除键值对的时候，并不会真正的删除，而是标记。那么随着键值对越来越多，会不会造成大量内存浪费？

首先答案是会的，很有可能导致 OOM，而且针对这个还有一个讨论：[https://github.com/golang/go/issues/20135](https://github.com/golang/go/issues/20135)。大致的意思就是在很大的 `map` 中，`delete` 操作没有真正释放内存而可能导致内存 OOM。

所以一般的做法：就是 **重建map**。而 `go-zero` 中内置了 `safemap` 的容器组件。`safemap` 在一定程度上可以避免这种情况发生。

那首先我们看看 `go` 原生提供的 `map` 是怎么删除的？

## 原生map删除

```go
1 package main
2
3 func main() {
4 m := make(map[int]string, 9)
5 m[1] = "hello"
6 m[2] = "world"
7 m[3] = "go"
8
9 v, ok := m[1]
10 _, _ = fn(v, ok)
11
12 delete(m, 1)
13 }
14
15 func fn(v string, ok bool) (string, bool) {
16 return v, ok
17 }
```

测试代码如上，我们可以通过 ` go tool compile -S -N -l testmap.go | grep "CALL"` ：

```go
0x0071 00113 (test/testmap.go:4) CALL runtime.makemap(SB)
0x0099 00153 (test/testmap.go:5) CALL runtime.mapassign_fast64(SB)
0x00ea 00234 (test/testmap.go:6) CALL runtime.mapassign_fast64(SB)
0x013b 00315 (test/testmap.go:7) CALL runtime.mapassign_fast64(SB)
0x0194 00404 (test/testmap.go:9) CALL runtime.mapaccess2_fast64(SB)
0x01f1 00497 (test/testmap.go:10) CALL "".fn(SB)
0x0214 00532 (test/testmap.go:12) CALL runtime.mapdelete_fast64(SB)
0x0230 00560 (test/testmap.go:7) CALL runtime.gcWriteBarrier(SB)
0x0241 00577 (test/testmap.go:6) CALL runtime.gcWriteBarrier(SB)
0x0252 00594 (test/testmap.go:5) CALL runtime.gcWriteBarrier(SB)
0x025c 00604 (test/testmap.go:3) CALL runtime.morestack_noctxt(SB)
```

执行第12行的 `delete`，实际执行的是 `runtime.mapdelete_fast64`。

这些函数的参数类型是具体的 `int64`，`mapdelete_fast64` 跟原始的 `delete` 操作一样的，所以我们来看看 `mapdelete`。

### mapdelete

*长图预警！！！*

![](https://oscimg.oschina.net/oscnet/up-8c0678b8fb76fd4fd92c8d78bba0264aa53.png)

大致代码分析如上，具体代码就留给大家去阅读了。其实大致过程：

1. 写保护，防止并发写
2. 查询要删除的 `key` 是否存在
3. 存在则对其标志做删除标记
4. `count--`

所以你在大面积删除 `key` ，实际 `map` 存储的 `key` 是不会删除的，只是标记当前的key状态为 `empty`。

其实出发点，和 `mysql` 的标记删除类似，防止后续会有相同的 `key` 插入，省去了扩缩容的操作。

但是这个对有些场景是不妥的，如果开发者在未来时间内都不会再插入相同的 `key` ，很可能会导致 `OOM`。

所以针对以上情况，`go-zero` 开发了 `safemap` 。下面我们看看 `safemap` 是如何避免这个问题的？

## safemap