diff --git a/wrp.go b/wrp.go index 3b94c4e..f8199b6 100644 --- a/wrp.go +++ b/wrp.go @@ -181,9 +181,16 @@ func capture(gourl string, w int64, h int64, s float64, co int, p int64, i bool, chromedp.Nodes("a", &nodes, chromedp.ByQueryAll)) if err != nil { - log.Printf("%s %s", c, err) - fmt.Fprintf(out, "
%s
", err) - return + if err.Error() == "context canceled" { + log.Printf("%s Contex cancelled, try again", c) + fmt.Fprintf(out, "
%s
-- restarting, try again", err) + ctx, cancel = chromedp.NewContext(context.Background()) + return + } else { + log.Printf("%s %s", c, err) + fmt.Fprintf(out, "
%s
", err) + return + } } log.Printf("%s Landed on: %s, Nodes: %d\n", c, loc, len(nodes))