add context cancelation

2020-03-11 16:52:07 +11:00
parent a49cf9470a
commit da80521ed1
1 changed files with 5 additions and 0 deletions
--- a/search.go
+++ b/search.go
@@ -102,6 +102,11 @@ func Search(ctx context.Context, searchTerm string, opts ...SearchOptions) ([]Re
 		sel := e.DOM

 		for i := range sel.Nodes {
+			if err := ctx.Err(); err != nil {
+				rErr = err
+				return
+			}
+
 			item := sel.Eq(i)

 			rDiv := item.Find("div.r")