public PartitionInfo GetPartition(TopicPartition topicPartition) { return partitionsByTopicPartition[topicPartition]; }
public Node GetLeaderFor(TopicPartition topicPartition) { PartitionInfo info = partitionsByTopicPartition[topicPartition]; if (info == null) return null; else return info.Leader; }