mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-12-22 10:11:52 +01:00
optimize code for initializing Kafka consumers and processors
Signed-off-by: Rajiv Ranjan Singh <rajivperfect007@gmail.com>
This commit is contained in:
parent
ec83d01dca
commit
832a5de758
@ -85,16 +85,13 @@ trait UnifiedGraphWriter[
|
||||
}
|
||||
}
|
||||
|
||||
private def initRecosHoseKafka(
|
||||
queue: java.util.Queue[Array[RecosHoseMessage]],
|
||||
queuelimit: Semaphore,
|
||||
): Unit = {
|
||||
private def initRecosHoseKafka(queue: java.util.Queue[Array[RecosHoseMessage]], queuelimit: Semaphore): Unit = {
|
||||
try {
|
||||
consumers = (0 until consumerNum).map { index =>
|
||||
new ThreadSafeKafkaConsumerClient(
|
||||
kafkaConsumerBuilder.clientId(s"clientId-$index").enableAutoCommit(false).config)
|
||||
val consumers = (0 until consumerNum).map { index =>
|
||||
new ThreadSafeKafkaConsumerClient(kafkaConsumerBuilder.clientId(s"clientId-$index").enableAutoCommit(false).config)
|
||||
}
|
||||
processors = consumers.zipWithIndex.map {
|
||||
|
||||
val processors = consumers.zipWithIndex.map {
|
||||
case (consumer, index) =>
|
||||
val bufferedWriter = BufferedEdgeCollector(bufferSize, queue, queuelimit, statsReceiver)
|
||||
val processor = RecosEdgeProcessor(bufferedWriter)(statsReceiver)
|
||||
@ -112,23 +109,16 @@ trait UnifiedGraphWriter[
|
||||
}
|
||||
|
||||
log.info(s"starting ${processors.size} recosKafka processors")
|
||||
processors.foreach { processor =>
|
||||
processor.start()
|
||||
}
|
||||
processors.foreach(_.start())
|
||||
|
||||
} catch {
|
||||
case e: Throwable =>
|
||||
e.printStackTrace()
|
||||
log.error(e, e.toString)
|
||||
processors.foreach { processor =>
|
||||
processor.close()
|
||||
}
|
||||
processors = Seq.empty
|
||||
consumers.foreach { consumer =>
|
||||
consumer.close()
|
||||
}
|
||||
consumers = Seq.empty
|
||||
}
|
||||
processors.foreach(_.close())
|
||||
consumers.foreach(_.close())
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Initialize the graph writers,
|
||||
|
Loading…
Reference in New Issue
Block a user