pass context to generator

Author: nxdir-s

README.md

@@ -13,7 +13,7 @@ stream := pipelines.StreamSlice(ctx, data)
 Or from a generator function
 
 ```go
-func GenerateData() int { return rand.Intn(10) }
+func GenerateData(ctx context.Context) int { return rand.Intn(10) }
 
 stream := pipelines.GenerateStream(ctx, GenerateData)
 ```
@@ -57,7 +57,7 @@ const (
     MaxFan int = 3
 )
 
-func GenerateData() int {
+func GenerateData(ctx context.Context) int {
     return rand.Intn(5)
 }
 

pipelines.go

@@ -6,7 +6,7 @@ import (
 )
 
 // GenerateStream takes a function that generates data and returns a channel of type T
-func GenerateStream[T any](ctx context.Context, fn func() T) <-chan T {
+func GenerateStream[T any](ctx context.Context, fn func(context.Context) T) <-chan T {
 	stream := make(chan T)
 
 	go func() {
@@ -16,7 +16,7 @@ func GenerateStream[T any](ctx context.Context, fn func() T) <-chan T {
 			select {
 			case <-ctx.Done():
 				return
-			case stream <- fn():
+			case stream <- fn(ctx):
 			}
 		}
 	}()