是否可以使用密钥和分区使用kafka消息?

时间:2019-11-06 06:57:45

标签: java apache-kafka kafka-partition

我正在使用 kafka_2.12 版本 2.3.0 ,其中我使用分区和密钥将数据发布到kafka主题中。我需要找到一种方法,可以使用键和分区组合使用主题中的特定消息。这样,我将不必消耗所有消息并为正确的消息进行迭代。

现在我只能这样做

KafkaConsumer<String, String> consumer = new KafkaConsumer<String, String>(props)
consumer.subscribe(Collections.singletonList("topic"))
ConsumerRecords<String, String> records = consumer.poll(100)
def data = records.findAll {
    it -> it.key().equals(key)
}

2 个答案:

答案 0 :(得分:1)

您不能“从Kafka通过密钥获取消息”。

如果可行的话,一种解决方案是拥有与键一样多的分区,并始终将键的消息路由到同一分区。

消息密钥作为分区

kafkaConsumer.assign(topicPartitions);
    kafkaConsumer.seekToBeginning(topicPartitions);

    // Pull records from kafka, keep polling until we get nothing back
    final List<ConsumerRecord<byte[], byte[]>> allRecords = new ArrayList<>();
    ConsumerRecords<byte[], byte[]> records;
    do {
        // Grab records from kafka
        records = kafkaConsumer.poll(2000L);
        logger.info("Found {} records in kafka", records.count());

        // Add to our array list
        records.forEach(allRecords::add);

    }
    while (!records.isEmpty());

仅使用主题名称访问主题的消息

KafkaConsumer<String, String> consumer = new KafkaConsumer<>(props);
     consumer.subscribe(Arrays.asList(<Topic Name>,<Topic Name>));
     while (true) {
         ConsumerRecords<String, String> records = consumer.poll(100);
         for (ConsumerRecord<String, String> record : records)
             System.out.printf("offset = %d, key = %s, value = %s%n", record.offset(), record.key(), record.value());
     }

答案 1 :(得分:1)

有两种使用主题/分区的方法:

  1. KafkaConsumer.assign():Document link
  2. KafkaConsumer.subscribe():Document link

因此,您不能通过按键获取消息。

如果您没有扩展分区的计划,请考虑使用assign()方法。因为带有特定密钥的所有消息都将进入同一分区。

使用方法:

KafkaConsumer<String, String> consumer = new KafkaConsumer<String, String>(properties);
TopicPartition partition = new TopicPartition("some-topic", 0);
consumer.assign(Arrays.asList(partition));

while(true){
    ConsumerRecords<String, String> records = consumer.poll(Duration.ofMillis(100));
    String data = records.findAll {
        it -> it.key().equals(key)
    }
}