golang快速实现网页截图

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

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

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

该服务有如下特点:

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

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

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

image.png

您可以前往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)
 }
}

本文来自:Segmentfault

感谢作者:xx19941215

查看原文:golang快速实现网页截图

<a href="/user/xx19941215" title="@xx19941215">@xx19941215</a> 第一种方案 实测OK,做成单独的服务对外提供API

评论于 2021年03月03日 09:09:35

截下来的图不能看呀

评论于 2021年03月02日 21:31:59