golang快速實現(xiàn)網(wǎng)頁截圖的方法

更新時間：2021年03月04日 09:21:24 投稿：mrr

這篇文章主要介紹了golang快速實現(xiàn)網(wǎng)頁截圖的方法,本文通過實例代碼給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下

golang是近幾年發(fā)展非常迅猛的一款服務器端的語言，其生態(tài)也日益豐富。對于使用golang實現(xiàn)網(wǎng)頁截圖這個需求，筆者在經(jīng)過一番調(diào)研之后發(fā)現(xiàn)大家有推薦Selenium方案，但是這一方案問題較多:

需要安裝Selenium或是PhantomJS
PhantomJS已經(jīng)停止維護了
Selenium的運行需要安裝chromedriver
chromedriver對chrome的版本有一定要求

還有推薦headless chrome方案的，但是這一方案也需要自行在服務器端安裝一款瀏覽器，對服務器的壓力比較大，并發(fā)性能也不好，所以最后筆者直接選用了一款第三方API的截圖服務。

該服務有如下特點：

支持多線路
支持登錄截圖
支持UA變換
支持Cookie注入實現(xiàn)登錄截圖
支持自定義寬高
支持網(wǎng)頁滾動截圖
支持設定截圖前的延遲時間

使用方式也很簡單，我們打開它的首頁 https://www.screenshotmaster.com/ 注冊一個賬號，然后你會在用戶中心獲取到一個唯一的Token，保存好這個Token 不要泄漏！

下面來看看它支持的參數(shù)：

您可以前往 API文檔頁面查看更多的參數(shù)。

使用golang調(diào)用截屏大師的接口獲取截圖：

package main

import (
  "fmt"
  "io"
  "io/ioutil"
  "net/http"
  url2 "net/url"
  "os"
)

func main() {
  // 參數(shù)
  token := "YOUR_API_TOKEN"
  url := url2.QueryEscape("https://www.baidu.com")
  width := 1280
  height := 800
  full_page := 1

  // 構(gòu)造URL
  query := "https://www.screenshotmaster.com/api/v1/screenshot"
  query += fmt.Sprintf("?token=%s&url=%s&width=%d&height=%d&full_page=%s",
    token, url, width, height, full_page)

  // 調(diào)用API
  resp, err := http.Get(query)
  if err != nil {
    panic(err)
  }
  defer resp.Body.Close()

  // 檢查是否調(diào)用成功
  if resp.StatusCode != 200 {
    errorBody, _ := ioutil.ReadAll(resp.Body)
    panic(fmt.Errorf("error while calling api %s", errorBody))
  }

  // 保存截圖
  file, err := os.Create("./screenshot.png")
  if err != nil {
    panic(err)
  }
  defer file.Close()

  _, err = io.Copy(file, resp.Body)
  if err != nil {
    panic(err)
  }
}

到此這篇關于golang快速實現(xiàn)網(wǎng)頁截圖的文章就介紹到這了,更多相關go實現(xiàn)網(wǎng)頁截圖內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: