Merge pull request #67 from TYashodaKrishna/maxBytes

Add kafka producer option top set max record size
unbxd · Nov 30, 2023 · d468c4a · d468c4a
2 parents fe16520 + 7829666
commit d468c4a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/transport/kafka/producer.go b/transport/kafka/producer.go
@@ -50,6 +50,12 @@ func WithMaxAttemptsProducerOption(attempts int) ProducerOption {
 	return func(p *Producer) { p.config.MaxAttempts = attempts }
 }
 
+// WithMaxBatchBytesOption sets the maximum bytes of record size
+// kafka producer will try to produce
+func WithMaxBatchBytesOption(batchBytes int64) ProducerOption {
+	return func(p *Producer) { p.config.BatchBytes = int(batchBytes) }
+}
+
 // WithQueueCapacityProducerOption sets the internal buffer capacity
 // used to cache incoming messages before publishing on kafka
 func WithQueueCapacityProducerOption(qc int) ProducerOption {