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

本文涉及的产品
检索分析服务 Elasticsearch 版,2核4GB开发者规格 1个月
实时数仓Hologres,5000CU*H 100GB 3个月
智能开放搜索 OpenSearch行业算法版,1GB 20LCU 1个月
简介: Reddit是一个社交新闻网站,用户可以发布各种主题的内容,包括图片。本文将介绍如何使用Go语言和Colly库编写一个简单的爬虫程序,从Reddit网站上下载指定主题的图片,并保存到本地文件夹中。为了避免被目标网站反爬,我们还将使用爬虫代理服务,通过动态切换代理IP来提高爬取效率和稳定性。

亿牛云.png

概述

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

正文

Colly库简介

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

亿牛云爬虫代理简介

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

爬虫程序设计

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

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

爬虫程序代码

package main

import (
    "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) {
   
   
        // 获取图片URL
        url := 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))
}

结语

本文介绍了如何使用Go语言和Colly库编写一个简单的爬虫程序,从Reddit网站上下载指定主题的图片,并保存到本地文件夹中。同时,我们还使用了亿牛云爬虫代理服务,通过动态切换代理IP来提高爬取效率和稳定性,希望本文对你有所帮助。

相关文章
|
14天前
|
Shell Go API
Go语言grequests库并发请求的实战案例
Go语言grequests库并发请求的实战案例
|
8天前
|
存储 缓存 Go
go语言编程系列(五)
go语言编程系列(五)
|
8天前
|
搜索推荐 Java 编译器
go语言编程系列(四)
go语言编程系列(四)
|
8天前
|
存储 JSON 安全
go语言编程系列(七)
go语言编程系列(七)
|
8天前
|
存储 安全 编译器
go语言编程系列(六)
go语言编程系列(六)
|
8天前
|
自然语言处理 Java 测试技术
go语言编程系列(二)
go语言编程系列(二)
|
8天前
|
编译器 Go
go语言编程系列(三)
go语言编程系列(三)
|
8天前
|
安全 Java Go
Go语言编程系列(一)
Go语言编程系列(一)
|
27天前
|
Kubernetes Go Docker
在K8s编程中如何使用Go
一文带你了解在K8s编程中如何使用Go
44 3
|
2月前
|
存储 JSON 前端开发
一文搞懂 Go 1.21 的日志标准库 - slog
一文搞懂 Go 1.21 的日志标准库 - slog
56 2
下一篇
无影云桌面