golang快速实现网页截图的方法_Golang

golang快速实现网页截图的方法

2021-04-01 00:51go语言中文网 Golang

这篇文章主要介绍了golang快速实现网页截图的方法,本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下

golang是近几年发展非常迅猛的一款服务器端的语言，其生态也日益丰富。对于使用golang实现网页截图这个需求，笔者在经过一番调研之后发现大家有推荐Selenium方案，但是这一方案问题较多:

需要安装Selenium或是PhantomJS
PhantomJS已经停止维护了
Selenium的运行需要安装chromedriver
chromedriver对chrome的版本有一定要求

还有推荐headless chrome方案的，但是这一方案也需要自行在服务器端安装一款浏览器，对服务器的压力比较大，并发性能也不好，所以最后笔者直接选用了一款第三方API的截图服务。

该服务有如下特点：

支持多线路
支持登录截图
支持UA变换
支持Cookie注入实现登录截图
支持自定义宽高
支持网页滚动截图
支持设定截图前的延迟时间

使用方式也很简单，我们打开它的首页 https://www.screenshotmaster.com/ 注册一个账号，然后你会在用户中心获取到一个唯一的Token，保存好这个Token 不要泄漏！

下面来看看它支持的参数：

golang快速实现网页截图的方法

您可以前往 API文档页面查看更多的参数。

使用golang调用截屏大师的接口获取截图：

									package main

									import (

									  "fmt"

									  "io"

									  "io/ioutil"

									  "net/http"

									  url2 "net/url"

									  "os"

									)

									func main() {

									  // 参数

									  token := "YOUR_API_TOKEN"

									  url := url2.QueryEscape("https://www.baidu.com")

									  width := 1280

									  height := 800

									  full_page := 1

									  // 构造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)

									  // 调用API

									  resp, err := http.Get(query)

									  if err != nil {

									    panic(err)

									  }

									  defer resp.Body.Close()

									  // 检查是否调用成功

									  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)

									  }

									}