使用Channel向Go Goroutine分发任务遇死锁问题求助

嘿，这个问题我太熟了！之前用Go做批量任务处理的时候也踩过一模一样的坑——固定几个worker Goroutine处理任务，跑着跑着就死锁在半路上。先给你吃个定心丸：完全可以用Channel给运行中的Goroutine分发任务，并且在所有任务完成后优雅终止它们，你的问题大概率是Channel的关闭逻辑或者同步机制没处理对。

先分析你可能遇到的死锁原因 #

结合你说的“处理完前4个文件后陷入死锁”，我猜你应该是启动了4个固定的worker Goroutine，常见的问题有这几个：

• 没关闭任务Channel：主Goroutine把前4个文件发完后，还在继续遍历剩余文件，但如果任务Channel是无缓冲的，或者worker处理速度跟不上，主Goroutine会阻塞在发送任务；更糟的是，如果你遍历完所有文件后没关闭Channel，worker会一直阻塞在<-taskChan等待新任务，而主Goroutine又在等worker完成，形成循环等待，直接死锁。
• WaitGroup使用错误：比如你给每个任务加一次WaitGroup计数，但没在任务完成时减，或者给worker的计数没加够，导致主Goroutine的wg.Wait()一直等不到所有任务完成。
• 任务Channel缓冲不匹配：如果你的Channel是无缓冲的，且worker数量是4，主Goroutine发完4个任务后，第5个任务会因为没有worker接收而阻塞，此时所有worker都在处理任务，主Goroutine卡在发送，双方都动不了，直接死锁。

正确的实现方式 #

核心思路就是：任务分发完就关闭Channel，用WaitGroup等待所有worker退出。下面给你一个完整的示例代码，对应你的文件渲染场景：

package main

import (
	"fmt"
	"sync"
	// 根据你的需求引入其他包，比如os、image等
)

// 模拟你的文件渲染逻辑
func renderFile(filePath string) {
	// 这里替换成你实际的渲染代码
	fmt.Printf("✅ 完成文件渲染：%s\n", filePath)
}

func main() {
	// 模拟你要处理的文件列表
	files := []string{
		"chart1.json", "chart2.json", "chart3.json", "chart4.json",
		"chart5.json", "chart6.json", "chart7.json", "chart8.json",
	}
	workerCount := 4 // 你设置的4个worker Goroutine

	// 创建任务Channel，缓冲大小设为workerCount，避免主Goroutine发送时阻塞
	taskChan := make(chan string, workerCount)
	var wg sync.WaitGroup

	// 启动worker Goroutine
	for i := 0; i < workerCount; i++ {
		wg.Add(1)
		go func(workerID int) {
			defer wg.Done() // worker退出时通知WaitGroup
			fmt.Printf("🔧 Worker %d 启动，等待任务...\n", workerID)

			// 循环接收任务，直到Channel被关闭
			for file := range taskChan {
				fmt.Printf("🔄 Worker %d 开始处理：%s\n", workerID, file)
				renderFile(file)
			}

			fmt.Printf("🛑 Worker %d 无任务可处理，退出\n", workerID)
		}(i)
	}

	// 分发任务：遍历所有文件，发送到任务Channel
	fmt.Println("📤 开始分发任务...")
	for _, file := range files {
		taskChan <- file
	}
	close(taskChan) // 关键！分发完所有任务后关闭Channel，让worker知道没有新任务了

	// 等待所有worker完成所有任务
	wg.Wait()
	fmt.Println("\n🎉 所有文件渲染任务完成！")
}

关键细节解释 #

1. 关闭任务Channel：这是解决死锁的核心！当主Goroutine把所有任务都发送到Channel后，必须调用close(taskChan)。worker里的for file := range taskChan循环会在Channel为空且关闭时自动退出，不会一直阻塞等待新任务。
2. WaitGroup的正确使用：每个worker启动时调用wg.Add(1)，worker退出前用defer wg.Done()确保计数正确减少。主Goroutine的wg.Wait()会一直阻塞，直到所有worker都退出，保证所有任务都处理完毕。
3. Channel缓冲设置：把任务Channel的缓冲大小设为worker数量，可以避免主Goroutine在发送任务时频繁阻塞，提升整体处理效率。当然你也可以根据实际情况调整缓冲大小，甚至用无缓冲Channel（但要确保发送和接收的节奏匹配）。

回到你的代码问题 #

对照上面的示例，你可以检查自己的代码：

• 是不是发完所有任务后没关闭任务Channel？
• WaitGroup的计数是不是正确？有没有给每个worker都加1，并且在worker退出时减1？
• 任务Channel是不是无缓冲的，导致主Goroutine发送第5个任务时阻塞？

只要把这几个点修正，你的程序就能正常处理所有文件，不会再死锁啦。

内容的提问来源于stack exchange，提问作者bemeyer