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

Add consumer rebalance callbacks

Browse Source
This commit is contained in:
Matias Fontanini
2016-05-22 15:11:32 -07:00
parent 58258750df
commit 29989ea346
4 changed files with 86 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
include/cppkafka/consumer.h
View File

@@ -4,6 +4,7 @@
#include <vector>
#include <string>
#include <chrono>
#include <functional>
#include "kafka_handle_base.h"
#include "message.h"
@@ -14,14 +15,22 @@ class TopicConfiguration;
class Consumer : public KafkaHandleBase {
public:
Consumer(const Configuration& config);
using AssignmentCallback = std::function<void(const TopicPartitionList&)>;
using RevocationCallback = std::function<void(const TopicPartitionList&)>;
using RebalanceErrorCallback = std::function<void(rd_kafka_resp_err_t)>;
Consumer(Configuration config);
void set_timeout(const std::chrono::milliseconds timeout);
void set_assignment_callback(AssignmentCallback callback);
void set_revocation_callback(RevocationCallback callback);
void set_rebalance_error_callback(RebalanceErrorCallback callback);
void subscribe(const std::vector<std::string>& topics);
void unsubscribe();
void assign(const TopicPartitionList& topic_partitions);
void unassign();
void close();
void commit(const Message& msg);
@@ -38,10 +47,17 @@ public:
private:
static const std::chrono::milliseconds DEFAULT_TIMEOUT;
static void rebalance_proxy(rd_kafka_t *handle, rd_kafka_resp_err_t error,
rd_kafka_topic_partition_list_t *partitions, void *opaque);
void commit(const Message& msg, bool async);
void commit(const TopicPartitionList& topic_partitions, bool async);
void handle_rebalance(rd_kafka_resp_err_t err, const TopicPartitionList& topic_partitions);
std::chrono::milliseconds timeout_ms_;
AssignmentCallback assignment_callback_;
RevocationCallback revocation_callback_;
RebalanceErrorCallback rebalance_error_callback_;
};
} // cppkafka

6
include/cppkafka/topic_partition_list.h
View File

@@ -12,6 +12,8 @@ class TopicPartition;
class TopicPartitionList {
public:
static TopicPartitionList make_non_owning(rd_kafka_topic_partition_list_t* handle);
TopicPartitionList();
TopicPartitionList(rd_kafka_topic_partition_list_t* handle);
TopicPartitionList(size_t size);
@@ -40,11 +42,15 @@ public:
private:
static const size_t DEFAULT_CONTAINER_SIZE;
struct NonOwningTag { };
using HandlePtr = std::unique_ptr<rd_kafka_topic_partition_list_t,
decltype(&rd_kafka_topic_partition_list_destroy)>;
static HandlePtr make_handle(rd_kafka_topic_partition_list_t* ptr);
TopicPartitionList(rd_kafka_topic_partition_list_t* handle, NonOwningTag);
rd_kafka_topic_partition_t* get_topic_partition(const TopicPartition& topic_partition) const;
HandlePtr handle_;