先聊聊「内存分配」,再聊聊Go的「逃逸分析」

开发 后端
通过本文的介绍,相信你一定加深了堆栈的理解;搞清楚逃逸分析的作用和原理之后能够指导我们写出更优雅的代码。

要搞清楚GO的逃逸分析一定要先搞清楚内存分配和堆栈:

内存分配既可以分配到堆中,也可以分配到栈中。

什么样的数据会被分配到栈中,什么样的数据又会被分配到堆中呢?

GO语言是如何进行内存分配的呢?其设计初衷和实现原理是什么呢?

我们先来聊一下内存管理、堆、栈的知识点:

内存管理

内存管理主要包括两个动作:分配与释放。逃逸分析就是服务于内存分配。

在Go中,栈的内存是由编译器自动进行分配和释放,栈区往往存储着函数参数、局部变量和调用函数帧,它们随着函数的创建而分配,函数的退出而销毁。

一个goroutine对应一个栈,栈是调用栈(call stack)的简称。一个栈通常又包含了许多栈帧(stack frame),它描述的是函数之间的调用关系,每一帧对应一个尚未返回的函数调用,它本身也是以栈形式存放数据。

与栈不同的是,应用程序在运行时只会存在一个堆。

我们可以简单理解为:我们用GO语言开发过程中,要考虑的内存管理只是针对堆内存而言的。

程序在运行期间可以主动从堆上申请内存,这些内存通过Go的内存分配器分配,并由垃圾收集器回收。

堆和栈的对比

加锁

  • 栈不需要加锁:栈是每个goroutine独有的,这就意味着栈上的内存操作是不需要加锁的。
  • 堆有时需要加锁:堆上的内存,有时需要加锁防止多线程冲突

延伸知识点:为什么堆上的内存有时需要加锁?而不是一直需要加锁呢?

因为Go的内存分配策略学习了TCMalloc的线程缓存思想,他为每个处理器P分配了一个mcache,从mcache分配内存也是无锁的

性能

  • 栈内存管理 性能好:栈上的内存,它的分配与释放非常高效的。简单地说,它只需要两个CPU指令:一个是分配入栈,另外一个是栈内释放。只需要借助于栈相关寄存器即可完成。
  • 堆内存管理 性能差:对于程序堆上的内存回收,还需要通过标记清除阶段,例如Go采用的三色标记法。

缓存策略

  • 栈缓存性能更好
  • 堆缓存性能较差

原因是:栈内存能更好地利用CPU的缓存策略,因为栈空间相较于堆来说是更连续的。

逃逸分析

上面说了这么多堆和栈的知识点,目的是为了让大家更好的理解逃逸分析。

正如上面讲的,相比于把内存分配到堆中,分配到栈中优势更明显。

Go语言也是这么做的:Go编译器会尽可能将变量分配到到栈上。

但是,当编译器无法证明函数返回的变量有没有被引用时,编译器就必须在堆上分配该变量,以此避免悬挂指针(dangling pointer)的问题。

另外,如果局部变量占用内存非常大,也会将其分配在堆上。

Go是如何确定内存是分配到栈上还是堆上的呢?

答案就是:逃逸分析。

编译器通过逃逸分析技术去选择堆或者栈,逃逸分析的基本思想如下:检查变量的生命周期是否是完全可知的,如果通过检查,则在栈上分配。否则,就是所谓的逃逸,必须在堆上进行分配。

逃逸分析原则

Go语言虽然没有明确说明逃逸分析原则,但是有以下几点准则,是可以参考的。

  • 不同于JAVA JVM的运行时逃逸分析,Go的逃逸分析是在编译期完成的:编译期无法确定的参数类型必定放到堆中;
  • 如果变量在函数外部存在引用,则必定放在堆中;
  • 如果变量占用内存较大时,则优先放到堆中;
  • 如果变量在函数外部没有引用,则优先放到栈中;

逃逸分析举例

我们使用这个命令来查看逃逸分析的结果:go build -gcflags '-m -m -l'

1.参数是interface类型

package main

import "fmt"

func main() {
a := 666
fmt.Println(a)
}

运行结果

图片

原因分析

因为Println(a ...interface{})的参数是interface{}类型,编译期无法确定其具体的参数类型,所以内存分配到堆中。

图片

2. 变量在函数外部有引用

package main

func test() *int {
a := 10
return &a
}

func main() {
_ = test()
}

运行结果

图片

原因分析

变量a在函数外部存在引用。

