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

有志者，事竟成！如果你在学习Golang，那么本文《Golang 函数：如何避免 goroutine 泄漏？》，就很适合你！文章讲解的知识点主要包括，若是你对本文感兴趣，或者是想搞懂其中某个知识点，就请你继续往下看吧~

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

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

简介

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

goroutine 泄漏

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

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

避免 goroutine 泄漏

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

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 泄漏？》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！