public class StickyAssignorConsumerRebalanceListener implements ConsumerRebalanceListener {

    ...
 
    private Collection<TopicPartition> revokedPartitions;
 
    @Override
    public void onPartitionsAssignedonPartitionsRevoked(Collection<TopicPartition> partitions) {
        this.revokedPartitions = partitions;
    }
 
    @Override
    public void onPartitionsRevokedonPartitionsAssigned(Collection<TopicPartition> partitions) {
        this.revokedPartitions.removeAll(partitions);
        cleanup(this.revokedPartitions);
    }
 
    ...
}