文章首发于Secin:Golang爬虫框架初探
学到协程通信后感觉难理解了很多,目前在平时也用不到,所以Go方向就先学到这里吧,以最后的爬虫做个收尾,如果后期再用到的话再补充。
Golang标准库中文文档
http包提供了HTTP客户端和服务端的实现。
import (
"fmt"
"io/ioutil"
"net/http"
)
常规请求方式
//GET
resp, err := http.Get(url)
//POST
resp, err := http.Post("http://example.com/upload", "image/jpeg", &buf)
//PostForm
resp, err := http.PostForm("http://example.com/form",url.Values{"key": {"Value"}, "id": {"123"}})
复杂请求方式
当需要在请求的时候设置头参数、cookie之类的数据,就可以使用http.Do
方法。
client := &http.Client{}
req, _ := http.NewRequest("GET", url, nil)
//(参数头设置)
resp, err := client.Do(req)
参数头设置
设置请求header和cookie
req.Header.Set("User-Agent", "*****")
req.Header.Set("Cookie", "****")
//也可以用Add方法
req.Header.Add("User-Agent", "*****")
req.Header.Add("Cookie", "****")
区别:
发出请求
resp, err := client.Do(req)
if err != nil {
fmt.Println("Http get err:", err)
return ""
}
if resp.StatusCode != 200 {
fmt.Println("Http status code:", resp.StatusCode)
return ""
}
defer resp.Body.Close() //程序在使用完回复后必须关闭回复的主体。
body, err := ioutil.ReadAll(resp.Body) //ioutil.ReadAll用于读取文件或者网络请求
if err != nil {
fmt.Println("Read error", err)
return ""
}
重点注意下这两个地方
defer resp.Body.Close() //程序在使用完回复后必须关闭回复的主体。
body, err := ioutil.ReadAll(resp.Body) //ioutil.ReadAll用于读取文件或者网络请求
最终调用一下定义的方法,返回请求内容
package main
import (
"fmt"
"io/ioutil"
"net/http"
)
func fetch(url string) string {
client := &http.Client{}
req, _ := http.NewRequest("GET", url, nil)
req.Header.Set("User-Agent", "1")
req.Header.Set("Cookie", "1")
resp, err := client.Do(req)
//resp, err := http.Get(url) //若不需要请求头信息上边的内容换成这一条就可以了
if err != nil {
fmt.Println("Http get err:", err)
return ""
}
if resp.StatusCode != 200 {
fmt.Println("Http status code:", resp.StatusCode)
return ""
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body) //读取文件或者网络请求时
if err != nil {
fmt.Println("Read error", err)
return ""
}
return string(body)
}
func main() {
s := fetch("https://blog.csdn.net/weixin_54902210?type=lately")
fmt.Println(s)
}
多了正则表达式部分
import (
"fmt"
"io/ioutil"
"net/http"
"regexp"
"strings"
)
通过正则表达式获取当前页面的,边栏链接信息
func parse(html string) {
// 替换空格
html = strings.Replace(html, "\n", "", -1)
//正则匹配内容块
r_sidebar := regexp.MustCompile(``)
//从传入的html开始匹配
sidebar := r_sidebar.FindString(html)
//正则匹配链接
r_link := regexp.MustCompile(`href="(.*?)"`)
//从上方匹配的内容中再次匹配链接
link := r_link.FindAllString(sidebar, -1)
url := "https://gorm.io/zh_CN/docs/"
//切片遍历链接
for _, v := range link {
s := v[6 : len(v)-1]
url1 := url + s
fmt.Println(url1)
body := fetch(url)
}
}
加上前一部分的代码
package main
import (
"fmt"
"io/ioutil"
"net/http"
"regexp"
"strings"
)
func fetch(url string) string {
client := &http.Client{}
req, _ := http.NewRequest("GET", url, nil)
req.Header.Set("User-Agent", "1")
req.Header.Set("Cookie", "1")
resp, err := client.Do(req)
//resp, err := http.Get(url) //若不需要请求头信息上边的内容换成这一条就可以了
if err != nil {
fmt.Println("Http get err:", err)
return ""
}
if resp.StatusCode != 200 {
fmt.Println("Http status code:", resp.StatusCode)
return ""
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body) //读取文件或者网络请求时
if err != nil {
fmt.Println("Read error", err)
return ""
}
return string(body)
}
func parse(html string) {
// 替换空格
html = strings.Replace(html, "\n", "", -1)
//正则匹配内容块
r_sidebar := regexp.MustCompile(``)
//从传入的html开始匹配
sidebar := r_sidebar.FindString(html)
//正则匹配链接
r_link := regexp.MustCompile(`href="(.*?)"`)
//从上方匹配的内容中再次匹配链接
link := r_link.FindAllString(sidebar, -1)
url := "https://gorm.io/zh_CN/docs/"
//切片遍历链接
for _, v := range link {
s := v[6 : len(v)-1]
url1 := url + s
fmt.Println(url1)
}
}
func main() {
s := fetch("https://gorm.io/zh_CN/docs/")
parse(s)
}
成功爬取边栏链接
在成功提取边栏链接后,便可对各链接中的内容进行提取,这里以提取标题为例:
要提取的主体内容在 这里的切片从42开始是因为提取title后,前边会有长度42的源代码需要去掉 之后就是需要调用 调用 提取出我们想要的内容后,下一步就是保存内容了,实现起来也比较简单用的是os库 定义一个save方法 在 成功写入 goquery package - github.com/PuerkitoBio/goquery - Go Packages goquery为Go语言带来了一种语法和一组类似于jQuery的功能。它基于Go的net/html包和CSS Selector库cascadia。由于net/html解析器返回节点,而不是功能齐全的DOM树,因此jQuery的有状态操作函数(如height(),css(),detach())被遗漏了。Go的著名爬虫框架colly就是基于goquery实现的。 创建文档 内容查找 主要用 内容获取 属性获取 遍历 Demo 成功获取边栏链接 在爬虫的简单实现部分列举了 GoQuery不知道要对哪个HTML文档执行操作。所以需要告诉它,这就是 Document 类的用途。它保存要操作的根文档节点,并且可以对此文档进行选择。 NewDocumentFromResponse NewDocumentFromResponse 是一个 Document 构造函数,它将 http 响应作为参数。它加载指定响应的文档,对其进行分析,并存储根文档节点,以便随时进行操作。响应的主体在返回时关闭。 这里是结合前边的http库一起使用 NewDocumentFromReader NewDocumentFromReader 从 io 返回一个 Document。读者。如果无法将读取器的数据解析为 html,它将返回错误作为第二个值。它不检查读取器是否也是 io。更近一点,提供的读取器永远不会被此调用关闭。如果需要,调用方有责任将其关闭。 传入的内容其实就是html的元素数据 上述是通过http请求的方式传入的参数,并不是通过io读取,这里再用io读取举例: 除此外也可以从字符串中读取内容 结果: Demo 选择表示与某些条件匹配的节点的集合,可以使用 Document.Find 创建初始选择。 上边其实已经通过爬取页面举过例了,所以这里就简单看一下。 通过id查找 通过class查找 通过标签查找 Demo 结果: http方式中,提取了页面信息并本地保存了下来,这里再用Goquery实践一下 Final Demo 了解了一些基础爬虫和Goquery框架后,就到了Go比较核心的爬虫框架Colly 了 colly package - github.com/gocolly/colly - Go Packages Colly 是用于构建 Web 爬虫的 Golang 框架。使用 Colly 你可以构建各种复杂的 Web 抓取工具,从简单的抓取工具到处理数百万个网页的复杂的异步网站抓取工具。 Colly 提供了一个 API,用于执行网络请求和处理接收到的内容(例如,与 HTML 文档的 DOM 树进行交互)。 先看下基本实现,后续会对函数进行补充 OnRequest OnError OnResponseHeaders OnResponse OnHTML OnXML OnScraped OnXML回调后调用 回调顺序 顺序可以用代码测试出来: 除此外还有一些常用方法 设置 设置 设置环境变量 HTTP 配置 Final Demo Golang的爬虫框架先了解这么多,最后以爬取Suyoleaves师傅的博客做个结尾吧,所以可以参考上方的正则提取方式,将标题提取出来
,后边的-5也是同理
长度为5
func parse2(body string) {
// 替换空格
body = strings.Replace(body, "\n", "", -1)
//正则匹配内容块
r_content := regexp.MustCompile(`
(.*?)
`)
//从上方匹配的内容中再次匹配标题
title := r_title.FindString(content)
//切片提取title
title = title[42 : len(title)-5]
fmt.Println(title)
}
parse2()
了,在parse()
的for循环最后加上:body := fetch(url1)
//启动另外一个线程处理
go parse2(body)
fetch()
方法,将正则获取的url
的内容解析出来赋值给body
,再将body
传给parse2()
,获取标题内容保存文件
import (
"fmt"
"io/ioutil"
"net/http"
"os"
"regexp"
"strings"
)
func save(title string, content string) {
err := os.WriteFile("./crawler/file/"+title+".html", []byte(content), 0644)
if err != nil {
panic(err)
}
}
parse2()
的最后调用save(title, content)
,将标题和内容写入到文件中package main
import (
"fmt"
"io/ioutil"
"net/http"
"os"
"regexp"
"strings"
)
func fetch(url string) string {
client := &http.Client{}
req, _ := http.NewRequest("GET", url, nil)
req.Header.Set("User-Agent", "1")
req.Header.Set("Cookie", "1")
resp, err := client.Do(req)
//resp, err := http.Get(url) //若不需要请求头信息上边的内容换成这一条就可以了
if err != nil {
fmt.Println("Http get err:", err)
return ""
}
if resp.StatusCode != 200 {
fmt.Println("Http status code:", resp.StatusCode)
return ""
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body) //读取文件或者网络请求时
if err != nil {
fmt.Println("Read error", err)
return ""
}
return string(body)
}
func parse(html string) {
// 替换空格
html = strings.Replace(html, "\n", "", -1)
//正则匹配内容块
r_sidebar := regexp.MustCompile(``)
//从传入的html开始匹配
sidebar := r_sidebar.FindString(html)
//正则匹配链接
r_link := regexp.MustCompile(`href="(.*?)"`)
//从上方匹配的内容中再次匹配链接
link := r_link.FindAllString(sidebar, -1)
url := "https://gorm.io/zh_CN/docs/"
//切片遍历链接
for _, v := range link {
s := v[6 : len(v)-1]
url1 := url + s
fmt.Println(url1)
body := fetch(url1)
//启动另外一个线程处理
go parse2(body)
}
}
func parse2(body string) {
// 替换空格
body = strings.Replace(body, "\n", "", -1)
//正则匹配内容块
r_content := regexp.MustCompile(`
(.*?)
`)
//从上方匹配的内容中再次匹配标题
title := r_title.FindString(content)
title = title[42 : len(title)-5]
fmt.Println(title)
save(title, content)
}
func save(title string, content string) {
err := os.WriteFile("./crawler/file/"+title+".html", []byte(content), 0644)
if err != nil {
panic(err)
}
}
func main() {
s := fetch("https://gorm.io/zh_CN/docs/")
parse(s)
}
Goquery
安装
go get github.com/PuerkitoBio/goquery
Goquery爬虫简单实现
dom, err := goquery.NewDocument(url)
dom, err := goquery.NewDocumentFromResponse(url)
dom, err := goquery.NewDocumentFromReader(reader io.Reader)
Find()
方法,个人感觉还是通过class查找方式精确一些dom.Find("#sidebar") //通过id查找
dom.Find(".sidebar-link") //通过class查找
dom.Find("asideid").Find("a") //通过标签链式查找
selection.Html()
selection.Text()
selection.Attr("href")
selection.AttrOr("href", "")
dom.Find(".sidebar-link").Each(func(i int, selection *goquery.Selection) {
}
package main
import (
"fmt"
"github.com/PuerkitoBio/goquery"
)
func main() {
url := "https://gorm.io/zh_CN/docs/"
dom, _ := goquery.NewDocument(url)
dom.Find(".sidebar-link").Each(func(i int, selection *goquery.Selection) {
href, _ := selection.Attr("href")
text := selection.Text()
fmt.Println(i, url+href, text)
})
}
常见api
创建文档
、内容查找
等方法,这里对上述用法做一点小总结Document
client := &http.Client{}
req, _ := http.NewRequest("GET", url, nil)
resp, _ := client.Do(req)
dom, _ := goquery.NewDocumentFromResponse(resp)
dom.Find(".sidebar-link").Each(func(i int, selection *goquery.Selection) {
href, _ := selection.Attr("href")
text := selection.Text()
fmt.Println(i, url+href, text)
})
resp, _ := http.Get(url)
dom, _ := goquery.NewDocumentFromReader(resp.Body)
dom.Find(".sidebar-link").Each(func(i int, selection *goquery.Selection) {
href, _ := selection.Attr("href")
text := selection.Text()
fmt.Println(i, url+href, text)
})
f, _ := os.Open("crawler/file/Context.html")
dom, _ := goquery.NewDocumentFromReader(f)
dom.Find("h1").Each(func(i int, selection *goquery.Selection) {
text := selection.Text()
fmt.Println(text)
})
data := `
Hello
`
dom, _ := goquery.NewDocumentFromReader(strings.NewReader(data))
dom.Find("title").Each(func(i int, selection *goquery.Selection) {
text := selection.Text()
fmt.Println(text)
})
Sentiment
package main
import (
"fmt"
"github.com/PuerkitoBio/goquery"
"net/http"
"os"
)
func Document(url string) {
client := &http.Client{}
req, _ := http.NewRequest("GET", url, nil)
resp, _ := client.Do(req)
dom, _ := goquery.NewDocumentFromResponse(resp)
dom.Find(".sidebar-link").Each(func(i int, selection *goquery.Selection) {
href, _ := selection.Attr("href")
text := selection.Text()
fmt.Println(i, url+href, text)
})
}
func Document1() {
f, _ := os.Open("crawler/file/Context.html")
dom, _ := goquery.NewDocumentFromReader(f)
dom.Find("h1").Each(func(i int, selection *goquery.Selection) {
text := selection.Text()
fmt.Println(i, text)
})
}
func main() {
Document("https://gorm.io/zh_CN/docs/")
Document1()
}
选择器
data := `
Hello
`
dom, _ := goquery.NewDocumentFromReader(strings.NewReader(data))
dom.Find("#t1").Each(func(i int, selection *goquery.Selection) {
text := selection.Text()
fmt.Println(text)
})
data := `
Hello
//class="c1"
`
dom, _ := goquery.NewDocumentFromReader(strings.NewReader(data))
dom.Find(".c1").Each(func(i int, selection *goquery.Selection) {
text := selection.Text()
fmt.Println(text)
})
data := `
Hello
`
dom, _ := goquery.NewDocumentFromReader(strings.NewReader(data))
dom.Find("title").Each(func(i int, selection *goquery.Selection) {
text := selection.Text()
fmt.Println(text)
})
package main
import (
"fmt"
"github.com/PuerkitoBio/goquery"
"strings"
)
func getId() {
data := `
Hello
`
dom, _ := goquery.NewDocumentFromReader(strings.NewReader(data))
dom.Find("#t1").Each(func(i int, selection *goquery.Selection) {
text := selection.Text()
fmt.Println(text)
})
}
func getClass() {
data := `
Hello
`
dom, _ := goquery.NewDocumentFromReader(strings.NewReader(data))
dom.Find(".c1").Each(func(i int, selection *goquery.Selection) {
text := selection.Text()
fmt.Println(text)
})
}
func find() {
data := `
Hello
`
dom, _ := goquery.NewDocumentFromReader(strings.NewReader(data))
dom.Find("title").Each(func(i int, selection *goquery.Selection) {
text := selection.Text()
fmt.Println(text)
})
}
func main() {
getId() //Sentiment
getClass() //Hello
find() //Sentiment
}
Sentiment
Hello
Sentiment
Goquery爬虫重构
package main
import (
"fmt"
"github.com/PuerkitoBio/goquery"
"os"
)
func save(title string, content string) {
err := os.WriteFile("./crawler/file1/"+title+".html", []byte(content), 0644)
if err != nil {
panic(err)
}
}
func main() {
url := "https://gorm.io/zh_CN/docs/"
dom, _ := goquery.NewDocument(url)
dom.Find(".sidebar-link").Each(func(i int, selection *goquery.Selection) {
//获取链接
href, _ := selection.Attr("href")
text := selection.Text()
fmt.Println(i, url+href, text)
dom, _ = goquery.NewDocument(url + href)
//获取标题
title := dom.Find(".article-title").Text()
//fmt.Println(title)
//获取内容
content, _ := dom.Find(".article").Html()
//fmt.Println(content)
save(title, content)
})
}
Colly
安装
go get -u github.com/gocolly/colly/...
特性
Colly爬虫简单实现
package main
import (
"fmt"
"github.com/gocolly/colly"
)
func main() {
c := colly.NewCollector()
c.OnHTML(".sidebar-link", func(element *colly.HTMLElement) {
element.Request.Visit(element.Attr("href"))
})
c.OnRequest(func(request *colly.Request) {
fmt.Println(request.URL)
})
c.Visit("https://gorm.io/zh_CN/docs/")
}
回调函数
在请求之前调用,这里只是简单打印请求的 URL。
如果请求期间发生错误则调用,这里简单打印 URL 和错误信息。
在收到响应标头后调用
收到回复后调用,这里也只是简单的打印 URL 和响应大小。
OnResponse如果收到的内容是HTML ,则在之后调用
OnHTML如果接收到的内容是HTML或XML ,则在之后调用OnRequest—>OnResponse—>OnHTML->OnXML—>OnScraped
package main
import (
"fmt"
"github.com/gocolly/colly"
)
func main() {
c := colly.NewCollector()
c.OnRequest(func(r *colly.Request) {
fmt.Println("请求前调用:OnRequest")
})
c.OnError(func(_ *colly.Response, err error) {
fmt.Println("发生错误调用:OnError")
})
c.OnResponse(func(r *colly.Response) {
fmt.Println("获得响应后调用:OnResponse")
})
c.OnHTML("a[href]", func(e *colly.HTMLElement) {
fmt.Println("OnResponse收到html请求后调用:OnHTML")
})
c.OnXML("//h1", func(e *colly.XMLElement) {
fmt.Println("OnResponse收到xml内容后调用:OnXML")
})
c.OnScraped(func(r *colly.Response) {
fmt.Println("结束", r.Request.URL)
})
c.Visit("https://gorm.io/zh_CN/docs/")
}
Attr(k string)
:返回当前元素的属性,上面示例中我们使用e.Attr("href")
获取了href
属性;ChildAttr(goquerySelector, attrName string)
:返回goquerySelector
选择的第一个子元素的attrName
属性;ChildAttrs(goquerySelector, attrName string)
:返回goquerySelector
选择的所有子元素的attrName
属性,以[]string
返回;ChildText(goquerySelector string)
:拼接goquerySelector
选择的子元素的文本内容并返回;ChildTexts(goquerySelector string)
:返回goquerySelector
选择的子元素的文本内容组成的切片,以[]string
返回。ForEach(goquerySelector string, callback func(int, *HTMLElement))
:对每个goquerySelector
选择的子元素执行回调callback
;Unmarshal(v interface{})
:通过给结构体字段指定 goquerySelector 格式的 tag,可以将一个 HTMLElement 对象 Unmarshal 到一个结构体实例中。参数设置
User-Agent
c := colly.NewCollector()
//User-Agent
c.UserAgent = ("Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/103.0.5060.53 Safari/537.36 Edg/103.0.1264.37")
Cookie
c.OnRequest(func(request *colly.Request) {
request.Headers.Add("Cookie", "_ga=GA1.2.927242199.1656415218; _gid=GA1.2.165227307.1656415218; __atuvc=4%7C26; __atuvs=62bb96de45b68598000")
})
------------------------------------------------------------------
//也可以使用获取到的url中的cookie
cookie := c.Cookies("https://gorm.io/zh_CN/docs/")
c.SetCookies("", cookie)
c := colly.NewCollector()
c.WithTransport(&http.Transport{
Proxy: http.ProxyFromEnvironment,
DialContext: (&net.Dialer{
Timeout: 30 * time.Second,
KeepAlive: 30 * time.Second,
DualStack: true,
}).DialContext,
MaxIdleConns: 100,
IdleConnTimeout: 90 * time.Second,
TLSHandshakeTimeout: 10 * time.Second,
ExpectContinueTimeout: 1 * time.Second,
})
c.Visit("https://gorm.io/zh_CN/docs/")
Colly爬虫重构
package main
import (
"fmt"
"github.com/gocolly/colly"
"os"
)
func save2(title string, content string) {
err := os.WriteFile("./crawler/file2/"+title+".html", []byte(content), 0644)
if err != nil {
panic(err)
}
}
func main() {
var t1 string
var c1 string
c := colly.NewCollector()
c.OnHTML(".sidebar-link", func(e *colly.HTMLElement) {
href := e.Attr("href")
if href != "index.html" {
c.Visit(e.Request.AbsoluteURL(href))
save2(t1, c1)
}
})
c.OnHTML(".article-title", func(t *colly.HTMLElement) {
title := t.Text
t1 = title
})
c.OnHTML(".article", func(c *colly.HTMLElement) {
content, _ := c.DOM.Html()
c1 = content
})
c.OnRequest(func(request *colly.Request) {
fmt.Println(request.URL.String())
})
c.Visit("https://gorm.io/zh_CN/docs/")
}
c.Visit
开始访问第一个页面后,先进行OnRequest进行回调,之后当遇到class标签的值为sidebar-link
时回调,并取出对应的href
的值,之后再通过AbsoluteURL
获取href的绝对路径,再次通过c.Visit
请求,当遇到class的值为article-title
,article
时,再次回调对应的OnHTML
函数,并将对讲的title值和content内容分别返回给全局变量t1
、c1
,在进行一轮回调结束后执行save2()
,将内容全部写入练习
package main
import (
"fmt"
"github.com/gocolly/colly"
"os"
"time"
)
func save3(title string, content string) {
err := os.WriteFile("crawler/Sentiment/"+title+".html", []byte(content), 0644)
if err != nil {
panic(err)
}
}
func main() {
var t1 string
var c1 string
c := colly.NewCollector()
c.Limit(&colly.LimitRule{
DomainRegexp: `https://www.fishfond.guru/`,
Delay: 1 * time.Second,
})
c.OnHTML(".post-title", func(e *colly.HTMLElement) {
href := e.Attr("href")
//fmt.Println("href:", href)
c.Visit(e.Request.AbsoluteURL(href))
if href != "index.html" {
c.Visit(e.Request.AbsoluteURL(href))
save3(t1, c1)
}
})
c.OnHTML("title", func(t *colly.HTMLElement) {
title := t.Text
t1 = string(title[0 : len(title)-12])
//fmt.Println("t1:", t1)
})
c.OnHTML(".entry-content", func(c *colly.HTMLElement) {
content, _ := c.DOM.Html()
c1 = content
//fmt.Println("c1", c1)
})
c.OnRequest(func(request *colly.Request) {
fmt.Println(request.URL.String())
})
c.OnError(func(response *colly.Response, err error) {
panic(err.Error())
})
c.Visit("https://www.fishfond.guru/")
}