Fix rest build due to upstream changes in ce-kafka (#18)

* Fix rest build due to upstream changes in ce-kafka

* Update KafkaConsumerState.java

* Use Duration.ofSeconds

---------

Co-authored-by: Akhilesh Chaganti <[email protected]>

Author: shubh-ranade

kafka-rest/src/main/java/io/confluent/kafkarest/v2/KafkaConsumerState.java

@@ -36,6 +36,7 @@
 import java.time.Instant;
 import java.util.ArrayDeque;
 import java.util.Collection;
+import java.util.Collections;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -269,7 +270,8 @@ public synchronized ConsumerCommittedResponse committed(ConsumerCommittedRequest
     if (consumer != null) {
       for (io.confluent.kafkarest.entities.v2.TopicPartition t : request.getPartitions()) {
         TopicPartition partition = new TopicPartition(t.getTopic(), t.getPartition());
-        OffsetAndMetadata offsetMetadata = consumer.committed(partition);
+        OffsetAndMetadata offsetMetadata =
+            consumer.committed(Collections.singleton(partition)).get(partition);
         if (offsetMetadata != null) {
           offsets.add(
               new TopicPartitionOffsetMetadata(
@@ -381,7 +383,7 @@ synchronized ConsumerRecord<KafkaKeyT, KafkaValueT> next() {
    * be invoked with the lock held, i.e. after startRead().
    */
   private synchronized void getOrCreateConsumerRecords() {
-    ConsumerRecords<KafkaKeyT, KafkaValueT> polledRecords = consumer.poll(0);
+    ConsumerRecords<KafkaKeyT, KafkaValueT> polledRecords = consumer.poll(Duration.ofSeconds(0L));
     // drain the iterator and buffer to list
     for (ConsumerRecord<KafkaKeyT, KafkaValueT> consumerRecord : polledRecords) {
       consumerRecords.add(consumerRecord);