1. 网页截图

在这篇文章中，我们将研究如何利用Chrome的调试协议来加载网页并截图。通过一个名为的程序包chromedp，一切都可以实现，该程序包使我们可以通过Go代码控制Chrome实例。您还需要安装Chrome或使用类似于chrome/headless-shellDocker映像的工具。

我们将代码中的过程分为：

  • 启动Chrome
  • 运行任务：例如加载网页并截图
  • 将屏幕截图保存到文件 
  1. package main
  3. import (
  4.     "context"
  5.     "io/ioutil"
  6.     "log"
  8.     "github.com/chromedp/cdproto/page"
  9.     "github.com/chromedp/chromedp"
  10. )
  12. func main() {
  14.     // Start Chrome
  15.     // Remove the 2nd param if you don't need debug information logged
  16.     ctx, cancel := chromedp.NewContext(context.Background(), chromedp.WithDebugf(log.Printf))
  17.     defer cancel()
  19.     url := "https://golangcode.com/"
  20.     filename := "golangcode.png"
  22.     // Run Tasks
  23.     // List of actions to run in sequence (which also fills our image buffer)
  24.     var imageBuf []byte
  25.     if err := chromedp.Run(ctx, ScreenshotTasks(url, &imageBuf)); err != nil {
  26.         log.Fatal(err)
  27.     }
  29.     // Write our image to file
  30.     if err := ioutil.WriteFile(filename, imageBuf, 0644); err != nil {
  31.         log.Fatal(err)
  32.     }
  33. }
  35. func ScreenshotTasks(url string, imageBuf *[]byte) chromedp.Tasks {
  36.     return chromedp.Tasks{
  37.         chromedp.Navigate(url),
  38.         chromedp.ActionFunc(func(ctx context.Context) (err error) {
  39.             *imageBuf, err = page.CaptureScreenshot().WithQuality(90).Do(ctx)
  40.             return err
  41.         }),
  42.     }
  43. }

topgoer

另外，如果您想将页面另存为pdf而不是图像，则可以将CaptureScreenshot行替换为以下内容：

  1.     *imageBuf, _, err = page.PrintToPDF().WithPrintBackground(false).Do(ctx)