关于 Go 的内存分配
在 Go
语言里,从内存的分配到不再使用后内存的回收等等这些内存管理工作都是由 Go
在底层完成的。虽然开发者在写代码时不必过度关心内存从分配到回收这个过程,但是 Go
的内存分配策略里有不少有意思的设计,通过了解他们有助于我们自身的提高,也让我们能写出更高效的 Go
程序。
Go 内存管理的设计旨在在并发环境中快速运行,并与垃圾回收器集成在一起。让我们看一个简单的示例:
package main
type smallStruct struct {
a, b int64
c, d float64
}
func main() {
smallAllocation()
}
//go:noinline
func smallAllocation() *smallStruct {
return &smallStruct{}
}
函数上面的注释//go:noinline
将禁止Go
对该函数进行内联,这样main
函数就会使用smallAllocation
函数返回的指针变量,因为被多个函数使用,返回的这个变量将被分配到堆上。
关于内联的概念之前的文章有说过:
内联是一种手动或编译器优化,用于将简短函数的调用替换为函数体本身。这么做的原因是它可以消除函数调用本身的开销,也使得编译器能更高效地执行其他的优化策略。
所以如果上面的例子不干预编译器的话,编译器通过内联将smallAllocation
函数体里的内容直接放到main
函数里,这样就不会产生smallAllocation
这个函数的调用了,所有的变量都是main
函数内这个范围使用的,也就不在需要将变量往堆上分配了。
继续说上面那个例子,通过逃逸分析命令 go tool compile -m main.go 可以确认我们上面的分析,&smallStruct{}
会被分配到堆上去。
➜ go tool compile -m main.go
main.go:12:6: can inline main
main.go:10:9: &smallStruct literal escapes to heap
借助命令go tool compile -S main.go,可以显示该程序的汇编代码,也可以明确地向我们展示内存的分配:
0x001d 00029 (main.go:10) LEAQ type."".smallStruct(SB), AX
0x0024 00036 (main.go:10) PCDATA $2, $0
0x0024 00036 (main.go:10) MOVQ AX, (SP)
0x0028 00040 (main.go:10) CALL runtime.newobject(SB)
内置函数newobject
会通过调用另外一个内置函数mallocgc
在堆上分配新内存。在Go里面有两种内存分配策略,一种适用于程序里小内存块的申请,另一种适用于大内存块的申请,大内存块指的是大于32KB。
下面我们来细聊一下这两种策略。
小于 32KB 内存块的分配策略
当程序里发生了32kb
以下的小块内存申请时,Go会从一个叫做的mcache
的本地缓存给程序分配内存。这个本地缓存mcache
持有一系列的大小为32kb
的内存块,这样的一个内存块里叫做mspan
,它是要给程序分配内存时的分配单元。
在Go的调度器模型里,每个线程M
会绑定给一个处理器P
,在单一粒度的时间里只能做多处理运行一个goroutine
,每个P
都会绑定一个上面说的本地缓存mcache
。当需要进行内存分配时,当前运行的goroutine
会从mcache
中查找可用的mspan
。从本地mcache
里分配内存时不需要加锁,这种分配策略效率更高。
那么有人就会问了,有的变量很小就是数字,有的却是一个复杂的结构体,申请内存时都分给他们一个mspan
这样的单元会不会产生浪费。其实mcache
持有的这一系列的mspan
并不都是统一大小的,而是按照大小,从8字节到32KB分了大概70类的msapn
。
就文章开始的那个例子来说,那个结构体的大小是32字节,正好32字节的这种mspan
能满足需求,那么分配内存的时候就会给它分配一个32字节大小的mspan
。
现在,我们可能会好奇,如果分配内存时mcachce
里没有空闲的32字节的mspan
了该怎么办?Go
里还为每种类别的mspan
维护着一个mcentral
。
mcentral
的作用是为所有mcache
提供切分好的mspan
资源。每个central
会持有一种特定大小的全局mspan
列表,包括已分配出去的和未分配出去的。每个mcentral
对应一种mspan
,当工作线程的mcache
中没有合适(也就是特定大小的)的mspan
时就会从mcentral
去获取。mcentral
被所有的工作线程共同享有,存在多个goroutine
竞争的情况,因此从mcentral
获取资源时需要加锁。
mcentral
的定义如下:
//runtime/mcentral.go
type mcentral struct {
// 互斥锁
lock mutex
// 规格
sizeclass int32
// 尚有空闲object的mspan链表
nonempty mSpanList
// 没有空闲object的mspan链表,或者是已被mcache取走的msapn链表
empty mSpanList
// 已累计分配的对象个数
nmalloc uint64
}
mcentral
里维护着两个双向链表,nonempty表示链表里还有空闲的mspan
待分配。empty表示这条链表里的mspan
都被分配了object
。
如果上面我们那个程序申请内存的时候,mcache
里已经没有合适的空闲mspan
了,那么工作线程就会像下图这样去mcentral
里去申请。
简单说下mcache
从mcentral
获取和归还mspan
的流程:
- 获取 加锁;从
nonempty
链表找到一个可用的mspan
;并将其从nonempty
链表删除;将取出的mspan
加入到empty
链表;将mspan
返回给工作线程;解锁。 - 归还 加锁;将
mspan
从empty
链表删除;将mspan
加入到nonempty
链表;解锁。
当mcentral
没有空闲的mspan
时,会向mheap
申请。而mheap
没有资源时,会向操作系统申请新内存。mheap
主要用于大对象的内存分配,以及管理未切割的mspan
,用于给mcentral
切割成小对象。
同时我们也看到,mheap
中含有所有规格的mcentral
,所以,当一个mcache
从mcentral
申请mspan
时,只需要在独立的mcentral
中使用锁,并不会影响申请其他规格的mspan
。
上面说了每种尺寸的mspan
都有一个全局的列表存放在mcentral
里供所有线程使用,所有mcentral
的集合则是存放于mheap
中的。mheap
里的arena
区域是真正的堆区,运行时会将 8KB
看做一页,这些内存页中存储了所有在堆上初始化的对象。运行时使用二维的 runtime.heapArena 数组管理所有的内存,每个 runtime.heapArena 都会管理 64MB 的内存。
如果 arena
区域没有足够的空间,会调用 runtime.mheap.sysAlloc 从操作系统中申请更多的内存。
大于32KB内存块的分配策略
Go 没法使用工作线程的本地缓存 mcache
和全局中心缓存 mcentral
上管理超过 32KB 的内存分配,所以对于那些超过32KB的内存申请,会直接从堆上(mheap
)上分配对应的数量的内存页(每页大小是8KB)给程序。
总结
我们把内存分配管理涉及的所有概念串起来,可以勾画出 Go 内存管理的一个全局视图:
Go语言的内存分配非常复杂,这个文章从一个比较粗的角度来看Go的内存分配,并没有深入细节。一般而言,了解它的原理,到这个程度也就可以了(应付面试)。
总结起来关于Go内存分配管理的策略有如下几点:
- Go在程序启动时,会向操作系统申请一大块内存,由
mheap
结构全局管理。 - Go内存管理的基本单元是
mspan
,每种mspan
可以分配特定大小的object
。 mcache
,mcentral
,mheap
是Go
内存管理的三大组件,mcache
管理线程在本地缓存的mspan
;mcentral
管理全局的mspan
供所有线程使用;mheap
管理Go
的所有动态分配内存。- 一般小对象通过
mspan
分配内存;大对象则直接由mheap
分配内存。
评论区