0

0

Go 并发爬虫中正确判断任务完成与通道关闭的时机

霞舞

霞舞

发布时间:2026-01-07 15:55:45

|

431人浏览过

|

来源于php中文网

原创

Go 并发爬虫中正确判断任务完成与通道关闭的时机

go 并发爬虫中,不能依赖 channel 长度或盲目关闭 channel 来判断任务结束;应使用 `sync.waitgroup` 精确跟踪活跃 goroutine 数量,并在所有任务完成后统一关闭 channel 或终止主流程。

在实现并发 Web 爬虫(如 Go Tour 中的 Exercise: Web Crawler)时,一个常见误区是试图通过检查 channel 缓冲区长度(如 len(stor.Queue) == 0)来判断“是否还有任务待处理”,进而决定何时关闭 channel 或退出主循环。这种做法不仅逻辑错误(channel 长度仅反映当前缓冲区中未读数据量,无法反映已启动但尚未入队、或正在执行中的 goroutine),而且极易导致死锁提前终止——正如原代码中注释所警示的那样。

正确的方案是:用 sync.WaitGroup 显式管理 goroutine 生命周期。其核心思想是:

  • 每启动一个新 goroutine 执行 Crawl,就调用 wg.Add(1);
  • 每个 Crawl 函数结束前,必须调用 defer wg.Done();
  • 主 goroutine 调用 wg.Wait() 阻塞等待,直到所有子任务完成。

这种方式不依赖 channel 状态,也不需要共享 channel 引用或手动关闭 channel,彻底规避了竞态与死锁风险。

以下是关键实践要点与优化后的结构化示例:

Tome
Tome

先进的AI智能PPT制作工具

下载

推荐写法(无 channel 队列,纯 WaitGroup 驱动):

package main

import (
    "fmt"
    "sync"
)

var wg sync.WaitGroup
var visited = make(map[string]int) // 全局共享,需注意并发安全(本例中无写竞争,因每个 URL 首次访问才写)

type Result struct {
    Url   string
    Depth int
}

type Fetcher interface {
    Fetch(url string) (body string, urls []string, err error)
}

func Crawl(res Result, fetcher Fetcher) {
    defer wg.Done() // 确保无论何种路径退出,都计数减一

    if res.Depth <= 0 {
        return
    }

    url := res.Url
    if visited[url] > 0 { // 已访问过,跳过
        fmt.Println("skip:", url)
        return
    }
    visited[url] = 1 // 标记为已访问(首次写入)

    body, urls, err := fetcher.Fetch(url)
    if err != nil {
        fmt.Printf("fetch error %s: %v\n", url, err)
        return
    }
    fmt.Printf("found: %s %q\n", url, body)

    // 为每个新 URL 启动并发爬取
    for _, u := range urls {
        wg.Add(1)
        go Crawl(Result{u, res.Depth - 1}, fetcher)
    }
}

func main() {
    wg.Add(1)
    go Crawl(Result{"http://golang.org/", 4}, fetcher)
    wg.Wait() // 主协程在此阻塞,直到所有爬取完成
}

⚠️ 注意事项:

  • 避免共享可变状态的竞争:本例中 visited 是全局 map,虽当前逻辑保证每个 URL 最多被一个 goroutine 首次写入(因 visited[url] > 0 判断在写之前),但在更复杂场景中,应使用 sync.Map 或 mutex 保护。
  • *不要传递 `Stor或 channel 给每个 goroutine**:原设计中stor.Queue` 作为任务分发通道,反而引入了关闭时机难题;WaitGroup 方案更简洁、正交,职责分离清晰(goroutine 管理 vs 数据通信)。
  • 若仍需 channel 收集结果:可在 Crawl 中将结果发送到一个只读 channel(如 results chan<- string),由主 goroutine 单独接收并关闭;此时 channel 关闭应由主 goroutine 在 wg.Wait() 后显式执行,而非任何 worker。

? 总结:判断“是否还有数据/任务”的本质,不是观察 channel 的瞬时状态,而是追踪任务的生命周期。sync.WaitGroup 是 Go 官方推荐、轻量且可靠的同步原语,适用于所有“主协程等待一组子协程完成”的场景。在并发爬虫中,它让逻辑更健壮、代码更易读、调试更简单。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

WorkBuddy
WorkBuddy

腾讯云推出的AI原生桌面智能体工作台

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
string转int
string转int

在编程中,我们经常会遇到需要将字符串(str)转换为整数(int)的情况。这可能是因为我们需要对字符串进行数值计算,或者需要将用户输入的字符串转换为整数进行处理。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

1030

2023.08.02

golang map内存释放
golang map内存释放

本专题整合了golang map内存相关教程,阅读专题下面的文章了解更多相关内容。

77

2025.09.05

golang map相关教程
golang map相关教程

本专题整合了golang map相关教程,阅读专题下面的文章了解更多详细内容。

40

2025.11.16

golang map原理
golang map原理

本专题整合了golang map相关内容,阅读专题下面的文章了解更多详细内容。

67

2025.11.17

java判断map相关教程
java判断map相关教程

本专题整合了java判断map相关教程,阅读专题下面的文章了解更多详细内容。

47

2025.11.27

Golang channel原理
Golang channel原理

本专题整合了Golang channel通信相关介绍,阅读专题下面的文章了解更多详细内容。

261

2025.11.14

golang channel相关教程
golang channel相关教程

本专题整合了golang处理channel相关教程,阅读专题下面的文章了解更多详细内容。

351

2025.11.17

C# ASP.NET Core微服务架构与API网关实践
C# ASP.NET Core微服务架构与API网关实践

本专题围绕 C# 在现代后端架构中的微服务实践展开,系统讲解基于 ASP.NET Core 构建可扩展服务体系的核心方法。内容涵盖服务拆分策略、RESTful API 设计、服务间通信、API 网关统一入口管理以及服务治理机制。通过真实项目案例,帮助开发者掌握构建高可用微服务系统的关键技术,提高系统的可扩展性与维护效率。

76

2026.03.11

Go高并发任务调度与Goroutine池化实践
Go高并发任务调度与Goroutine池化实践

本专题围绕 Go 语言在高并发任务处理场景中的实践展开,系统讲解 Goroutine 调度模型、Channel 通信机制以及并发控制策略。内容包括任务队列设计、Goroutine 池化管理、资源限制控制以及并发任务的性能优化方法。通过实际案例演示,帮助开发者构建稳定高效的 Go 并发任务处理系统,提高系统在高负载环境下的处理能力与稳定性。

38

2026.03.10

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Go 教程
Go 教程

共32课时 | 6.2万人学习

Go语言实战之 GraphQL
Go语言实战之 GraphQL

共10课时 | 0.9万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号