为了避免 go 中的 goroutine 泄漏，可以通过使用 sync.waitgroup 确保 goroutine 结束（例如，用于异步文件下载），或使用上下文传递取消信号来管理 goroutine 生命周期（例如，在需要优雅地结束 goroutine 时）。
Golang 函数：如何避免 goroutine 泄漏？
简介
Goroutine 是 Go 语言中的轻量级协程，用于并发编程。但是，如果 goroutine 未正确管理，可能会导致泄漏，从而消耗系统资源并导致应用程序不稳定。
goroutine 泄漏
立即学习“go语言免费学习笔记（深入）”；
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)
} 
