首頁 > 後端開發 > Golang > 主體

golang 不定向爬蟲

PHPz
發布: 2023-05-10 10:06:06
原創
478 人瀏覽過

一、前言

隨著網路的發展,網路爬蟲的應用範圍越來越廣。在日常生活中,我們可以透過網路爬蟲獲取各種信息,如新聞、股票、天氣、電影、音樂等。特別是在大數據分析和人工智慧領域,網路爬蟲更是扮演著重要的角色。本文主要說明如何使用golang語言編寫一個不定向(即沒有特定目標網站)的爬蟲,以獲取網路上的資訊。

二、golang簡介

golang是Google開發的程式語言,由於其並發性、高效能、簡潔易學等特點,越來越受到程式設計師的青睞。本文所使用的golang版本是1.14.2。

三、實現想法

本次爬蟲主要分為以下幾步:

  1. #取得起始網址

可以透過手動輸入網址、從檔案讀取網址、從資料庫讀取網址等方式取得起始網址。

  1. 發送http請求

透過Get或Post等方式發送http請求,取得回應資料。

  1. 解析回應資料

根據回應資料的格式,使用正規表示式或第三方程式庫解析資料。

  1. 儲存資料

可以將資料儲存到檔案中、儲存到資料庫中,或使用其他儲存方式,具體根據需求決定。

  1. 解析新的網址

根據回應資料中的超連結等信息,解析新的網址,作為下一個需要爬取的網址。

  1. 重複上述步驟

根據新的網址,再次發送http請求,解析回應數據,儲存數據,解析新的網址,重複進行,直到沒有新的網址為止。

四、程式碼實作

在golang中,使用net/http套件來傳送http請求,使用regexp套件或第三方函式庫來解析回應數據,而本文所使用的是goquery函式庫。

  1. 初始化函數

首先,我們需要定義一個初始函數,負責取得起始網址、設定http客戶端等作業。

func init() {
    // 获取起始网址
    flag.StringVar(&startUrl, "url", "", "请输入起始网址")
    flag.Parse()

    // 设置http客户端
    client = &http.Client{
        Timeout: 30 * time.Second,
        CheckRedirect: func(req *http.Request, via []*http.Request) error {
            return http.ErrUseLastResponse
        },
    }
}
登入後複製
  1. 發送http請求函數

定義一個函數,負責傳送http請求,取得回應資料。

func GetHtml(url string) (string, error) {
    resp, err := client.Get(url)
    if err != nil {
        log.Println(err)
        return "", err
    }
    defer resp.Body.Close()

    body, err := ioutil.ReadAll(resp.Body)
    if err != nil {
        log.Println(err)
        return "", err
    }

    return string(body), nil
}
登入後複製
  1. 解析回應資料函數

使用goquery函式庫解析回應數據,具體實作方式如下:

func ParseSingleHTML(html string, query string) []string {
    doc, err := goquery.NewDocumentFromReader(strings.NewReader(html))
    if err != nil {
        log.Println(err)
        return nil
    }

    result := make([]string, 0)
    doc.Find(query).Each(func(i int, selection *goquery.Selection) {
        href, ok := selection.Attr("href")
        if ok {
            result = append(result, href)
        }
    })

    return result
}
登入後複製
  1. 儲存資料函數

定義一個函數,負責將資料儲存到檔案中。

func SaveData(data []string) error {
    file, err := os.OpenFile("data.txt", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
    if err != nil {
        log.Println(err)
        return err
    }
    defer file.Close()

    writer := bufio.NewWriter(file)
    for _, line := range data {
        _, err := writer.WriteString(line + "
")
        if err != nil {
            log.Println(err)
            return err
        }
    }
    writer.Flush()

    return nil
}
登入後複製
  1. 解析新的網址函數

使用正規表示式解析超連結中的新網址。

func ParseHref(url, html string) []string {
    re := regexp.MustCompile(`<a[sS]+?href="(.*?)"[sS]*?>`)
    matches := re.FindAllStringSubmatch(html, -1)

    result := make([]string, 0)
    for _, match := range matches {
        href := match[1]
        if strings.HasPrefix(href, "//") {
            href = "http:" + href
        } else if strings.HasPrefix(href, "/") {
            href = strings.TrimSuffix(url, "/") + href
        } else if strings.HasPrefix(href, "http://") || strings.HasPrefix(href, "https://") {
            // do nothing
        } else {
            href = url + "/" + href
        }
        result = append(result, href)
    }

    return result
}
登入後複製
  1. 主函數

最後,我們需要定義一個主函數,實作整個爬蟲的流程。

func main() {
    // 确认起始网址是否为空
    if startUrl == "" {
        fmt.Println("请指定起始网址")
        return
    }

    // 初始化待访问队列
    queue := list.New()
    queue.PushBack(startUrl)

    // 初始化已访问集合
    visited := make(map[string]bool)

    // 循环爬取
    for queue.Len() > 0 {
        // 从队列中弹出一个网址
        elem := queue.Front()
        queue.Remove(elem)
        url, ok := elem.Value.(string)
        if !ok {
            log.Println("网址格式错误")
            continue
        }

        // 确认该网址是否已经访问过
        if visited[url] {
            continue
        }
        visited[url] = true

        // 发送http请求,获取响应数据
        html, err := GetHtml(url)
        if err != nil {
            continue
        }

        // 解析响应数据,获取新的网址
        hrefs := ParseHref(url, html)
        queue.PushBackList(list.New().Init())
        for _, href := range hrefs {
            if !visited[href] {
                hrefHtml, err := GetHtml(href)
                if err != nil {
                    continue
                }
                hrefUrls := ParseSingleHTML(hrefHtml, "a")

                // 将新的网址加入队列
                queue.PushBackList(list.New().Init())
                for _, hrefUrl := range hrefUrls {
                    queue.PushBack(hrefUrl)
                }
            }
        }

        // 存储数据到文件
        data := ParseSingleHTML(html, "title")
        err = SaveData(data)
        if err != nil {
            continue
        }
    }
}
登入後複製

五、總結

以上是使用golang編寫不定向爬蟲的基本流程和實作方式。當然,這只是一個簡單的範例,實際開發中還需要考慮反爬蟲策略、線程安全等問題。希望能夠對讀者有幫助。

以上是golang 不定向爬蟲的詳細內容。更多資訊請關注PHP中文網其他相關文章!

來源:php.cn
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板