Exclude context cancellation from error stats

Author: koenrh

candidates.go

@@ -44,9 +44,6 @@ func (p *Producer) ProduceWordList(ctx context.Context, names []string, list str
 	scanner := bufio.NewScanner(file)
 
 	for scanner.Scan() {
-		if ctx.Err() != nil {
-			return ctx.Err()
-		}
 		line := scanner.Text()
 		for _, n := range names {
 			if !p.Produce(ctx, n, line) {

resolver.go

@@ -116,6 +116,9 @@ func (s *DNSResolver) IsBucket(ctx context.Context, name string) bool {
 	records, err := s.resolveName(ctx, fmt.Sprintf("%s.%s", name, s3GlobalSuffix))
 
 	if err != nil {
+		if ctx.Err() != nil {
+			return false
+		}
 		atomic.AddUint64(&s.errors, 1)
 		s.classifyError(err)
 		return false