Golang如何实现简单的函数闭包

闭包是引用外部变量的函数值,可“记住”定义时环境。如counter函数返回操作count的闭包,每次调用均累加并返回新值;循环中创建闭包需注意变量绑定问题。

在Go语言中,闭包是通过匿名函数捕获其外部作用域中的变量来实现的。闭包让函数可以“记住”定义时的环境,即使外部函数已经执行完毕,内部函数依然能访问那些变量。

什么是闭包

闭包是一个函数值,它引用了其外部函数中的变量。这些被引用的变量会随着闭包的生命周期继续存在,即使外部函数已经返回。

简单示例:计数器

下面是一个使用闭包实现计数器的例子:

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

func main() {
    c := counter()
    fmt.Println(c()) // 输出: 1
    fmt.Println(c()) // 输出: 2
    fmt.Println(c()) // 输出: 3
}

在这个例子中,count 是外部函数 counter 中的局部变量。返回的匿名函数引用了这个变量,形成了闭包。每次调用 c(),都会修改并返回更新后的 count 值。

捕获循环变量的注意事项

在使用闭包时,特别是在循环中创建多个闭包,需要注意变量绑定问题:

for i := 0; i < 3; i++ {
    go func() {
        fmt.Println(i)
    }()
}

上面代码可能会全部输出 3,因为所有 goroutine 共享同一个变量 i。正确的做法是将变量作为参数传入:

for i := 0; i < 3; i++ {
    go func(val int) {
        fmt.Println(val)
    }(i)
}

或者在循环内创建一个新的变量副本:

for i := 0; i < 3; i++ {
    i := i // 创建局部副本
    go func() {
        fmt.Println(i)
    }()
}

基本上就这些。Go的闭包简洁实用,适合用于状态保持、延迟计算和函数式编程风格的场景。关键是理解变量的引用关系,避免意外共享。