小言_互联网的博客

Kafka学习笔记: 拦截器

451人阅读  评论(0)

 

Kafka 拦截器分为生产者拦截器和消费者拦截器。生产者拦截器允许你在发送消息前以及消息提交成功后植入你的拦截器逻辑;而消费者拦截器支持在消费消息前以及提交位移后编写特定逻辑。 这两种拦截器都支持链的方式,即你可以将一组拦截器串连成一个大的拦截器,Kafka 会按照添加顺序依次执行拦截器逻辑。

 

假设你想在生产消息前执行两个“前置动作”:第一个是为消息增加一个头信息,封装发送该消息的时间第二个是更新发送消息数字段,那么当你将这两个拦截器串联在一起统一指定给 Producer 后,Producer 会按顺序执行上面的动作,然后再发送消息。

当前 Kafka 拦截器的设置方法是通过参数配置完成的。生产者和消费者两端有一个相同的参数,名字叫 interceptor.classes,它指定的是一组类的列表,每个类就是特定逻辑的拦截器实现类。拿上面的例子来说,假设

第一个拦截器的完整类路径是 com.yourcompany.kafkaproject.interceptors.AddTimeStampInterceptor,

第二个类是 com.yourcompany.kafkaproject.interceptors.UpdateCounterInterceptor,可以按照以下方法在 Producer 端指定拦截器:


  
  1. Properties props = new Properties();
  2. List<String> interceptors = new ArrayList<>();
  3. interceptors. add( "com.yourcompany.kafkaproject.interceptors.AddTimestampInterceptor"); // 拦截器1
  4. interceptors. add( "com.yourcompany.kafkaproject.interceptors.UpdateCounterInterceptor"); // 拦截器2
  5. props.put(ProducerConfig.INTERCEPTOR_CLASSES_CONFIG, interceptors);
  6. ……

 

生产者拦截器 要实现 : org.apache.kafka.clients.producer.ProducerInterceptor 接口。

1. onSend:该方法会在消息发送之前被调用。

2.onAcknowledgement:该方法会在消息成功提交或发送失败之后被调用。onAcknowledgement 的调用要早于 callback 的调用。值得注意的是,这个方法和 onSend 不是在同一个线程中被调用的,因此如果你在这两个方法中调用了某个共享可变对象,一定要保证线程安全哦。还有一点很重要,这个方法处在 Producer 发送的主路径中,所以最好别放一些太重的逻辑进去,否则你会发现你的 Producer TPS 直线下降。

 

消费者拦截器 要实现org.apache.kafka.clients.consumer.ConsumerInterceptor 接口.

1. onConsume:该方法在消息返回给 Consumer 程序之前调用。

2. onCommit:Consumer 在提交位移之后调用该方法。通常你可以在该方法中做一些记账类的动作,比如打日志等。

 

实例

某个业务只有一个 Producer 和一个 Consumer,他们想知道该业务消息从被生产出来到最后被消费的平均总时长是多少.

思路:

生产者: 在发送消息前更新总的已发送消息数。假设数据被保存在 Redis 中。

消费者:  首先更新了它们的总延时,方法就是用当前的时钟时间减去封装在消息中的创建时间,然后累计得到这批消息总的端到端处理延时并更新到 Redis 中。最后分别从 Redis 中读取更新过的总延时和总消息数,两者相除即得到端到端消息的平均处理延时。

 

生产者代码:


  
  1. public class AvgLatencyProducerInterceptor implements ProducerInterceptor<String, String> {
  2. private Jedis jedis; // 省略Jedis初始化
  3. @Override
  4. public ProducerRecord<String, String> onSend(ProducerRecord<String, String> record) {
  5. jedis.incr( "totalSentMessage");
  6. return record;
  7. }
  8. @Override
  9. public void onAcknowledgement(RecordMetadata metadata, Exception exception) {
  10. }
  11. @Override
  12. public void close() {
  13. }
  14. @Override
  15. public void configure(Map<java.lang.String, ?> configs) {
  16. }

消费者代码:


  
  1. public class AvgLatencyConsumerInterceptor implements ConsumerInterceptor<String, String> {
  2. private Jedis jedis; //省略Jedis初始化
  3. @Override
  4. public ConsumerRecords<String, String> onConsume(ConsumerRecords<String, String> records) {
  5. long lantency = 0L;
  6. for (ConsumerRecord<String, String> record : records) {
  7. lantency += (System.currentTimeMillis() - record.timestamp());
  8. }
  9. jedis.incrBy( "totalLatency", lantency);
  10. long totalLatency = Long.parseLong(jedis.get( "totalLatency"));
  11. long totalSentMsgs = Long.parseLong(jedis.get( "totalSentMessage"));
  12. jedis.set( "avgLatency", String.valueOf(totalLatency / totalSentMsgs));
  13. return records;
  14. }
  15. @Override
  16. public void onCommit(Map<TopicPartition, OffsetAndMetadata> offsets) {
  17. }
  18. @Override
  19. public void close() {
  20. }
  21. @Override
  22. public void configure(Map<String, ?> configs) {

 

 

 

 

 

 

 

 

 

参考引用:

Kafka核心技术与实战 - 胡夕
 

 

 

 

 

 

 

 


转载:https://blog.csdn.net/zhanglong_4444/article/details/103686235
查看评论
* 以上用户言论只代表其个人观点,不代表本网站的观点或立场