Use shardBatch instead of records for failedRecords (#248)

etspaceman · Nov 9, 2023 · 24f67da · 24f67da
1 parent 69060e0
commit 24f67da
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/shared/src/main/scala/kinesis4cats/producer/Producer.scala b/shared/src/main/scala/kinesis4cats/producer/Producer.scala
@@ -142,7 +142,7 @@ abstract class Producer[F[_], PutReq, PutRes] private[kinesis4cats] (
               .parTraverseN(config.shardParallelism) { shardBatch =>
                 putImpl(asPutRequest(shardBatch))
                   .map(resp =>
-                    failedRecords(records, resp)
+                    failedRecords(shardBatch, resp)
                       .map(Producer.Result.putFailures[PutRes])
                       .fold(
                         Producer.Result.success(resp)