Add back LiKafkaProducer.partitionsFor() that takes a set of topics (#…

…101)
linkedin · Dec 10, 2018 · 756b70f · 756b70f
1 parent 1967b6b
commit 756b70f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/...ache-kafka-clients/src/main/java/com/linkedin/kafka/clients/producer/LiKafkaProducer.java b/...ache-kafka-clients/src/main/java/com/linkedin/kafka/clients/producer/LiKafkaProducer.java
@@ -6,6 +6,10 @@
 
 import com.linkedin.kafka.clients.annotations.InterfaceOrigin;
 import java.util.concurrent.TimeUnit;
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+import org.apache.kafka.common.PartitionInfo;
 import org.apache.kafka.clients.producer.Producer;
 
 
@@ -23,4 +27,7 @@ public interface LiKafkaProducer<K, V> extends Producer<K, V> {
    */
   @InterfaceOrigin.LiKafkaClients
   void flush(long timeout, TimeUnit timeUnit);
+
+  @InterfaceOrigin.LiKafkaClients
+  Map<String, List<PartitionInfo>> partitionsFor(Set<String> topics);
 }
diff --git a/...-kafka-clients/src/main/java/com/linkedin/kafka/clients/producer/LiKafkaProducerImpl.java b/...-kafka-clients/src/main/java/com/linkedin/kafka/clients/producer/LiKafkaProducerImpl.java
@@ -351,6 +351,7 @@ public List<PartitionInfo> partitionsFor(String topic) {
     return _producer.partitionsFor(topic);
   }
 
+  @Override
   public Map<String, List<PartitionInfo>> partitionsFor(Set<String> topics) {
     //TODO come back here when upstream API settles
     throw new UnsupportedOperationException("not implemented");