在 Spring 中管理生产者拦截器

从 3.0.0 版本开始，对于生产者拦截器，您可以直接让 Spring 将其管理为 Bean，而不是将拦截器的类名提供给 Apache Kafka 生产者配置。如果您采用这种方法，则需要在KafkaTemplate上设置此生产者拦截器。以下示例使用与上面相同的MyProducerInterceptor，但已更改为不使用内部配置属性。

public class MyProducerInterceptor implements ProducerInterceptor<String, String> {

    private final SomeBean bean;

    public MyProducerInterceptor(SomeBean bean) {
        this.bean = bean;
    }

    @Override
    public void configure(Map<String, ?> configs) {
    }

    @Override
    public ProducerRecord<String, String> onSend(ProducerRecord<String, String> record) {
        this.bean.someMethod("producer interceptor");
        return record;
    }

    @Override
    public void onAcknowledgement(RecordMetadata metadata, Exception exception) {
    }

    @Override
    public void close() {
    }

}

@Bean
public MyProducerInterceptor myProducerInterceptor(SomeBean someBean) {
  return new MyProducerInterceptor(someBean);
}

@Bean
public KafkaTemplate<String, String> kafkaTemplate(ProducerFactory<String, String> pf, MyProducerInterceptor myProducerInterceptor) {
   KafkaTemplate<String, String> kafkaTemplate = new KafkaTemplate<>(pf);
   kafkaTemplate.setProducerInterceptor(myProducerInterceptor);
}

在记录发送之前，会调用生产者拦截器的 onSend 方法。一旦服务器在发布数据后发送确认，就会调用 onAcknowledgement 方法。onAcknowledgement 在生产者调用任何用户回调之前被调用。

如果您有多个通过 Spring 管理的生产者拦截器需要应用于 KafkaTemplate，则需要使用 CompositeProducerInterceptor。CompositeProducerInterceptor 允许以顺序的方式添加各个生产者拦截器。来自底层 ProducerInterceptor 实现的方法将按照它们添加到 CompositeProducerInterceptor 的顺序被调用。