Go 语言内置运行时(就是 runtime),抛弃了传统的内存分配方式,改为自主管理。这样可以自主地实现更好的内存使用模式,比如内存池、预分配等等。这样,不会每次内存分配都需要进行系统调用。

Golang 运行时的内存分配算法主要源自 Google 为 C 语言开发的TCMalloc 算法,全称Thread-Caching Malloc。核心思想就是把内存分为多级管理,从而降低锁的粒度。它将可用的堆内存采用二级分配的方式进行管理:每个线程都会自行维护一个独立的内存池,进行内存分配时优先从该内存池中分配,当内存池不足时才会向全局内存池申请,以避免不同线程对全局内存池的频繁竞争。

为了更好的阅读体验,手动贴上文章目录:

图解Go语言内存分配 - 知乎 - 图1

基础概念

Go 在程序启动的时候,会先向操作系统申请一块内存(注意这时还只是一段虚拟的地址空间,并不会真正地分配内存),切成小块后自己进行管理。

申请到的内存块被分配了三个区域,在 X64 上分别是 512MB,16GB,512GB 大小。

图解Go语言内存分配 - 知乎 - 图2

arena 区域就是我们所谓的堆区,Go 动态分配的内存都是在这个区域,它把内存分割成8KB大小的页,一些页组合起来称为mspan

bitmap 区域标识arena区域哪些地址保存了对象,并且用4bit标志位表示对象是否包含指针、GC标记信息。bitmap中一个byte大小的内存对应arena区域中 4 个指针大小(指针大小为 8B )的内存,所以bitmap区域的大小是512GB/(4*8B)=16GB

图解Go语言内存分配 - 知乎 - 图3
图解Go语言内存分配 - 知乎 - 图4

从上图其实还可以看到 bitmap 的高地址部分指向 arena 区域的低地址部分,也就是说 bitmap 的地址是由高地址向低地址增长的。

spans 区域存放mspan(也就是一些arena分割的页组合起来的内存管理基本单元,后文会再讲)的指针,每个指针对应一页,所以spans区域的大小就是512GB/8KB*8B=512MB。除以 8KB 是计算arena区域的页数,而最后乘以 8 是计算spans区域所有指针的大小。创建mspan的时候,按页填充对应的spans区域,在回收object时,根据地址很容易就能找到它所属的mspan

内存管理单元

mspan:Go 中内存管理的基本单元,是由一片连续的8KB的页组成的大块内存。注意,这里的页和操作系统本身的页并不是一回事,它一般是操作系统页大小的几倍。一句话概括:mspan是一个包含起始地址、mspan规格、页的数量等内容的双端链表。

每个mspan按照它自身的属性Size Class的大小分割成若干个object,每个object可存储一个对象。并且会使用一个位图来标记其尚未使用的object。属性Size Class决定object大小,而mspan只会分配给和object尺寸大小接近的对象,当然,对象的大小要小于object大小。还有一个概念:Span Class,它和Size Class的含义差不多,

  1. Size_Class = Span_Class / 2

这是因为其实每个 Size Class有两个mspan,也就是有两个Span Class。其中一个分配给含有指针的对象,另一个分配给不含有指针的对象。这会给垃圾回收机制带来利好,之后的文章再谈。

如下图,mspan由一组连续的页组成,按照一定大小划分成object

图解Go语言内存分配 - 知乎 - 图5

Go1.9.2 里mspanSize Class共有 67 种,每种mspan分割的 object 大小是 8*2n 的倍数,这个是写死在代码里的:

  1. // path: /usr/local/go/src/runtime/sizeclasses.go
  2. const _NumSizeClasses = 67
  3. var class_to_size = [_NumSizeClasses]uint16{0, 8, 16, 32, 48, 64, 80, 96, 112, 128, 144, 160, 176, 192, 208, 224, 240, 256, 288, 320, 352, 384, 416, 448, 480, 512, 576, 640, 704, 768, 896, 1024, 1152, 1280, 1408, 1536,1792, 2048, 2304, 2688, 3072, 3200, 3456, 4096, 4864, 5376, 6144, 6528, 6784, 6912, 8192, 9472, 9728, 10240, 10880, 12288, 13568, 14336, 16384, 18432, 19072, 20480, 21760, 24576, 27264, 28672, 32768}

