Spring Integration根据时间戳从Kafka主题读取消息

时间:2019-02-13 07:30:34

标签: spring-integration kafka-consumer-api spring-integration-dsl

使用spring kafka时,我可以使用以下代码基于时间戳读取来自主题的消息-

                ConsumerRecords<String, String> records = consumer.poll(100);
                if (flag) {
                    Map<TopicPartition, Long> query = new HashMap<>();
                    query.put(new TopicPartition(kafkaTopic, 0), millisecondsFromEpochToReplay);

                    Map<TopicPartition, OffsetAndTimestamp> result = consumer.offsetsForTimes(query);
                    if(result != null)
                    {
                        records = ConsumerRecords.empty();
                    }

                    result.entrySet().stream()
                            .forEach(entry -> consumer.seek(entry.getKey(), entry.getValue().offset()));

                    flag = false;
                }

如何使用 spring集成DSL KafkaMessageDrivenChannelAdapter来实现相同的功能? 我们如何设置集成流程并根据时间戳从主题读取消息?

1 个答案:

答案 0 :(得分:2)

使用ConsumerAwareRebalanceListener配置适配器的侦听器容器,并在分配分区后执行查找/查找。

编辑

使用Spring Boot(但是您可以在创建容器的同时配置容器)...

spring.kafka.consumer.enable-auto-commit=false
spring.kafka.consumer.auto-offset-reset=earliest
spring.kafka.consumer.group-id=so54664761

@SpringBootApplication
public class So54664761Application {

    public static void main(String[] args) {
        SpringApplication.run(So54664761Application.class, args);
    }

    @Bean
    public ApplicationRunner runner(KafkaTemplate<String, String> template) {
        return args -> template.send("so54664761", "foo");
    }

    @Bean
    public NewTopic topic() {
        return new NewTopic("so54664761", 1, (short) 1);
    }

    @Bean
    public IntegrationFlow flow(ConcurrentKafkaListenerContainerFactory<String, String> containerFactory) {
        ConcurrentMessageListenerContainer<String, String> container = container(containerFactory);
        return IntegrationFlows.from(new KafkaMessageDrivenChannelAdapter<>(container))
                .handle(System.out::println)
                .get();
    }

    @Bean
    public ConcurrentMessageListenerContainer<String, String> container(
            ConcurrentKafkaListenerContainerFactory<String, String> containerFactory) {

        ConcurrentMessageListenerContainer<String, String> container = containerFactory.createContainer("so54664761");
        container.getContainerProperties().setConsumerRebalanceListener(new ConsumerAwareRebalanceListener() {

            @Override
            public void onPartitionsAssigned(Consumer<?, ?> consumer, Collection<TopicPartition> partitions) {
                System.out.println("Partitions assigned - do the lookup/seeks here");
            }

        });
        return container;
    }

}

Partitions assigned - do the lookup/seeks here
GenericMessage [payload=foo, headers={kafka_offset=0, kafka_consumer=org.apache.kafka.clients.consumer.KafkaConsumer@2f5b2297, kafka_timestampType=CREATE_TIME, kafka_receivedMessageKey=null, kafka_receivedPartitionId=0, kafka_receivedTopic=so54664761, kafka_receivedTimestamp=1550241100112}]