用go语言完成爬虫

2021-05-06 16:27:22 浏览数 (1)

除了python是一个爬虫的好工具,go也可以完成同样的操作 如下,爬取的百度贴吧的代码:

代码语言:javascript复制
package main

import (
	"fmt"
	"io"
	"net/http"
	"os"
	"strconv"
)

func main() {
	// 指定爬取起始,终止页
	var start, end int
	fmt.Print("请输入爬取的起始页:")
	fmt.Scan(&start)
	fmt.Println("请输入终止页")
	fmt.Scan(&end)

	working(start, end)
}

func working(start int, end int) {
	fmt.Printf("正在爬取第%d页到%d页", start, end)
	// 循环读取页数
	for i := start; i <= end; i   {
		url := "https://tieba.baidu.com/f?kw=绝地求生&ie=utf-8&pn="   strconv.Itoa((i-1)*50)
		result, err := HttpGet(url)
		if err != nil {
			fmt.Println("httpGet err", err)
			continue
		}
		//fmt.Println("result=", result)
		f, err := os.Create("第"   strconv.Itoa(i)   "页"   ".html")
		if err != nil {
			fmt.Println("Create err", err)
			continue
		}
		f.WriteString(result)
		f.Close() // 保存好一个文件,关闭一个文件
	}

}

func HttpGet(url string) (result string, err error) {
	resp, err1 := http.Get(url)
	if err1 != nil {
		err = err1 // 将封装函数的内部的错误抛给调用者
		return
	}
	defer resp.Body.Close()
	// 循环读取,传出给调用者
	buf := make([]byte, 4096)
	for {
		n, err2 := resp.Body.Read(buf)
		if n == 0 {
			fmt.Println("读取网页完成")
			break
		}
		if err2 != nil && err2 != io.EOF {
			err = err2
			return
		}
		// 累加每一次循环读到的buf数据,存入result一次性读取
		result  = string(buf[:n])
	}
	return

}

可以看见实现完后,项目根目录下有了html文件:

0 人点赞