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

Add more wrappers for Consumer

Browse Source
This commit is contained in:
Matias Fontanini
2016-05-22 09:32:17 -07:00
parent c491136e0e
commit 83c1d304c6
4 changed files with 35 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
include/cppkafka/consumer.h
View File

@@ -23,14 +23,18 @@ public:
void unsubscribe();
void assign(const TopicPartitionList& topic_partitions);
void close();
void commit(const Message& msg);
void async_commit(const Message& msg);
void commit(const TopicPartitionList& topic_partitions);
void async_commit(const TopicPartitionList& topic_partitions);
TopicPartitionList get_committed(const TopicPartitionList& topic_partitions);
TopicPartitionList get_position(const TopicPartitionList& topic_partitions);
TopicPartitionList get_offsets_committed(const TopicPartitionList& topic_partitions);
TopicPartitionList get_offsets_position(const TopicPartitionList& topic_partitions);
TopicPartitionList get_subscription();
TopicPartitionList get_assignment();
Message poll();
private:

1
include/cppkafka/topic_partition_list.h
View File

@@ -13,6 +13,7 @@ class TopicPartition;
class TopicPartitionList {
public:
TopicPartitionList();
TopicPartitionList(rd_kafka_topic_partition_list_t* handle);
TopicPartitionList(size_t size);
template <typename ForwardIterator>
TopicPartitionList(ForwardIterator start, const ForwardIterator& end)

25
src/consumer.cpp
View File

@@ -45,6 +45,11 @@ void Consumer::assign(const TopicPartitionList& topic_partitions) {
check_error(error);
}
void Consumer::close() {
rd_kafka_resp_err_t error = rd_kafka_consumer_close(get_handle());
check_error(error);
}
void Consumer::commit(const Message& msg) {
commit(msg, false);
}
@@ -61,7 +66,7 @@ void Consumer::async_commit(const TopicPartitionList& topic_partitions) {
commit(topic_partitions, true);
}
TopicPartitionList Consumer::get_committed(const TopicPartitionList& topic_partitions) {
TopicPartitionList Consumer::get_offsets_committed(const TopicPartitionList& topic_partitions) {
// Copy the list, let rd_kafka change it and return it
TopicPartitionList output = topic_partitions;
rd_kafka_resp_err_t error = rd_kafka_committed(get_handle(), output.get_handle(),
@@ -70,7 +75,7 @@ TopicPartitionList Consumer::get_committed(const TopicPartitionList& topic_parti
return output;
}
TopicPartitionList Consumer::get_position(const TopicPartitionList& topic_partitions) {
TopicPartitionList Consumer::get_offsets_position(const TopicPartitionList& topic_partitions) {
// Copy the list, let rd_kafka change it and return it
TopicPartitionList output = topic_partitions;
rd_kafka_resp_err_t error = rd_kafka_position(get_handle(), output.get_handle());
@@ -78,6 +83,22 @@ TopicPartitionList Consumer::get_position(const TopicPartitionList& topic_partit
return output;
}
TopicPartitionList Consumer::get_subscription() {
rd_kafka_resp_err_t error;
rd_kafka_topic_partition_list_t* list = nullptr;
error = rd_kafka_subscription(get_handle(), &list);
check_error(error);
return TopicPartitionList(list);
}
TopicPartitionList Consumer::get_assignment() {
rd_kafka_resp_err_t error;
rd_kafka_topic_partition_list_t* list = nullptr;
error = rd_kafka_assignment(get_handle(), &list);
check_error(error);
return TopicPartitionList(list);
}
Message Consumer::poll() {
rd_kafka_message_t* message = rd_kafka_consumer_poll(get_handle(), timeout_ms_.count());
return Message(message);

5
src/topic_partition_list.cpp
View File

@@ -11,6 +11,11 @@ TopicPartitionList::TopicPartitionList()
}
TopicPartitionList::TopicPartitionList(rd_kafka_topic_partition_list_t* handle)
: handle_(make_handle(handle)) {
}
TopicPartitionList::TopicPartitionList(size_t size)
: handle_(make_handle(rd_kafka_topic_partition_list_new(size))) {