Added context to fetch method

README.md

@@ -24,7 +24,7 @@ func NewUserLoader() *UserLoader {
 	return &UserLoader{
 		wait:     2 * time.Millisecond,
 		maxBatch: 100,
-		fetch: func(keys []string) ([]*User, []error) {
+		fetch: func(ctx context.Context, keys []string) ([]*User, []error) {
 			users := make([]*User, len(keys))
 			errors := make([]error, len(keys))
 
@@ -41,7 +41,7 @@ Then wherever you want to call the dataloader
 ```go
 loader := NewUserLoader()
 
-user, err := loader.Load("123")
+user, err := loader.Load(context.Background(), "123")
 ```
 
 This method will block for a short amount of time, waiting for any other similar requests to come in, call your fetch
@@ -87,7 +87,7 @@ func NewLoader(ctx context.Context) *UserLoader {
 	return &UserLoader{
 		wait:     2 * time.Millisecond,
 		maxBatch: 100,
-		fetch: func(keys []string) ([]*User, []error) {
+		fetch: func(ctx context.Context, keys []string) ([]*User, []error) {
 			// you now have a ctx to work with
 		},
 	}

example/benchmark_test.go

@@ -1,6 +1,7 @@
 package example
 
 import (
+	"context"
 	"fmt"
 	"math/rand"
 	"strconv"
@@ -13,7 +14,7 @@ func BenchmarkLoader(b *testing.B) {
 	dl := &UserLoader{
 		wait:     500 * time.Nanosecond,
 		maxBatch: 100,
-		fetch: func(keys []string) ([]*User, []error) {
+		fetch: func(fCtx context.Context, keys []string) ([]*User, []error) {
 			users := make([]*User, len(keys))
 			errors := make([]error, len(keys))
 
@@ -33,7 +34,7 @@ func BenchmarkLoader(b *testing.B) {
 	b.Run("caches", func(b *testing.B) {
 		thunks := make([]func() (*User, error), b.N)
 		for i := 0; i < b.N; i++ {
-			thunks[i] = dl.LoadThunk(strconv.Itoa(rand.Int() % 300))
+			thunks[i] = dl.LoadThunk(context.Background(), strconv.Itoa(rand.Int()%300))
 		}
 
 		for i := 0; i < b.N; i++ {
@@ -44,7 +45,7 @@ func BenchmarkLoader(b *testing.B) {
 	b.Run("random spread", func(b *testing.B) {
 		thunks := make([]func() (*User, error), b.N)
 		for i := 0; i < b.N; i++ {
-			thunks[i] = dl.LoadThunk(strconv.Itoa(rand.Int()))
+			thunks[i] = dl.LoadThunk(context.Background(), strconv.Itoa(rand.Int()))
 		}
 
 		for i := 0; i < b.N; i++ {
@@ -58,7 +59,7 @@ func BenchmarkLoader(b *testing.B) {
 			wg.Add(1)
 			go func() {
 				for j := 0; j < b.N; j++ {
-					dl.Load(strconv.Itoa(rand.Int()))
+					dl.Load(context.Background(), strconv.Itoa(rand.Int()))
 				}
 				wg.Done()
 			}()

example/slice/user.go

@@ -3,6 +3,7 @@
 package slice
 
 import (
+	"context"
 	"strconv"
 	"time"
 
@@ -13,7 +14,7 @@ func NewLoader() *UserSliceLoader {
 	return &UserSliceLoader{
 		wait:     2 * time.Millisecond,
 		maxBatch: 100,
-		fetch: func(keys []int) ([][]example.User, []error) {
+		fetch: func(ctx context.Context, keys []int) ([][]example.User, []error) {
 			users := make([][]example.User, len(keys))
 			errors := make([]error, len(keys))