根据mspanSize Class可以得到它划分的object大小。 比如Size Class等于 3,object大小就是 32B。 32B 大小的 object 可以存储对象大小范围在 17B~32B 的对象。而对于微小对象(小于 16B),分配器会将其进行合并,将几个对象分配到同一个object中。

数组里最大的数是 32768,也就是 32KB,超过此大小就是大对象了,它会被特别对待,这个稍后会再介绍。顺便提一句,类型Size Class为 0 表示大对象,它实际上直接由堆内存分配,而小对象都要通过mspan来分配。

对于 mspan 来说,它的Size Class会决定它所能分到的页数,这也是写死在代码里的:

  1. // path: /usr/local/go/src/runtime/sizeclasses.go
  2. const _NumSizeClasses = 67
  3. var class_to_allocnpages = [_NumSizeClasses]uint8{0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 1, 2, 1, 2, 1, 3, 2, 3, 1, 3, 2, 3, 4, 5, 6, 1, 7, 6, 5, 4, 3, 5, 7, 2, 9, 7, 5, 8, 3, 10, 7, 4}

比如当我们要申请一个object大小为32Bmspan的时候,在 class_to_size 里对应的索引是 3,而索引 3 在class_to_allocnpages数组里对应的页数就是 1。

mspan结构体定义:

  1. // path: /usr/local/go/src/runtime/mheap.go
  2. type mspan struct {
  3. //链表前向指针,用于将span链接起来
  4. next *mspan
  5. //链表前向指针,用于将span链接起来
  6. prev *mspan
  7. // 起始地址,也即所管理页的地址
  8. startAddr uintptr
  9. // 管理的页数
  10. npages uintptr
  11. // 块个数,表示有多少个块可供分配
  12. nelems uintptr
  13. //分配位图,每一位代表一个块是否已分配
  14. allocBits *gcBits
  15. // 已分配块的个数
  16. allocCount uint16
  17. // class表中的class ID,和Size Classs相关
  18. spanclass spanClass
  19. // class表中的对象大小,也即块大小
  20. elemsize uintptr
  21. }

我们将mspan放到更大的视角来看:

图解Go语言内存分配 - 知乎 - 图6

上图可以看到有两个S指向了同一个mspan,因为这两个S指向的P是同属一个mspan的。所以,通过arena上的地址可以快速找到指向它的S,通过S就能找到mspan,回忆一下前面我们说的mspan区域的每个指针对应一页。

假设最左边第一个mspanSize Class等于 10,根据前面的class_to_size数组,得出这个msapn分割的object大小是 144B,算出可分配的对象个数是8KB/144B=56.89个,取整 56 个,所以会有一些内存浪费掉了,Go 的源码里有所有Size Classmspan浪费的内存的大小;再根据class_to_allocnpages数组,得到这个mspan只由 1 个page组成;假设这个mspan是分配给无指针对象的,那么spanClass等于 20。

startAddr直接指向arena区域的某个位置,表示这个mspan的起始地址,allocBits指向一个位图,每位代表一个块是否被分配了对象;allocCount则表示总共已分配的对象个数。

这样,左起第一个mspan的各个字段参数就如下图所示:

图解Go语言内存分配 - 知乎 - 图7

内存管理组件

内存分配由内存分配器完成。分配器由 3 种组件构成:mcache, mcentral, mheap

mcache

mcache:每个工作线程都会绑定一个 mcache,本地缓存可用的mspan资源,这样就可以直接给 Goroutine 分配,因为不存在多个 Goroutine 竞争的情况,所以不会消耗锁资源。

