change cancel place (#82)

harlow · Feb 18, 2019 · 245d1bd · 245d1bd
1 parent 2037463
commit 245d1bd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/consumer.go b/consumer.go
@@ -135,15 +135,15 @@ func (c *Consumer) Scan(ctx context.Context, fn func(*Record) ScanStatus) error
 			defer wg.Done()
 
 			if err := c.ScanShard(ctx, shardID, fn); err != nil {
+				cancel()
+
 				select {
 				case errc <- fmt.Errorf("shard %s error: %v", shardID, err):
 					// first error to occur
 				default:
 					// error has already occured
 				}
 			}
-
-			cancel()
 		}(shardID)
 	}