go写爬虫

朴灿烈づ我的快乐病毒、 2023-06-05 08:06 13阅读 0赞

package main
    
    import (
    	"fmt"
    	"net/http"
    	"os"
    	"regexp"
    	"strconv"
    	"strings"
    )
    
    func main() { 
    	var start, end int
    	fmt.Printf("请输入起始页( >= 1) :")
    	fmt.Scan(&start)
    	fmt.Printf("请输入终止页( >= 起始页) :")
    	fmt.Scan(&end)
    
    	DoWork(start, end) //工作函数
    }
    
    //开启协程
    func DoWork(start, end int) { 
    	fmt.Printf("准备爬取第%d页到%d页的网址\n", start, end)
    
    	page := make(chan int)
    
    	for i := start; i <= end; i++ { 
    		//开协程爬
    		go SpiderPape(i, page)
    	}
    
    	for i := start; i <= end; i++ { 
    		fmt.Printf("第%d个页面爬取完成\n", <-page)
    	}
    }
    
    //爬取页面
    func SpiderPape(i int, page chan int) { 
    
    	//这里需要分析页面的规律
    	url := "https://m.pengfu.com/xiaohua_" + strconv.Itoa(i) + ".html"
    	fmt.Printf("正在爬取第%d个网页：%s\n", i, url)
    
    	//开始爬取页面内容
    	result, err := HttpGet(url)
    	if err != nil { 
    		fmt.Println("HttpGet err = ", err)
    		return
    	}
    
    	//这里需要分析内容页面url的规律
    	re := regexp.MustCompile(`<h1 class="f18"><a href="(?s:(.*?))"`)
    	if re == nil { 
    		fmt.Println("regexp.MustCompile err")
    		return
    	}
    
    	//取内容的url
    	joyUrls := re.FindAllStringSubmatch(result, -1)
    
    	fileTitle := make([]string, 0)
    	fileContent := make([]string, 0)
    
    	//根据内容url爬取标题和内容
    	//第一个返回下标，第二个返回内容
    	for _, data := range joyUrls { 
    
    		//开始爬取每一个笑话，每一个段子
    		title, content, err := SpiderOneJoy(data[1])
    		if err != nil { 
    			fmt.Println("SpiderOneJoy err = ", err)
    			continue
    		}
    
    		fileTitle = append(fileTitle, title)       //追加内容
    		fileContent = append(fileContent, content) //追加内容
    	}
    
    	//把内容写入到文件
    	fmt.Println("title = ", fileTitle)
    	fmt.Println("content = ", fileContent)
    	StoreJoyToFile(i, fileTitle, fileContent)
    
    	page <- i //写内容，写num
    
    }
    
    //通过get请求获取网页内容
    func HttpGet(url string) (result string, err error) { 
    	resp, err1 := http.Get(url) //发送get请求
    	if err1 != nil { 
    		err = err1
    		return
    	}
    
    	defer resp.Body.Close()
    
    	//读取网页内容
    	buf := make([]byte, 4*1024)
    	for { 
    		n, _ := resp.Body.Read(buf)
    		if n == 0 { 
    			break
    		}
    		result += string(buf[:n]) //累加读取的内容
    	}
    	return
    }
    
    //开始爬取每一个笑话，每一个段子 title, content, err := SpiderOneJoy(url)
    func SpiderOneJoy(url string) (title, content string, err error) { 
    
    	//开始爬取页面内容
    	result, err1 := HttpGet(url)
    	if err1 != nil { 
    		err = err1
    		return
    	}
    
    	//这里需要分析标题的规律
    	re1 := regexp.MustCompile(`<title>(?s:(.*?))</title>`)
    	if re1 == nil { 
    		//fmt.Println("regexp.MustCompile err")
    		err = fmt.Errorf("%s", "regexp.MustCompile err")
    		return
    	}
    	//取内容
    	tmpTitle := re1.FindAllStringSubmatch(result, 1) //最后一个参数为1，只过滤第一个
    	for _, data := range tmpTitle { 
    		title = data[1]
    		title = strings.Replace(title, "\t", "", -1)
    		break
    	}
    
    	//这里需要分析内容的规律
    	re2 := regexp.MustCompile(`<div class="con-txt">(?s:(.*?))</div`)
    	if re2 == nil { 
    		err = fmt.Errorf("%s", "regexp.MustCompile err2")
    		return
    	}
    
    	//取内容
    	tmpContent := re2.FindAllStringSubmatch(result, -1)
    	for _, data := range tmpContent { 
    		content = data[1]
    		content = strings.Replace(content, "\t", "", -1)
    		content = strings.Replace(content, "\n", "", -1)
    		content = strings.Replace(content, "\r", "", -1)
    		content = strings.Replace(content, "<br />", "", -1)
    		break
    	}
    
    	return
    }
    
    //把内容写入到文件
    func StoreJoyToFile(i int, fileTitle, fileContent []string) { 
    	//新建文件
    	f, err := os.Create(strconv.Itoa(i) + ".txt")
    	if err != nil { 
    		fmt.Println("os.Create err = ", err)
    		return
    	}
    
    	defer f.Close()
    
    	//写内容
    	n := len(fileTitle)
    	for i := 0; i < n; i++ { 
    		//写标题
    		f.WriteString(fileTitle[i] + "\n")
    		//写内容
    		f.WriteString(fileContent[i] + "\n")
    
    		f.WriteString("\n=================================================================\n")
    	}
    
    }