Handle context cancel on race method

Author: vardius

README.md

@@ -34,11 +34,11 @@ HOW TO USE
 goos: darwin
 goarch: amd64
 pkg: github.com/vardius/gollback
-BenchmarkRace-4         10000000               219 ns/op               0 B/op          0 allocs/op
-BenchmarkAll-4           5000000               281 ns/op              40 B/op          1 allocs/op
-BenchmarkRetry-4         2000000               579 ns/op             208 B/op          3 allocs/op
+BenchmarkRace-4    	 5000000	       223 ns/op	       0 B/op	       0 allocs/op
+BenchmarkAll-4     	 5000000	       281 ns/op	      40 B/op	       1 allocs/op
+BenchmarkRetry-4   	200000000	         6.68 ns/op	       0 B/op	       0 allocs/op
 PASS
-ok      github.com/vardius/gollback     23.742s
+ok  	github.com/vardius/gollback	25.130s
 ```
 
 ## Race

gollback.go

@@ -2,9 +2,7 @@ package gollback
 
 import (
 	"context"
-	"errors"
 	"sync"
-	"time"
 )
 
 // AsyncFunc represents asynchronous function
@@ -41,15 +39,18 @@ func (p *gollback) Race(fns ...AsyncFunc) (interface{}, error) {
 			for {
 				select {
 				case <-ctx.Done():
+					if index == len(fns)-1 {
+						cancel()
+						out <- &response{
+							err: ctx.Err(),
+						}
+					}
+
 					return
 				default:
 					var r response
 					r.res, r.err = f(ctx)
 
-					if ctx.Err() != nil {
-						return
-					}
-
 					if r.err == nil || index == len(fns)-1 {
 						cancel()
 						out <- &r
@@ -96,8 +97,6 @@ func (p *gollback) Retry(retires int, fn AsyncFunc) (interface{}, error) {
 
 	for {
 		select {
-		case <-time.After(1 * time.Second):
-			return nil, errors.New("timeout")
 		case <-p.ctx.Done():
 			return nil, p.ctx.Err()
 		default:
@@ -113,13 +112,6 @@ func (p *gollback) Retry(retires int, fn AsyncFunc) (interface{}, error) {
 	}
 }
 
-func (p *gollback) RetryWithTimeout(retires int, duration time.Duration, fn AsyncFunc) (interface{}, error) {
-	ctx, cancel := context.WithTimeout(p.ctx, duration)
-	defer cancel()
-
-	return fn(ctx)
-}
-
 // New creates new gollback
 func New(ctx context.Context) Gollback {
 	if ctx == nil {