

新闻资讯
技术学院使用 `sync.waitgroup` 配合带缓冲的通道和结构化结果类型,是 go 中处理未知深度递归爬虫并安全关闭通道的惯用方案。
在 Go 的并发编程中,递归启动 Goroutine(如网页爬虫)时,常面临一个经典难题:如何在所有子 Goroutine 完成后,优雅地停止从结果通道读取,避免死锁或资源泄漏? 由于递归分支数量动态不可知,无法预先关闭通道;而若在主 Goroutine 中直接 close() 通道,又可能因竞态导致 panic 或漏读数据。
标准、符合 Go 惯用法(idiomatic Go)的解法是 “WaitGroup + 结构化结果通道 + 单独消费协程” 模式:
以下是关键逻辑精简示例(省略 fakeFetcher 等辅助代码):
func Crawl(wg *sync.WaitGroup, url string, depth int, fetcher Fetcher, cache *UrlCache, results *Results) {
defer wg.Done()
if depth <= 0 || !cache.AtomicSet(url) {
return
}
body, urls, err := fetcher.Fetch(url)
if err != nil {
results.Error <- err // 缓冲通道,不会阻塞
return
}
results.Data <- [2]string{url, body}
for _, u := range urls {
wg.Add(1)
go Crawl(wg, u, depth-1, fetcher, cache, results)
}
}
func main() {
var wg sync.WaitGroup
cache := NewUrlCache()
results := NewResults()
defer results.Close() // 确保退出前关闭通道
wg.Add(1)
go Crawl(&wg, "http://golang.org/", 4, fetcher, cache, results)
go results.Read() // 启动非阻塞消费者
wg.Wait() // 等待所有爬取完成
}⚠️ 注意事项:
后 这正是 Tour of Go 第 73 节所期望的思维范式:用组合代替继承,用明确的同步原语(WaitGroup)替代隐式控制流,用结构化通道通信替代共享内存——简洁、健壮、且一眼可知其并发契约。