find topic and partitions info from ZkUtils

KafkaTopicTest.scala

import kafka.utils.ZkUtils

object KafkaTopicTest {
  def createZkUtils(): ZkUtils = {
    val zookeepers = "localhost:2181"
    val timeout = 10000
    ZkUtils(zookeepers, timeout, timeout, isZkSecurityEnabled = false)
  }

  def getPartitionsForTopics(topics: Seq[String]): Map[String, Seq[Int]] = {
      zkUtils.getPartitionsForTopics(topics).toMap
    }
    
  def getPartitionsForTopic(topic: String): Option[Seq[Int]] = {
    zkUtils.getPartitionsForTopics(Seq(topic)).get(topic)
  }
  
  def main(args: Array[String]): Unit = {
    val topicName = args(0)
    val topicAndPartitionMap = getPartitionsForTopic(topicName)

    println(topicAndPartitionMap)
  }
}