我们来分析一下执行过程:当函数执行完毕,对应的栈帧就被销毁,但是引用已经被返回到函数之外。如果这时外部通过引用地址取值,虽然地址还在,但是这块内存已经被释放回收了,这就是非法内存。

为了避免上述非法内存的情况,在这种情况下变量的内存分配必须分配到堆上。

3. 变量内存占用较大

package main

func test() {
a := make([]int, 10000, 10000)
for i := 0; i < 10000; i++ {
a[i] = i
}
}

func main() {
test()
}

运行结果

图片

原因分析

我们定义了一个容量为10000的int类型切片,发生了逃逸,内存分配到了堆上(heap)。

注意看:

我们再简单修改一下代码,将切片的容量和长度修改为1,再次查看逃逸分析的结果,我们发现,没有发生逃逸,内存默认分类到了栈上。

图片

所以,当变量占用内存较大时,会发生逃逸分析,将内存分配到堆上。

4. 变量大小不确定时

我们再简单修改一下上面的代码:

package main

func test() {
l := 1
a := make([]int, l, l)
for i := 0; i < l; i++ {
a[i] = i
}
}

func main() {
test()
}

运行结果

图片

原因分析

我们通过控制台的输出结果可以很明显的看出:发生了逃逸,分配到了heap堆中。

原因是这样的:

我们虽然在代码段中给变量 l 赋值了1,但是编译期间只能识别到初始化int类型切片时,传入的长度和容量是变量l,编译期并不能确定变量l的值,所以发生了逃逸,会把内存分配到堆中。

思考题

好了,我们举了4个逃逸分析的经典案例,相信聪明的你已经理解了逃逸分析的作用和发生逃逸的场景。

我们来想一下,在理解逃逸分析的原理之后,在开发的过程中如何更好的编码,进而提高程序的效率,更好的利用内存呢?

如何实践?

理解逃逸分析一定能帮助我们写出更好的程序。知道变量分配在栈堆之上的差别后,我们就要尽量写出分配在栈上的代码。因为堆上的变量变少后,可以减轻内存分配的开销,减小GC的压力,提高程序的运行速度。

但是我们也要有过犹不及的指导思想。

我认为没有一成不变的开发模式,我们一定是在不断的需求变化,业务变化中求得平衡的:

举个例子

举个日常开发中函数传参的例子:

有些场景下我们不应该传递结构体指针,而应该直接传递结构体。

为什么会这样呢?虽然直接传递结构体需要值拷贝,但是这是在栈上完成的操作,开销远比变量逃逸后动态地在堆上分配内存少的多。

当然这种做法不是绝对的,要根据场景去分析:

  • 如果结构体较大,传递结构体指针更合适,因为指针类型相比值类型能节省大量的内存空间
  • 如果结构体较小,传递结构体更适合,因为在栈上分配内存,可以有效减少GC压力

总结

通过本文的介绍,相信你一定加深了堆栈的理解;搞清楚逃逸分析的作用和原理之后能够指导我们写出更优雅的代码。

我们在日常开发中,要根据实际场景考虑,如何将内存尽量分配到栈中,减少GC的压力,提高性能。

如何找到应用开发效率、程序运行效率、对机器的压力及负载的平衡点,是程序员进阶之旅中的必修课。

本文转载自微信公众号「 程序员升级打怪之旅」,作者「王中阳Go」,可以通过以下二维码关注。

转载本文请联系「 程序员升级打怪之旅」公众号。

责任编辑:武晓燕 来源: 程序员升级打怪之旅
相关推荐

2023-01-10 09:18:37

Go内存分配逃逸

2023-01-28 08:32:04

Go内存分配

2021-12-16 06:52:33

C语言内存分配

2017-03-16 13:17:54

TDD代码开发

2023-03-26 23:47:32

Go内存模型

2021-05-30 19:29:12

内存Go语言

2022-10-17 09:03:52

2024-04-07 11:33:02

Go逃逸分析

2022-02-11 09:31:23

IPV4IP地址IANA

2023-12-22 07:55:38

Go语言分配策略

2021-02-22 09:30:09

go开发环境桌面系统

2021-01-07 07:53:10

JavaScript内存管理

2017-03-16 13:28:34

TDD代码软件架构

2022-12-12 08:42:06

Java对象栈内存

2022-07-25 15:38:59

Go 语言Go 语言编译器内存逃逸

2024-01-15 06:42:00

高并发热点账户数据库

2023-01-30 23:04:10

B-Treegolang优化

2021-03-28 20:58:25

Go语言线程

2021-04-29 09:02:44

语言Go 处理

2022-09-28 11:34:27

用户行为数据业务
点赞
收藏

51CTO技术栈公众号