首次 golang爬虫插件gocolly/colly 使用经历

glqEason · · 2389 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

# 首次 golang爬虫插件gocolly/colly 使用经历 ##涉及问题 > * 各种包 例如:golang.org/x/net, golang/x/test 等的下载问题,有可能要翻墙 > * golang 与数据库交互 > * golang 文件读写 > * golang 多线程使用 > * golang 字符编码的转换 > * js 标签选择器 ## 参考 系列文章 https://www.cnblogs.com/majianguo/p/8146110.html # start ### js 选择器的问题 1). 一个controller 可以定义多个OnHTML 回调函数 2). "div > p" --> div 的直接子元素, "div p" --> div 下的所有子元素 ### golang数据库连接问题 参考 : https://blog.csdn.net/webxscan/article/details/70174658 ```golang import ( "database/sql" _ "github.com/go-sql-driver/mysql" ) //数据库配置 const ( userName = "root" password = "" ip = "127.0.0.1" port = "3306" dbName = "dbName" ) path := strings.Join([]string{userName, ":", password, "@tcp(",ip, ":", port, ")/", dbName, "?charset=utf8"}, "") DB, _ := sql.Open("mysql", path) //验证连接 if errConn := DB.Ping(); errConn != nil{ fmt.Println("open database fail") return } fmt.Println("connnect success") defer DB.Close() stmt, err := DB.Prepare("insert into user(name age) values(?, ?)") if err != nil { fmt.Println(err) } res, err := stmt.Exec("username", 18) if err != nil { fmt.Println(err) } // 获取新插入行的id fmt.Println(res.LastInsertId()) ``` ### golang文件写入 ```golang var fileName = "./flag.txt" var file *os.File var err error func main() { file = openFile(fileName) writeFile(file, "keep coding!!") } func openFile(fileName string) *os.File { if checkFileIsExist(fileName) { //如果文件存在 file, err = os.OpenFile(fileName, os.O_APPEND, 0666) } else { //创建文件 file, err = os.Create(fileName) } check(err) return file } func writeFile(file *os.File, content string) { writer := bufio.NewWriter(file) writer.WriteString(content) writer.WriteString("\r\n") writer.Flush() } func check(e error) { if e != nil { panic(e) } } func checkFileIsExist(filename string) bool { var exist = true if _, err := os.Stat(filename); os.IsNotExist(err) { exist = false } return exist } ``` ### golang字符编码的转换 抓取链家网数据时,网页本身是utf8编码,没有问题,可是抓取房天下的数据时,网页本身是gb2312,折腾了好一会, F**K ```golang //src为要转换的字符串 func coverGBKToUTF8(src string) string { // 网上搜有说要调用translate函数的,实测不用 return mahonia.NewDecoder("gbk").ConvertString(src) } ``` ### golang多线程,控制消息同步 附上测试代码,便于理解 ```golang func main() { // 我们还可以创建一个带缓冲的channel: //c := make(chan int, 1024) // 从带缓冲的channel中读数据 //for i:= range c { //} //此时,创建一个大小为1024的int类型的channel,即使没有读取方,写入方也可以一直往channel里写入,在缓冲区被填完之前都不会阻塞。 chs := make([] chan int, 10) for i := 0; i < 10; i ++ { go func(i int) { chs[i] = make(chan int) count(chs[i], i) //fmt.Println("run thread ", i) // 打印要放在向信道发消息之前 }(i) } for _, ch := range chs { value := <- ch close(ch) fmt.Println(value, " thread done") } fmt.Println("All done") } func count(ch chan int, i int) { ch <- i // 向信道发消息的过程一定要放在协程内部,才不会被主进程阻塞 } ``` ##最后,附上抓取放天下小区信息的源代码 ```golang package main import ( "bufio" "database/sql" "fmt" "github.com/PuerkitoBio/goquery" "github.com/axgle/mahonia" _ "github.com/go-sql-driver/mysql" "github.com/gocolly/colly" "os" "reflect" "strconv" "strings" ) //数据库配置 const ( userName = "root" password = "" ip = "127.0.0.1" port = "3306" dbName = "dbName" ) type XQinfo struct { name string addr string area string postCode string propertyRight string propertyType string buildTime string developer string buildType string buildArea string buildStruct string floorSpace string manageCompany string greenRatio string plotRatio string propretyFee string AdditionalInfo string waterSupply string heatSupply string elecSupply string gas string security string environment string parkingSpace string OtherInfo string } var flagCh = make(chan int) var count = 1 var fileName = "./flag.txt" var file *os.File var err error func main() { path := strings.Join([]string{userName, ":", password, "@tcp(",ip, ":", port, ")/", dbName, "?charset=utf8"}, "") max := 2 file = openFile(fileName) for i := 0; i < max; i ++ { for j := 1; j <= 10; j ++ { DB, _ := sql.Open("mysql", path) //验证连接 if errConn := DB.Ping(); errConn != nil{ fmt.Println("open database fail") return } fmt.Println("connnect success") defer DB.Close() link := "http://tj.esf.fang.com/housing/__0_0_0_0_" + strconv.Itoa(i * 10 + j) + "_0_0_0/" go work(link, DB, i * 10 + j) } } for { <- flagCh if count < max * 10 { fmt.Println("<- receive the " + strconv.Itoa(count) + " thread ending flag") count++ } else { break } fmt.Println("All "+ strconv.Itoa(count) + " has done") } defer file.Close() } func openFile(fileName string) *os.File { if checkFileIsExist(fileName) { //如果文件存在 file, err = os.OpenFile(fileName, os.O_APPEND, 0666) } else { //创建文件 file, err = os.Create(fileName) } check(err) return file } func writeFile(file *os.File, content string) { writer := bufio.NewWriter(file) writer.WriteString(content) writer.WriteString("\r\n") writer.Flush() } func check(e error) { if e != nil { panic(e) } } func checkFileIsExist(filename string) bool { var exist = true if _, err := os.Stat(filename); os.IsNotExist(err) { exist = false } return exist } func work(url string, DB *sql.DB, page int) { c := colly.NewCollector() detailLink := c.Clone() detailController := c.Clone() infos := make([]XQinfo, 0) c.OnHTML(".plotListwrap > dt > a", func(e *colly.HTMLElement) { link := e.Attr("href") fmt.Printf("link : %s \t", link) fmt.Println() detailLink.Visit(link) }) detailLink.OnHTML("#kesfxqxq_A01_03_01", func(e *colly.HTMLElement) { link := e.ChildAttr("a", "href") //content := e.ChildText("a") //fmt.Printf("detial link : %s \t", link) //fmt.Printf("detial content : %s \t", coverGBKToUTF8(content)) //fmt.Println() detailController.Visit(link) }) detailController.OnHTML("body", func(e *colly.HTMLElement) { info := XQinfo{} // 小区名称 name := e.DOM.Find(".ceninfo_sq > h1 > a").Text() info.name = coverString(name) e.DOM.Find(".inforwrap").Each(func(i int, selection *goquery.Selection) { // 模块名称 modelName := coverString(selection.Prev().Find("h3").Text()) //fmt.Println("h3 -> ", modelName) switch modelName { case "基本信息": dealInfo(selection, &info) case "配套设施": dealInfo(selection, &info) case "周边信息": selection.Find("dl dt").Each(func(_ int, otherSelect *goquery.Selection) { tab := coverString(otherSelect.Text()) del := strings.Index(tab, "本段合作") if del == -1 { info.OtherInfo = info.OtherInfo + tab + "|" } }) } }) infos = append(infos, info) }) c.OnRequest(func(r *colly.Request) { fmt.Println("Visiting", r.URL.String()) }) c.OnScraped(func(response *colly.Response) { for _, info := range infos{ insertDB(DB, info) } fmt.Println("the "+ strconv.Itoa(page) + " thread sending end flag ->") flagCh <- 1 }) c.Visit(url) } // 处理小区基础信息 func dealInfo(selection *goquery.Selection, info *XQinfo) { selection.Find("dl dd").Each(func(_ int, selectionbase *goquery.Selection) { setXQinfo(selectionbase, info) }) selection.Find("dl dt").Each(func(_ int, selectionbase *goquery.Selection) { setXQinfo(selectionbase, info) }) } func setXQinfo(selectionbase *goquery.Selection, info *XQinfo) { orgKey := coverString(selectionbase.Find("strong").Text()) index := strings.Index(orgKey, ":") var key string if index > 0 { key = orgKey[:index] } else { key = orgKey } var value string var fullValue string value,ok := selectionbase.Attr("title") if ok { value = coverString(value) } else { fullValue = coverString(selectionbase.Text()) value = fullValue[strings.Index(fullValue, ":") + 3:] } switch key { case "小区地址": info.addr = value case "所属区域": info.area = value case "邮编": info.postCode = value case "产权描述": info.propertyRight = value case "物业类别": info.propertyType = value case "建筑年代": info.buildTime = value case "开 发 商": info.developer = value case "建筑结构": info.buildStruct = value case "建筑类型": info.buildType = value case "建筑面积": info.buildArea = value case "占地面积": info.floorSpace = value case "物业公司": info.manageCompany = value case "绿 化 率": info.greenRatio = value case "容 积 率": info.plotRatio = value case "物 业 费": info.propretyFee = value case "附加信息": info.AdditionalInfo = value case "供水": info.waterSupply = value case "供暖": info.heatSupply = value case "供电": info.elecSupply = value case "燃气": info.gas = value case "安全管理": info.security = value case "卫生服务": info.environment = value case "停 车 位": info.parkingSpace = value } } //src为要转换的字符串 func coverGBKToUTF8(src string) string { return mahonia.NewDecoder("gbk").ConvertString(src) } func replaceNullHtml(src string) string { return strings.Replace(src, "聽", "", -1) } func coverString(src string) string { return replaceNullHtml(coverGBKToUTF8(src)) } func insertDB(DB *sql.DB, info XQinfo) { t := reflect.TypeOf(info) v := reflect.ValueOf(info) sql1 := "insert into rx_xiaoqu(" sql2 := ") values (" sql3 := ")" for i := 0; i < t.NumField(); i++ { sql1 = sql1 + t.Field(i).Name sql2 = sql2 + "'" + v.Field(i).String() + "'" if i != t.NumField() - 1 { sql1 = sql1 + ", " sql2 = sql2 + ", " } //fmt.Printf("key -> %s, value -> %s", t.Field(i).Name, v.Field(i)) //fmt.Println() } //fmt.Println(sql1, sql2) stmt, err := DB.Prepare(sql1 + sql2 + sql3) if err != nil { fmt.Println(sql1 + sql2) fmt.Println(err) } res, err := stmt.Exec() if err != nil { fmt.Println(err) } fmt.Println(res.LastInsertId()) } ```

有疑问加站长微信联系(非本文作者))

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

2389 次点击  
加入收藏 微博
1 回复  |  直到 2019-05-04 21:23:34
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传