mcache的结构体定义:

  1. //path: /usr/local/go/src/runtime/mcache.go
  2. type mcache struct {
  3. alloc [numSpanClasses]*mspan
  4. }
  5. numSpanClasses = _NumSizeClasses << 1

mcacheSpan Classes作为索引管理多个用于分配的mspan,它包含所有规格的mspan。它是_NumSizeClasses的 2 倍,也就是67*2=134,为什么有一个两倍的关系,前面我们提到过:为了加速之后内存回收的速度,数组里一半的mspan中分配的对象不包含指针,另一半则包含指针。

对于无指针对象的mspan在进行垃圾回收的时候无需进一步扫描它是否引用了其他活跃的对象。 后面的垃圾回收文章会再讲到,这次先到这里。

图解Go语言内存分配 - 知乎 - 图8

mcache在初始化的时候是没有任何mspan资源的,在使用过程中会动态地从mcentral申请,之后会缓存下来。当对象小于等于 32KB 大小时,使用mcache的相应规格的mspan进行分配。

mcentral

mcentral:为所有mcache提供切分好的mspan资源。每个central保存一种特定大小的全局mspan列表,包括已分配出去的和未分配出去的。 每个mcentral对应一种mspan,而mspan的种类导致它分割的object大小不同。当工作线程的mcache中没有合适(也就是特定大小的)的mspan时就会从mcentral获取。

mcentral被所有的工作线程共同享有,存在多个 Goroutine 竞争的情况,因此会消耗锁资源。结构体定义:

  1. //path: /usr/local/go/src/runtime/mcentral.go
  2. type mcentral struct {
  3. // 互斥锁
  4. lock mutex
  5. // 规格
  6. sizeclass int32
  7. // 尚有空闲object的mspan链表
  8. nonempty mSpanList
  9. // 没有空闲object的mspan链表,或者是已被mcache取走的msapn链表
  10. empty mSpanList
  11. // 已累计分配的对象个数
  12. nmalloc uint64
  13. }

图解Go语言内存分配 - 知乎 - 图9

empty表示这条链表里的mspan都被分配了object,或者是已经被cache取走了的mspan,这个mspan就被那个工作线程独占了。而nonempty则表示有空闲对象的mspan列表。每个central结构体都在mheap中维护。

简单说下mcachemcentral获取和归还mspan的流程:

  • 获取 加锁;从nonempty链表找到一个可用的mspan;并将其从nonempty链表删除;将取出的mspan加入到empty链表;将mspan返回给工作线程;解锁。
  • 归还 加锁;将mspanempty链表删除;将mspan加入到nonempty链表;解锁。

mheap

mheap:代表 Go 程序持有的所有堆空间,Go 程序使用一个mheap的全局对象_mheap来管理堆内存。

mcentral没有空闲的mspan时,会向mheap申请。而mheap没有资源时,会向操作系统申请新内存。mheap主要用于大对象的内存分配,以及管理未切割的mspan,用于给mcentral切割成小对象。

同时我们也看到,mheap中含有所有规格的mcentral,所以,当一个mcachemcentral申请mspan时,只需要在独立的mcentral中使用锁,并不会影响申请其他规格的mspan

mheap结构体定义:

  1. //path: /usr/local/go/src/runtime/mheap.go
  2. type mheap struct {
  3. lock mutex
  4. // spans: 指向mspans区域,用于映射mspan和page的关系
  5. spans []*mspan
  6. // 指向bitmap首地址,bitmap是从高地址向低地址增长的
  7. bitmap uintptr
  8. // 指示arena区首地址
  9. arena_start uintptr
  10. // 指示arena区已使用地址位置
  11. arena_used uintptr
  12. // 指示arena区末地址
  13. arena_end uintptr
  14. central [67*2]struct {
  15. mcentral mcentral
  16. pad [sys.CacheLineSize - unsafe.Sizeof(mcentral{})%sys.CacheLineSize]byte
  17. }
  18. }

