Inline kafkalib.batched (#519)

artie-labs · Oct 18, 2024 · 106c9d9 · 106c9d9
1 parent caad3d1
commit 106c9d9
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 44 deletions.
diff --git a/lib/kafkalib/batch.go b/lib/kafkalib/batch.go
diff --git a/lib/kafkalib/batch_test.go b/lib/kafkalib/batch_test.go
diff --git a/lib/kafkalib/writer.go b/lib/kafkalib/writer.go
@@ -5,6 +5,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"log/slog"
+	"slices"
 	"time"
 
 	"github.com/artie-labs/transfer/lib/jitter"
@@ -99,7 +100,7 @@ func (b *BatchWriter) Write(ctx context.Context, rawMsgs []lib.RawMessage) error
 		msgs = append(msgs, kafkaMsg)
 	}
 
-	for _, batch := range batched(msgs, int(b.cfg.GetPublishSize())) {
+	for batch := range slices.Chunk(msgs, int(b.cfg.GetPublishSize())) {
 		tags := map[string]string{
 			"what": "error",
 		}