如何在 Goroutine 中使用闭包捕获局部变量？

在 goroutine 中，闭包可用于捕获局部变量，使其在 goroutine 执行期间依然可用。通过将局部变量作为匿名函数的参数传递，可以在 goroutine 启动时捕获该变量。通过使用闭包，可以在并发执行的 goroutine 之间共享和修改数据，同时确保线程安全。

如何在 Goroutine 中使用闭包捕获局部变量

闭包是一个函数，它可以访问在其创建时就已经存在的外部变量或资源。在 Go 中，闭包通过使用匿名函数来创建。

捕获局部变量

当我们在 Goroutine 中启动匿名函数时，该函数可以捕获 Goroutine 启动时存在的所有局部变量。

以下是如何捕获局部变量的代码示例：

package main

import (
    "fmt"
    "time"
)

func main() {
    i := 0

    for i < 10 {
        go func(i int) {
            fmt.Println(i)
        }(i)
        i++
    }

    time.Sleep(time.Second)
}

登录后复制

实战案例

我们可以使用闭包来创建一个并行执行的计数器。每个 Goroutine 都有自己的私有计数器，可以安全地并发访问和修改：

package main

import (
    "fmt"
    "sync"
    "sync/atomic"
    "time"
)

func main() {
    var wg sync.WaitGroup
    var counter uint64

    for i := 0; i < 10; i++ {
        wg.Add(1)
        go func(i int) {
            defer wg.Done()
            for j := 0; j < 1000000; j++ {
                // 使用原子操作确保并发安全
                atomic.AddUint64(&counter, 1)
            }
            fmt.Println("Goroutine", i, "finished")
        }(i)
    }

    wg.Wait()
    fmt.Println("Total count:", counter)
}

登录后复制

这样，我们并行创建了 10 个 Goroutine，每个 Goroutine 都有自己的计数器，并发执行 100 万次自增操作，最后我们得到了总计数。由于每个 Goroutine 都使用闭包捕获了自己的局部计数器变量，因此并发访问和修改计数器是线程安全的。

以上就是如何在 Goroutine 中使用闭包捕获局部变量？的详细内容，更多请关注叮当号网其它相关文章！