Golang 闭包的性能优化技巧

前言
Go 语言中的闭包是一个函数，它可以访问函数外定义的变量。闭包可以捕获值，为不同的goroutine 提供一致的状态。然而，过度使用闭包可能会导致性能问题。本文介绍了优化 Golang 闭包性能的实用技巧。

优化技巧

1. 避免捕获大变量
闭包捕获所有引用变量的值，因此捕获大变量会增加内存占用和 GC 压力。尽量仅捕获小变量或值类型。

立即学习“go语言免费学习笔记（深入）”；

func main() {
    largeArray := make([]int, 1000000) // 避免捕获大数组
    foo := func() {
        fmt.Println(largeArray)
    }
    foo()
}

2. 减少闭包创建频率
重复创建的闭包会导致不必要的内存分配。考虑使用闭包函数指针来避免重复创建。

Golang Uber Dig

使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器，提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象，支持命名依赖等

下载

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
    }
}