juplo.de Git - demos/kafka/training/blob - src/main/java/de/juplo/kafka/ApplicationRebalanceListener.java

   1 package de.juplo.kafka;
   2
   3 import lombok.RequiredArgsConstructor;
   4 import lombok.extern.slf4j.Slf4j;
   5 import org.apache.kafka.clients.consumer.ConsumerRebalanceListener;
   6 import org.apache.kafka.common.TopicPartition;
   7
   8 import java.util.Collection;
   9 import java.util.HashMap;
  10 import java.util.Map;
  11
  12
  13 @RequiredArgsConstructor
  14 @Slf4j
  15 public class ApplicationRebalanceListener implements ConsumerRebalanceListener
  16 {
  17   private final ApplicationRecordHandler recordHandler;
  18   private final String id;
  19
  20   @Override
  21   public void onPartitionsAssigned(Collection<TopicPartition> partitions)
  22   {
  23     partitions.forEach(tp ->
  24     {
  25       Integer partition = tp.partition();
  26       log.info("{} - adding partition: {}", id, partition);
  27       recordHandler.addPartition(partition, new HashMap<>());
  28     });
  29   }
  30
  31   @Override
  32   public void onPartitionsRevoked(Collection<TopicPartition> partitions)
  33   {
  34     partitions.forEach(tp ->
  35     {
  36       Integer partition = tp.partition();
  37       log.info("{} - removing partition: {}", id, partition);
  38       Map<String, Long> removed = recordHandler.removePartition(partition);
  39       for (String key : removed.keySet())
  40       {
  41         log.info(
  42             "{} - Seen {} messages for partition={}|key={}",
  43             id,
  44             removed.get(key),
  45             partition,
  46             key);
  47       }
  48     });
  49   }
  50 }