Add partitioner callback to topic configuration

This commit is contained in:
Matias Fontanini
2016-06-04 19:15:32 -07:00
parent 8e37440f58
commit 4fccf277e0
7 changed files with 102 additions and 9 deletions

View File

@@ -2,26 +2,40 @@
#define CPPKAFKA_TOPIC_CONFIGURATION_H
#include <string>
#include <functional>
#include <librdkafka/rdkafka.h>
#include "clonable_ptr.h"
namespace cppkafka {
class Topic;
class Buffer;
class TopicConfiguration {
public:
using PartitionerCallback = std::function<int32_t(const Topic&, const Buffer& key,
int32_t partition_count)>;
TopicConfiguration();
void set(const std::string& name, const std::string& value);
void set_partitioner_callback(PartitionerCallback callback);
void set_as_opaque();
const PartitionerCallback& get_partitioner_callback() const;
rd_kafka_topic_conf_t* get_handle() const;
private:
using HandlePtr = ClonablePtr<rd_kafka_topic_conf_t, decltype(&rd_kafka_topic_conf_destroy),
using HandlePtr = ClonablePtr<rd_kafka_topic_conf_t,
decltype(&rd_kafka_topic_conf_destroy),
decltype(&rd_kafka_topic_conf_dup)>;
TopicConfiguration(rd_kafka_topic_conf_t* ptr);
static HandlePtr make_handle(rd_kafka_topic_conf_t* ptr);
HandlePtr handle_;
PartitionerCallback partitioner_callback_;
};
} // cppkafka