-package de.juplo.kafka;
-
-import lombok.RequiredArgsConstructor;
-import lombok.extern.slf4j.Slf4j;
-import org.apache.kafka.clients.consumer.ConsumerRebalanceListener;
-import org.apache.kafka.common.TopicPartition;
-import org.springframework.kafka.listener.ConsumerAwareRebalanceListener;
-
-import java.util.*;
-
-
-@RequiredArgsConstructor
-@Slf4j
-public class ApplicationRebalanceListener implements ConsumerAwareRebalanceListener
-{
- private final ApplicationRecordHandler recordHandler;
- private final AdderResults adderResults;
- private final StateRepository stateRepository;
- private final String id;
-
- private final Set<Integer> partitions = new HashSet<>();
-
- @Override
- public void onPartitionsAssigned(Collection<TopicPartition> partitions)
- {
- partitions.forEach(tp ->
- {
- Integer partition = tp.partition();
- log.info("{} - adding partition: {}", id, partition);
- this.partitions.add(partition);
- StateDocument document =
- stateRepository
- .findById(Integer.toString(partition))
- .orElse(new StateDocument(partition));
- recordHandler.addPartition(partition, document.state);
- for (String user : document.state.keySet())
- {
- log.info(
- "{} - Restored state for partition={}|user={}: {}",
- id,
- partition,
- user,
- document.state.get(user));
- }
- adderResults.addPartition(partition, document.results);
- });
- }
-
- @Override
- public void onPartitionsRevoked(Collection<TopicPartition> partitions)
- {
- partitions.forEach(tp ->
- {
- Integer partition = tp.partition();
- log.info("{} - removing partition: {}", id, partition);
- this.partitions.remove(partition);
- Map<String, AdderResult> state = recordHandler.removePartition(partition);
- for (String user : state.keySet())
- {
- log.info(
- "{} - Saved state for partition={}|user={}: {}",
- id,
- partition,
- user,
- state.get(user));
- }
- Map<String, List<AdderResult>> results = adderResults.removePartition(partition);
- stateRepository.save(new StateDocument(partition, state, results));
- });
- }
-}