通过Golang实现无头浏览器截图

package main

import (
   "context"
   "io/ioutil"
   "log"
   "time"

   "github.com/chromedp/cdproto/page"
   "github.com/chromedp/chromedp"
)

func main() {
   // 创建一个上下文
   ctx, cancel := chromedp.NewContext(context.Background())
   defer cancel()

   // 设置浏览器选项
   opts := append(chromedp.DefaultExecAllocatorOptions[:],
      chromedp.Flag("headless", true),
      chromedp.Flag("disable-gpu", true),
      chromedp.Flag("no-sandbox", true),
      chromedp.Flag("disable-dev-shm-usage", true),
      chromedp.Flag("remote-debugging-port", "9222"),
   )
   allocCtx, cancel := chromedp.NewExecAllocator(ctx, opts...)
   defer cancel()

   // 创建一个浏览器实例
   ctx, cancel = chromedp.NewContext(allocCtx)
   defer cancel()

   // 导航到指定的URL
   var buf []byte
   err := chromedp.Run(ctx, chromedp.Navigate("https://www.baidu.com"), chromedp.Sleep(2*time.Second), chromedp.ActionFunc(func(ctx context.Context) error {
      // 获取页面截图
      var err error
      buf, err = page.CaptureScreenshot().WithQuality(90).WithClip(&page.Viewport{X: 0, Y: 0, Width: 1920, Height: 1080, Scale: 1}).Do(ctx)
      if err != nil {
         return err
      }
      return nil
   }))
   if err != nil {
      log.Fatal(err)
   }
   // 将截图保存到文件
   err = ioutil.WriteFile("screenshot.png", buf, 0644)
   if err != nil {
      log.Fatal(err)
   }
}

package main

import (
 "github.com/go-rod/rod"
 "github.com/go-rod/rod/lib/proto"
 "github.com/go-rod/rod/lib/utils"
 "github.com/ysmood/gson"
)

func main() {
 page := rod.New().MustConnect().MustPage("https://baidu.com").MustWaitLoad()

 //简单模式，默认截图设置
 page.MustScreenshot("my.png")

 //自定义截图设置
 img, _ := page.Screenshot(true, &proto.PageCaptureScreenshot{
  Format:  proto.PageCaptureScreenshotFormatJpeg,
  Quality: gson.Int(90),
  Clip: &proto.PageViewport{
   X:      0,
   Y:      0,
   Width:  300,
   Height: 200,
   Scale:  1,
  },
  FromSurface: true,
 })
 _ = utils.OutputFile("my.jpg", img)
}