Golang 闭包的性能优化技巧
前言
Go 语言中的闭包是一个函数,它可以访问函数外定义的变量。闭包可以捕获值,为不同的goroutine 提供一致的状态。然而,过度使用闭包可能会导致性能问题。本文介绍了优化 Golang 闭包性能的实用技巧。
优化技巧
1. 避免捕获大变量
闭包捕获所有引用变量的值,因此捕获大变量会增加内存占用和 GC 压力。尽量仅捕获小变量或值类型。
立即学习“go语言免费学习笔记(深入)”;
func main() {
largeArray := make([]int, 1000000) // 避免捕获大数组
foo := func() {
fmt.Println(largeArray)
}
foo()
}
2. 减少闭包创建频率
重复创建的闭包会导致不必要的内存分配。考虑使用闭包函数指针来避免重复创建。
func main() {
foo := func() {
fmt.Println("Hello, world!")
}
for i := 0; i < 1000000; i++ {
go foo() // 重复创建闭包
}
}
改为:
func main() {
foo := func() {
fmt.Println("Hello, world!")
}
for i := 0; i < 1000000; i++ {
go foo // 使用闭包函数指针
}
}
3. 避免在闭包中修改外部变量
在闭包中修改外部变量会触发写屏障,增加 GC 开销。尽量在闭包之外修改外部变量。
func main() {
var counter int
foo := func() {
counter++ // 避免在闭包中修改外部变量
}
for i := 0; i < 1000000; i++ {
go foo()
}
}
改为:
func main() {
var counter int
foo := func() {
tmp := counter // 在闭包内创建临时变量
tmp++
counter = tmp
}
for i := 0; i < 1000000; i++ {
go foo()
}
}
实战案例
考虑以下示例:
import "context"
func main() {
ctx := context.Background()
for i := 0; i < 100; i++ {
go func() {
fmt.Println(ctx)
}()
}
}
在此示例中,每个goroutine 都捕获了 ctx 的副本,即使所有goroutine 都使用相同的值。可以通过使用闭包函数指针来优化此示例:
import "context"
func main() {
ctx := context.Background()
foo := func() {
fmt.Println(ctx)
}
for i := 0; i < 100; i++ {
go foo
}
}