Java代码:与具有特定主题的kafka使用者检索最后n条消息

时间:2019-04-18 11:02:59

标签: java apache-kafka messaging kafka-consumer-api kafka-records

kafka版本:0.9.0.1

如果n = 20, 我必须得到一个主题的最后20条消息。

目前,我正在使用

kafkaConsumer.seekToBeginning();

我收到所有消息。我正在写一些逻辑,以获得最后20个。

我的主题可能有数十万条记录

public List<JSONObject> consumeMessages(String kafkaTopicName) {
  KafkaConsumer<String, String> kafkaConsumer = null;
  boolean flag = true;
  List<JSONObject> messagesFromKafka = new ArrayList<>();
  int recordCount = 0;
  int i = 0;
  int maxMessagesToReturn = 20;

  Properties props = new Properties();         
  props.put("bootstrap.servers", "localhost:9092");
  props.put("group.id", "project.group.id");
  props.put("max.partition.fetch.bytes", "1048576000");
  props.put("key.deserializer", "org.apache.kafka.common.serialization.StringDeserializer");
  props.put("value.deserializer", "org.apache.kafka.common.serialization.StringDeserializer");
  kafkaConsumer = new KafkaConsumer<>(props);

  kafkaConsumer.subscribe(Arrays.asList(kafkaTopicName));
  TopicPartition topicPartition = new TopicPartition(kafkaTopicName, 0);
  LOGGER.info("Subscribed to topic " + kafkaConsumer.listTopics());
  while (flag) {
    // will consume all the messages and store in records
    ConsumerRecords<String, String> records = kafkaConsumer.poll(1000);

    kafkaConsumer.seekToBeginning(topicPartition);

    // getting total records count
    recordCount = records.count();
    LOGGER.info("recordCount " + recordCount);
    for (ConsumerRecord<String, String> record : records) {
      if(record.value() != null) {
        if (i >= recordCount - maxMessagesToReturn) {
          // adding last 20 messages to messagesFromKafka
          LOGGER.info("kafkaMessage "+record.value());
          messagesFromKafka.add(new JSONObject(record.value()));
        }
        i++;
      }
    }
    if (recordCount > 0) {
      flag = false;
    }
  }
  kafkaConsumer.close();
  return messagesFromKafka;
}

1 个答案:

答案 0 :(得分:1)

您可以使用kafkaConsumer.seekToEnd(Collection<TopicPartition> partitions)来查找给定分区的最后一个偏移量。根据文档:

  

“查找每个给定分区的最后一个偏移量。此函数延迟计算,仅在调用poll(Duration)position(TopicPartition)时才在所有分区中寻找最终偏移量。如果没有提供分区,请为所有当前分配的分区寻找最终偏移量。”

然后,您可以使用position(TopicPartition partition)来检索特定分区的位置。

然后,您可以减少20条消息,并使用kafkaConsumer.seek(TopicPartition partition, long offset)来获取最新的20条消息。

简单

kafkaConsumer.seekToEnd(partitionList);
long endPosition = kafkaConsumer.position(topicPartiton);
long recentMessagesStartPosition = endPosition - maxMessagesToReturn;
kafkaConsumer.seek(topicPartition, recentMessagesStartPosition);

现在您可以使用poll()

检索最近的20条消息

这是简单的逻辑,但是如果您有多个分区,则还必须考虑这些情况。我没有尝试过,但是希望您能理解。