个人技术分享

协程泄漏(Goroutine Leakage)是指那些已经没有任何用处（不再被使用或者无法到到达其执行路径），但由于某些原因未被收回的goroutine。这些泄漏的goroutine占用内存资源，可能会随着程序运行时间的增长而累积，最终导致内存耗尽或者程序性能下降。goroutine leakage的原因主要有以下几种：

1. 长时间运行或未正确终止的goroutine:

如果goroutine在执行长时间任务时没有适当的退出条件，或者其执行路径没有正确终结，那么这些goroutine就会一直存在。

2. 未处理的通道(Channel)操作：

• 发送操作未被接受： 如果一个goroutine向通道发送数据，而没有其他的goroutine来接收这些数据， 发送者可能会永远阻塞在那里，特别是在使用无缓冲通道的情况下。
• 接受操作没有数据可接受： 同样，如果一个goroutine在等待从通道接收数据，而这个通道再也没有数据发送，该goroutine也会永远阻塞。

3. 阻塞的系统调用：

一些系统调用或者操作，如文件操作或者网络请求，可能会因为外部操作不满足而阻塞，如果这些操作没有设置超时处理，相应的goroutine可能会永久阻塞。

4. 资源锁定：

goroutine在等待如互斥锁等同步原语时，如果锁由于编程错误而不被释放，依赖这些锁的goroutine可能会永久阻塞。

5. 循环等待：

如果goroutine之间形成了循环等待的死锁，这些goroutine都将无法进展。

解决方法：

1. 使用context控制goroutine的生命周期

package main

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

func doSomething(ctx context.Context) {
	for {
		select {
		case <-ctx.Done():
			fmt.Println("goroutine exiting")
			return // 正确退出goroutine
		default:
			// 模拟工作
			fmt.Println("working...")
			time.Sleep(1 * time.Second)
		}
	}
}

func main() {
    // 3s后自动取消
	ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
	defer cancel()
	// 监听，收到信号后退出
	go doSomething(ctx)

	// 主函数等待足够时间
	time.Sleep(5 * time.Second)
	fmt.Println("main function exiting")
}

2. 为通道设置超时操作

发送操作：

package main

import (
	"fmt"
	"time"
)

func main() {
	ch := make(chan int)

	go func() {
		select {
		case ch <- 1:
			fmt.Println("sent value")
		case <-time.After(1 * time.Second):
			fmt.Println("timeout on send")
		}
	}()

	time.Sleep(2 * time.Second) // 模拟延迟，没有接收器
}

接收操作：

package main

import (
	"fmt"
	"time"
)

func main() {
	ch := make(chan int)

	go func() {
		select {
		case v := <-ch:
			fmt.Println("received value", v)
		case <-time.After(1 * time.Second):
			fmt.Println("timeout on receive")
		}
	}()

	time.Sleep(2 * time.Second) // 模拟延迟，没有发送者
}

4. 避免死锁

// 确保多个`goroutine`不会因为循环依赖锁而产生死锁，确保多个`goroutine`按照相同的顺序获取锁
package main

import (
	"fmt"
	"sync"
	"time"
)

func main() {
	var lockA sync.Mutex
	var lockB sync.Mutex

	go func() {
		lockA.Lock()
		fmt.Println("Goroutine 1: Locked A")
		time.Sleep(1 * time.Second) // 模拟工作
		lockB.Lock()
		fmt.Println("Goroutine 1: Locked B")
		lockB.Unlock()
		lockA.Unlock()
	}()

	go func() {
		lockB.Lock()
		fmt.Println("Goroutine 2: Locked B")
		time.Sleep(1 * time.Second) // 模拟工作
		lockA.Lock()
		fmt.Println("Goroutine 2: Locked A")
		lockA.Unlock()
		lockB.Unlock()
	}()

	time.Sleep(3 * time.Second) // 等待足够时间
	fmt.Println("main function exiting")
}

5. sync.WaitGroup

sync.WaitGroup 主要用于等待一组 goroutine 完成，而并不直接解决 goroutine 泄漏的问题。它通过计数 goroutine 的数量来同步等待所有的 goroutine 正确退出。使用 WaitGroup 可以确保在所有相关的 goroutine 都执行完毕后，程序的主流程才会继续执行，从而避免在所有 goroutine 还未完成时程序就结束了。但如果 goroutine 中存在无限循环或阻塞等待资源的情况而没有适当的退出条件，使用 WaitGroup 也无法解决这些 goroutine 的泄漏问题。

package main

import (
	"fmt"
	"sync"
	"time"
)

func worker(id int, wg *sync.WaitGroup) {
	defer wg.Done() // 在函数退出时通知 WaitGroup 这个 goroutine 已完成
	fmt.Printf("Worker %d starting\n", id)
	time.Sleep(time.Second)
	fmt.Printf("Worker %d done\n", id)
}

func main() {
	var wg sync.WaitGroup

	for i := 1; i <= 5; i++ {
		wg.Add(1) // 为每个 goroutine 增加计数
		go worker(i, &wg)
	}

	wg.Wait() // 等待所有 goroutine 完成
	fmt.Println("All workers done")
}

在这个例子中，wg.Wait() 调用会阻塞，直到所有通过 wg.Add() 注册的 goroutine 都调用了 wg.Done()，从而确保所有 goroutine 都完成了它们的任务。然而，如果 goroutine 中存在逻辑错误或资源死锁，WaitGroup 并不能自动解决这些问题。

最后给大家推荐一个LinuxC/C++高级架构系统教程的学习资源与课程，可以帮助你有方向、更细致地学习C/C++后端开发，具体内容请见 https://xxetb.xetslk.com/s/1o04uB