Golang 函数：如何避免 goroutine 泄漏？-Golang-PHP中文网

Golang 函数：如何避免 goroutine 泄漏？

PHPz

发布： 2024-09-27 18:27:02

原创

612人浏览过

为了避免 go 中的 goroutine 泄漏，可以通过使用 sync.waitgroup 确保 goroutine 结束（例如，用于异步文件下载），或使用上下文传递取消信号来管理 goroutine 生命周期（例如，在需要优雅地结束 goroutine 时）。

Golang 函数：如何避免 goroutine 泄漏？

简介

Goroutine 是 Go 语言中的轻量级协程，用于并发编程。但是，如果 goroutine 未正确管理，可能会导致泄漏，从而消耗系统资源并导致应用程序不稳定。

goroutine 泄漏

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

goroutine 泄漏发生在以下情况下：

创建一个 goroutine 但从未结束
创建一个 goroutine，但它没有引用任何其他程序代码，因此在主程序函数退出后仍然运行

避免 goroutine 泄漏

有几种方法可以避免 goroutine 泄漏：

如知AI笔记

如知笔记——支持markdown的在线笔记，支持ai智能写作、AI搜索，支持DeepseekR1满血大模型

查看详情

1. 确保 goroutine 结束

使用 sync.WaitGroup 来确保 goroutine 在结束前等待所有正在进行的操作。例如：

package main

import (
    "fmt"
    "sync"
)

func main() {
    var wg sync.WaitGroup

    // 创建并启动 goroutine
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go func(n int) {
            defer wg.Done()
            fmt.Println(n)
        }(i)
    }

    // 等待所有 goroutine 结束
    wg.Wait()
}

登录后复制

2. 使用上下文

使用 [context.Context](https://go.dev/blog/context) 通过传递取消信号来管理 goroutine 的生命周期。当上下文被取消时，所有关联的 goroutine 将优雅地结束。例如：

package main

import (
    "context"
    "fmt"
    "time"
)

func main() {
    ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
    defer cancel()

    go func(ctx context.Context) {
        for {
            select {
            case <-ctx.Done():
                fmt.Println("Gracefully ending goroutine")
                return
            default:
                fmt.Println("Working...")
            }
        }
    }(ctx)

    time.Sleep(time.Second * 10)
}

登录后复制

实战案例

问题：需要异步下载多个文件，但要确保在主程序退出时所有下载都已完成。

解决方案：使用 sync.WaitGroup 确保在主程序退出前等待所有下载 goroutine 结束：

package main

import (
    "fmt"
    "io"
    "net/http"
    "os"
    "sync"
)

func main() {
    urls := []string{"url1", "url2", "url3"}
    var wg sync.WaitGroup

    // 创建并启动下载 goroutine
    for _, url := range urls {
        wg.Add(1)
        go downloadFile(url, wg)
    }

    // 等待所有下载结束
    wg.Wait()
}

func downloadFile(url string, wg sync.WaitGroup) {
    defer wg.Done()

    resp, err := http.Get(url)
    if err != nil {
        fmt.Println("Error downloading file:", err)
        return
    }
    defer resp.Body.Close()

    f, err := os.Create(url + ".txt")
    if err != nil {
        fmt.Println("Error creating file:", err)
        return
    }
    defer f.Close()

    _, err = io.Copy(f, resp.Body)
    if err != nil {
        fmt.Println("Error copying file:", err)
        return
    }

    fmt.Println("File downloaded successfully:", url)
}

登录后复制

以上就是Golang 函数：如何避免 goroutine 泄漏？的详细内容，更多请关注php中文网其它相关文章！