Go语言的内存分配由编译器自动管理，开发者不需要手动申请和释放内存，但了解变量在栈和堆上的分配规则，能够帮助我们写出性能更优的代码。栈是线程私有的内存区域，分配和回收速度极快，而堆是共享内存区域，分配和回收需要垃圾回收器参与，开销相对更大。
Go内存分配的基本规则
Go编译器会通过逃逸分析来决定变量的分配位置，逃逸分析的核心判断标准是：如果变量的生命周期可以明确限定在当前函数内，且不需要被外部引用，那么变量会分配在栈上；如果变量的引用会逃逸出当前函数，或者无法确定其生命周期，那么变量会分配在堆上。
常见的栈分配场景
以下情况中变量通常会被分配在栈上：
- 函数内部定义的局部变量，且没有将地址返回给外部
- 局部变量的大小在编译期可以确定，且不超过栈的最大限制
- 变量没有在函数外部被引用，也没有被闭包捕获
常见的堆分配场景
以下情况中变量通常会被分配在堆上：
- 将局部变量的指针返回给函数外部
- 局部变量被闭包捕获，生命周期超出当前函数
- 变量的大小在编译期无法确定，比如动态创建的切片、map等
- 变量被发送到channel中，或者被存储在全局变量里
代码示例验证分配位置
我们可以通过go build -gcflags="-m"命令查看编译器的逃逸分析结果，判断变量的分配位置。下面是几个典型示例：
栈分配示例
package main
func add(a, b int) int {
// 局部变量c只在函数内部使用，不会逃逸，分配在栈上
c := a + b
return c
}
func main() {
result := add(1, 2)
println(result)
}
执行go build -gcflags="-m"后，不会看到该变量的逃逸提示，说明c分配在栈上。
堆分配示例
package main
// 返回局部变量的指针，变量会逃逸到堆上
func getPointer() *int {
x := 10
return &x
}
func main() {
p := getPointer()
println(*p)
}
执行逃逸分析命令后，会看到leaked to heap的提示，说明x分配在堆上。
Pointer与内存分配的关系
Go中的Pointer类型是用来表示指针的特殊类型，它不能直接进行指针运算，主要用于和C语言交互或者进行底层内存操作。Pointer本身不会影响变量的分配位置，但是使用Pointer传递变量地址时，可能会触发逃逸分析。
比如下面的示例中，使用Pointer传递变量地址，变量依然会逃逸到堆上：
package main
import "unsafe"
func usePointer(p unsafe.Pointer) {
// 接收Pointer参数，原变量的引用会被传递到函数外部
_ = p
}
func main() {
x := 20
// 将x的地址转换为Pointer传递给外部函数，x会逃逸到堆上
usePointer(unsafe.Pointer(&x))
}
如果Pointer指向的变量没有被外部引用，那么变量依然会分配在栈上：
package main
import "unsafe"
func main() {
x := 30
// Pointer只在函数内部使用，没有传递到外部，x分配在栈上
p := unsafe.Pointer(&x)
println(*(*int)(p))
}
内存分配的优化建议
为了减少堆分配带来的垃圾回收压力，我们可以尽量遵循以下原则：
- 尽量不要返回局部变量的指针，除非确实需要
- 避免不必要的闭包捕获大对象
- 对于频繁创建的小对象，可以考虑使用对象池复用
- 编写代码后可以通过逃逸分析命令检查不必要的堆分配，针对性优化
理解Go的栈堆分配规则，能够帮助我们更好地把握代码的性能瓶颈，写出更符合Go语言设计理念的程序。