golang函数闭包的常见陷阱-Golang-PHP中文网

函数闭包在 go 中使用时有陷阱：变量共享：多个闭包引用同一变量时修改变量会影响所有闭包行为。变量类型转换：稍后将捕获变量类型转换为另一种类型会导致不可预期的结果。实战中，生成序列的函数闭包问题可以通过为每个闭包创建局部变量副本来解决，以避免修改捕获的变量。

golang函数闭包的常见陷阱

Go 函数闭包的常见陷阱

在 Go 中，闭包是指引用其创建时的局部变量的函数。它可以提供程序的灵活性，但如果使用不当，也会带来意外行为和难以调试的问题。

闭包陷阱

以下是一些常见的函数闭包陷阱：

1. 变量共享

当多个闭包引用同一个局部变量时，修改该变量将影响所有闭包的行为。例如：

func counter() int {
    count := 0
    return func() int {
        count++
        return count
    }
}

// 两个同时调用的闭包共享计数变量
c1 := counter()
c2 := counter()
c1()
c2()

会导致返回的值为 2，而不是预期的 1 和 2。

2. 变量类型转换

闭包捕获变量的类型，如果稍后将该变量类型转换为另一种类型，可能会导致不可预期的结果。例如：

Golang Uber Dig

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

下载

func intToString() string {
    i := 42
    return func() string {
        // 会返回一个空字符串，而不是 "42"
        return strconv.Itoa(i)
    }
}

因为闭包捕获的变量 i 被转换为 string 类型，导致 strconv.Itoa() 无法正确转换。

实战案例

期望：创建一个函数，根据传入的数字生成一个序列，直到达到最大值。

代码：

func generateSequence(max int) []int {
    seq := []int{}
    i := 0
    return func() int {
        i++
        if i > max {
            return 0
        }
        seq = append(seq, i)
        return i
    }
}

问题：此函数的闭包会捕获 i 变量，并在调用闭包时对其进行修改。但是，我们期望闭包每次被调用时都会返回一个新的 i 值。

修复：可以通过为每个闭包创建局部变量副本来解决此问题。

func generateSequence(max int) []int {
    seq := []int{}
    return func() int {
        i := 0
        i++
        if i > max {
            return 0
        }
        seq = append(seq, i)
        return i
    }
}

注意：修复后的代码不会再修改捕获到的 i 变量，而是每次创建一个新的副本。