Spring Integration 和 Kafka：如何根据消息头过滤消息

Question

我有一个基于这个问题的问题：Filter messages before deserialization based on headers

我想使用 Spring Integration DSL 按 kafka 消费者记录标头进行过滤。

目前我有这个流程：

@Bean
IntegrationFlow readTicketsFlow(KafkaProperties kafkaProperties,
                                ObjectMapper jacksonObjectMapper,
                                EventService<Ticket> service) {
    Map<String, Object> consumerProperties = kafkaProperties.buildConsumerProperties();
    DefaultKafkaConsumerFactory<String, String> consumerFactory = new DefaultKafkaConsumerFactory<>(consumerProperties);

    return IntegrationFlows.from(
            Kafka.messageDrivenChannelAdapter(
                    consumerFactory, TICKET_TOPIC))
            .transform(fromJson(Ticket.class, new Jackson2JsonObjectMapper(jacksonObjectMapper)))
            .handle(service)
            .get();
}

如何在此流程中注册org.springframework.kafka.listener.adapter.RecordFilterStrategy？

Answer 1

您可以简单地将.filter() 元素添加到流中。

.filter("!'bar'.equals(headers['foo'])")

将过滤掉（忽略）任何标头名为 foo 等于 bar 的消息。

注意 Spring Kafka 的 RecordFilterStrategy 具有 Spring Integration 过滤器的反向含义

public interface RecordFilterStrategy<K, V> {

    /**
     * Return true if the record should be discarded.
     * @param consumerRecord the record.
     * @return true to discard.
     */
    boolean filter(ConsumerRecord<K, V> consumerRecord);

}

如果过滤器返回 false，则 Spring Integration 过滤器会丢弃消息。

编辑

或者您可以将RecordFilterStrategy 添加到通道适配器。

return IntegrationFlows
        .from(Kafka.messageDrivenChannelAdapter(consumerFactory(), TEST_TOPIC1)
                .recordFilterStrategy(record -> {
                    Header header = record.headers().lastHeader("foo");
                    return header != null ? new String(header.value()).equals("bar") : false;
                })
                ...