当前位置：首页 > news >正文

Go编程：使用 Colly 库下载Reddit网站的图像

news 2025/8/18 19:15:20

亿牛云

概述

Reddit是一个社交新闻网站，用户可以发布各种主题的内容，包括图片。本文将介绍如何使用Go语言和Colly库编写一个简单的爬虫程序，从Reddit网站上下载指定主题的图片，并保存到本地文件夹中。为了避免被目标网站反爬，我们还将使用亿牛云爬虫代理服务，通过动态切换代理IP来提高爬取效率和稳定性。

正文

Colly库简介

Colly是一个用Go语言编写的功能强大的爬虫框架。它提供了简洁的API，拥有强劲的性能，可以自动处理cookie和session，还有提供灵活的扩展机制。Colly支持多种数据格式的解析，如HTML、XML、JSON等，还支持分布式爬取、限速、缓存、重试等功能。

亿牛云爬虫代理简介

亿牛云爬虫代理是一种专业的代理服务，通过固定云代理服务地址，建立专线网络链接，代理平台自动实现海量IP池管理及负载均衡，实时无感的毫秒级代理IP切换，提供企业级云服务的网络稳定性及请求响应速度，同时降低了客户端运算负载压力，避免了爬虫客户在代理IP策略优化上投入精力，整体提升了爬虫效率。

爬虫程序设计

我们的爬虫程序主要分为以下几个步骤：

创建一个Colly实例，并设置相关选项和回调函数
登录亿牛云后台，获取代理相关信息（域名、端口、用户名、密码），并设置到Colly实例中
访问Reddit网站，根据指定的主题（subreddit）和过滤条件（filter），获取图片链接
下载图片，并保存到本地文件夹中

爬虫程序代码

package mainimport ("fmt""io""log""net/http""net/url""os""path/filepath""strings""github.com/gocolly/colly/v2"
)// 定义常量
const (subreddit = "pics" // 主题filter    = "top"  // 过滤条件limit     = 10     // 图片数量限制output    = "images" // 输出文件夹// 亿牛云爬虫代理相关信息（需登录后台获取）proxyHost = "www.16yun.cn" // 域名proxyPort = 6443           // 端口proxyUser = "16YUN"        // 用户名proxyPass = "16IP"         // 密码
)func main() {// 创建一个Colly实例c := colly.NewCollector(colly.AllowedDomains("www.reddit.com", "old.reddit.com", "i.redd.it", "i.imgur.com"),)// 设置代理（使用亿牛云爬虫代理服务）err := c.SetProxyFunc(func(r *http.Request) (*url.URL, error) {return url.Parse(fmt.Sprintf("http://%s:%s@%s:%d", proxyUser, proxyPass, proxyHost, proxyPort))})if err != nil {log.Fatal(err)}// 创建一个计数器，用于限制图片数量count := 0// 注册HTML回调函数，用于解析图片链接c.OnHTML("a[href]", func(e *colly.HTMLElement) {// 获取链接地址link := e.Attr("href")// 过滤非图片链接if !strings.HasSuffix(link, ".jpg") && !strings.HasSuffix(link, ".png") {return}// 限制图片数量if count >= limit {return}// 访问图片链接c.Visit(link)})// 注册图片回调函数，用于下载图片c.OnResponse(func(r *colly.Response) {// 获取图片URLurl := r.Request.URL.String()// 获取图片文件名filename := filepath.Base(url)// 创建输出文件夹（如果不存在）if _, err := os.Stat(output); os.IsNotExist(err) {os.Mkdir(output, 0755)}// 创建图片文件file, err := os.Create(filepath.Join(output, filename))if err != nil {log.Fatal(err)}defer file.Close()// 写入图片数据file.Write(r.Body)fmt.Printf("下载图片：%s\n", url)count++})c.OnError(func(r *colly.Response, err error) {fmt.Printf("请求失败：%s\n", r.Request.URL)fmt.Println(err)})c.OnRequest(func(r *colly.Request) {fmt.Printf("请求开始：%s\n", r.URL)})c.OnResponse(func(r *colly.Response) {fmt.Printf("请求完成：%s\n", r.Request.URL)})c.OnScraped(func(r *colly.Response) {fmt.Println("爬取结束")})fmt.Println("爬取开始")c.Visit(fmt.Sprintf("https://old.reddit.com/r/%s/%s/", subreddit, filter))
}