图解Go语言内存分配 - 知乎 - 图10

上图我们看到,bitmap 和 arena_start 指向了同一个地址,这是因为 bitmap 的地址是从高到低增长的,所以他们指向的内存位置相同。

分配流程

上一篇文章《Golang 之变量去哪儿》中我们提到了,变量是在栈上分配还是在堆上分配,是由逃逸分析的结果决定的。通常情况下,编译器是倾向于将变量分配到栈上的,因为它的开销小,最极端的就是 “zero garbage”,所有的变量都会在栈上分配,这样就不会存在内存碎片,垃圾回收之类的东西。

Go 的内存分配器在分配对象时,根据对象的大小,分成三类:小对象(小于等于 16B)、一般对象(大于 16B,小于等于 32KB)、大对象(大于 32KB)。

大体上的分配流程:

  • 32KB 的对象,直接从 mheap 上分配;
  • <=16B 的对象使用 mcache 的 tiny 分配器分配;
  • (16B,32KB] 的对象,首先计算对象的规格大小,然后使用 mcache 中相应规格大小的 mspan 分配;
  • 如果 mcache 没有相应规格大小的 mspan,则向 mcentral 申请
  • 如果 mcentral 没有相应规格大小的 mspan,则向 mheap 申请
  • 如果 mheap 中也没有合适大小的 mspan,则向操作系统申请

总结

Go 语言的内存分配非常复杂,它的一个原则就是能复用的一定要复用。源码很难追,后面可能会再来一篇关于内存分配的源码阅读相关的文章。简单总结一下本文吧。

文章从一个比较粗的角度来看 Go 的内存分配,并没有深入细节。一般而言,了解它的原理,到这个程度也可以了。

  • Go 在程序启动时,会向操作系统申请一大块内存,之后自行管理。
  • Go 内存管理的基本单元是 mspan,它由若干个页组成,每种 mspan 可以分配特定大小的 object。
  • mcache, mcentral, mheap 是 Go 内存管理的三大组件,层层递进。mcache 管理线程在本地缓存的 mspan;mcentral 管理全局的 mspan 供所有线程使用;mheap 管理 Go 的所有动态分配内存。
  • 极小对象会分配在一个 object 中,以节省资源,使用 tiny 分配器分配内存;一般小对象通过 mspan 分配内存;大对象则直接由 mheap 分配内存。

参考资料

【简单易懂,非常清晰】https://yq.aliyun.com/articles/652551

【内存分配器的初始化过程,分配流程图很详细】https://www.jianshu.com/p/47691d870756

【全局的图】https://swanspouse.github.io/2018/08/22/golang-memory-model/

【雨痕 Go1.5 源码阅读】https://github.com/qyuhen/book

【图不错】https://www.jianshu.com/p/47691d870756

【整体感】https://juejin.im/post/59f2e19f5188253d6816d504

【源码解读】http://legendtkl.com/2017/04/02/golang-alloc/

【重点推荐 深入到晶体管了 图很好】https://www.linuxzen.com/go-memory-allocator-visual-guide.html

【总体描述对象分配流程】http://gocode.cc/project/4/article/103

【实际 Linux 命令】https://mikespook.com/2014/12/%E7%90%86%E8%A7%A3-go-%E8%AF%AD%E8%A8%80%E7%9A%84%E5%86%85%E5%AD%98%E4%BD%BF%E7%94%A8/

【整体流程图 对象分配函数调用链路】http://blog.newbmiao.com/2018/08/20/go-source-analysis-of-memory-alloc.html

【源码讲解 非常细致】https://www.cnblogs.com/zkweb/p/7880099.html

【源码阅读】https://zhuanlan.zhihu.com/p/34930748

原文:https://zhuanlan.zhihu.com/p/59125443