move chromedp.Sleep to just before screenshot to fix render issue

This commit is contained in:
Antoni Sawicki 2020-10-27 04:05:17 -07:00
parent 5dc4699ac9
commit b54ebbf9e5
1 changed files with 4 additions and 2 deletions

6
wrp.go
View File

@ -314,7 +314,6 @@ func (w wrpReq) capture() {
var pngcap []byte
chromedp.Run(ctx,
emulation.SetDeviceMetricsOverride(int64(float64(w.width)/w.scale), 10, w.scale, false),
chromedp.Sleep(time.Second*2),
chromedp.Location(&w.url),
chromedp.ComputedStyle("body", &styles, chromedp.ByQuery),
chromedp.ActionFunc(func(ctx context.Context) error {
@ -338,7 +337,10 @@ func (w wrpReq) capture() {
}
chromedp.Run(ctx, emulation.SetDeviceMetricsOverride(int64(float64(w.width)/w.scale), height, w.scale, false))
// Capture screenshot...
err = chromedp.Run(ctx, chromedp.CaptureScreenshot(&pngcap))
err = chromedp.Run(ctx,
chromedp.Sleep(time.Second*2),
chromedp.CaptureScreenshot(&pngcap),
)
if err != nil {
if err.Error() == "context canceled" {
log.Printf("%s Contex cancelled, try again", w.req.RemoteAddr)