wip: debug

pkg/utils/batch.go

@@ -2,12 +2,14 @@ package utils
 
 import (
 	"context"
+	"fmt"
 )
 
 // Batch reads from a channel and calls fn with a slice of batchSize.
 func Batch[T any](ctx context.Context, ch <-chan T, batchSize int, fn func([]T)) {
 	if batchSize <= 1 { // sanity check,
 		for v := range ch {
+			fmt.Println("<< 1")
 			fn([]T{v})
 		}
 		return
@@ -19,22 +21,26 @@ func Batch[T any](ctx context.Context, ch <-chan T, batchSize int, fn func([]T))
 		select {
 		case <-ctx.Done():
 			if len(batch) > 0 {
+				fmt.Println("<< 2")
 				fn(batch)
 			}
 			return
 		case v, ok := <-ch:
 			if !ok { // closed
+				fmt.Println("<< 3")
 				fn(batch)
 				return
 			}
 
 			batch = append(batch, v)
 			if len(batch) == batchSize { // full
+				fmt.Println("<< 4")
 				fn(batch)
 				batch = make([]T, 0, batchSize) // reset
 			}
 		default:
 			if len(batch) > 0 { // partial
+				fmt.Println("<< 5")
 				fn(batch)
 				batch = make([]T, 0, batchSize) // reset
 			} else { // empty