如何使用 spring-kafka 暂停和恢复 @KafkaListener

Question

我已经实现了 Kafka 消费者，现在我有一个场景。

1. 从 Kafka 流中读取数据 2.2.5.Release via Srpingboot
2. 加载到数据库table1中
3. 将table1中的数据复制到table2中
4. 清空表1

要执行上述操作，我需要使用石英的调度作业（已编写）暂停/恢复 Kafka 消费者，该作业将数据从表 1 复制到表 2。但在此活动期间，我希望我的 Kafka 侦听器暂停，一旦复制完成，它应该恢复。

我的实现：

@KafkaListener(topicPartitions =
    { @TopicPartition(topic = "data_pipe", partitions = { "0" })})
public void listen(ConsumerRecord<String, String> cr) throws Exception {

Answer 1

如果你使用 'kafkaListener annotation' 自动创建 KafkaListenerEndpointRegistry bean，那么，你可以像这样使用它：

@Component
public class KafkaManager {

    private final KafkaListenerEndpointRegistry registry;

    public KafkaManager(KafkaListenerEndpointRegistry registry) {
        this.registry = registry;
    }
    public void pause() {
        registry.getListenerContainers().forEach(MessageListenerContainer::pause);
    }

    public void resume() {
        registry.getListenerContainers().forEach(MessageListenerContainer::resume);
    }
}

文档：https://docs.spring.io/spring-kafka/reference/html/#pause-resume