Skip to content

Commit ce58b3c

Browse files
authored
Update Redeliverer.scala
1 parent d61fa44 commit ce58b3c

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

core/src/main/scala/com.softwaremill.kmq/redelivery/Redeliverer.scala

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,15 +21,20 @@ trait Redeliverer {
2121

2222
class DefaultRedeliverer(
2323
partition: Partition, producer: KafkaProducer[Array[Byte], Array[Byte]],
24-
config: KmqConfig, clients: KafkaClients)
24+
config: KmqConfig, clients: KafkaClients, extraConfig: Option[java.util.Map[String, Object]] = None)
2525
extends Redeliverer with StrictLogging {
2626

2727
private val SendTimeoutSeconds = 60L
2828

2929
private val tp = new TopicPartition(config.getMsgTopic, partition)
3030

3131
private val reader = {
32-
val c = clients.createConsumer(null, classOf[ByteArrayDeserializer], classOf[ByteArrayDeserializer])
32+
val c = extraConfig match {
33+
// extraConfig is not empty
34+
case Some(cfg) => clients.createConsumer(null, classOf[ByteArrayDeserializer], classOf[ByteArrayDeserializer], cfg)
35+
// extraConfig is empty
36+
case None => clients.createConsumer(null, classOf[ByteArrayDeserializer], classOf[ByteArrayDeserializer])
37+
}
3338
c.assign(Collections.singleton(tp))
3439
new SingleOffsetReader(tp, c)
3540
}
@@ -133,4 +138,4 @@ private class SingleOffsetReader(tp: TopicPartition, consumer: KafkaConsumer[Arr
133138
def close(): Unit = {
134139
consumer.close()
135140
}
136-
}
141+
}

0 commit comments

Comments
 (0)