Allow polling with a timeout

This commit is contained in:
Matias Fontanini
2016-10-08 10:08:15 -07:00
parent 8361e4ca1e
commit a46896790b
6 changed files with 38 additions and 6 deletions

View File

@@ -188,8 +188,11 @@ const Consumer::RebalanceErrorCallback& Consumer::get_rebalance_error_callback()
}
Message Consumer::poll() {
rd_kafka_message_t* message = rd_kafka_consumer_poll(get_handle(),
get_timeout().count());
return poll(get_timeout());
}
Message Consumer::poll(milliseconds timeout) {
rd_kafka_message_t* message = rd_kafka_consumer_poll(get_handle(), timeout.count());
return message ? Message(message) : Message();
}