Use tags on EOF and timeout callbacks on ConsumerDispatcher

This commit is contained in:
Matias Fontanini
2017-06-10 17:39:23 -07:00
parent f0ec0bfb10
commit 4369b75695
3 changed files with 31 additions and 16 deletions

View File

@@ -39,7 +39,7 @@ ConsumerRunner::ConsumerRunner(Consumer& consumer, size_t expected, size_t parti
}
},
// EOF callback
[&](const TopicPartition& topic_partition) {
[&](ConsumerDispatcher::EndOfFile, const TopicPartition& topic_partition) {
if (number_eofs != partitions) {
number_eofs++;
if (number_eofs == partitions) {
@@ -50,7 +50,7 @@ ConsumerRunner::ConsumerRunner(Consumer& consumer, size_t expected, size_t parti
}
},
// Timeout callback
[&]() {
[&](ConsumerDispatcher::Timeout) {
if (expected > 0 && messages_.size() == expected) {
dispatcher.stop();
}