scottk/wlan-cloud-lib-cppkafka
1
0
Fork 0
mirror of https://github.com/Telecominfraproject/wlan-cloud-lib-cppkafka.git synced 2025-11-01 02:57:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added member functions for static consumers

Browse Source
This commit is contained in:
Alexander Damian
2020-02-08 22:34:45 -05:00
parent 7d097df34d
commit 68ae525eba
2 changed files with 51 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
include/cppkafka/utils/poll_strategy_base.h
View File

@@ -84,6 +84,29 @@ public:
*/
Consumer& get_consumer() final;
/**
* \brief Creates partitions queues associated with the supplied partitions.
*
* This method contains a default implementation. It adds all the new queues belonging
* to the provided partition list and calls reset_state().
* To be used with static consumers.
*
* \param partitions Assigned topic partitions.
*/
virtual void assign(TopicPartitionList& partitions);
/**
* \brief Removes partitions queues associated with the supplied partitions.
*
* This method contains a default implementation. It removes all the queues
* belonging to the provided partition list and calls reset_state().
* To be used with static consumers.
*
* \param partitions Revoked topic partitions. If the partition list is empty
* all partitions will be revoked.
*/
virtual void revoke(const TopicPartitionList& partitions = {});
protected:
/**
* \brief Get the queues from all assigned partitions
@@ -111,8 +134,8 @@ protected:
/**
* \brief Function to be called when a new partition assignment takes place
*
* This method contains a default implementation. It adds all the new queues belonging
* to the provided partition list and calls reset_state().
* This method contains a default implementation. It calls assign()
* and invokes the user assignment callback.
*
* \param partitions Assigned topic partitions
*/
@@ -121,8 +144,8 @@ protected:
/**
* \brief Function to be called when an old partition assignment gets revoked
*
* This method contains a default implementation. It removes all the queues
* belonging to the provided partition list and calls reset_state().
* This method contains a default implementation. It calls revoke()
* and invokes the user revocation callback.
*
* \param partitions Revoked topic partitions
*/

34
src/utils/poll_strategy_base.cpp
View File

@@ -89,13 +89,35 @@ void PollStrategyBase::reset_state() {
}
void PollStrategyBase::on_assignment(TopicPartitionList& partitions) {
void PollStrategyBase::assign(TopicPartitionList& partitions) {
// populate partition queues
for (const auto& partition : partitions) {
// get the queue associated with this partition
partition_queues_.emplace(partition, QueueData{consumer_.get_partition_queue(partition), boost::any()});
}
reset_state();
}
void PollStrategyBase::revoke(const TopicPartitionList& partitions) {
if (partitions.empty()) {
//revoke everything
partition_queues_.clear();
}
else {
for (const auto &partition : partitions) {
// get the queue associated with this partition
auto toppar_it = partition_queues_.find(partition);
if (toppar_it != partition_queues_.end()) {
// remove this queue from the list
partition_queues_.erase(toppar_it);
}
}
}
reset_state();
}
void PollStrategyBase::on_assignment(TopicPartitionList& partitions) {
assign(partitions);
// call original consumer callback if any
if (assignment_callback_) {
assignment_callback_(partitions);
@@ -103,15 +125,7 @@ void PollStrategyBase::on_assignment(TopicPartitionList& partitions) {
}
void PollStrategyBase::on_revocation(const TopicPartitionList& partitions) {
for (const auto& partition : partitions) {
// get the queue associated with this partition
auto toppar_it = partition_queues_.find(partition);
if (toppar_it != partition_queues_.end()) {
// remove this queue from the list
partition_queues_.erase(toppar_it);
}
}
reset_state();
revoke(partitions);
// call original consumer callback if any
if (revocation_callback_) {
revocation_callback_(partitions);