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

Allow using generic message builders not tied to Buffer class

Browse Source
This commit is contained in:
Matias Fontanini
2017-04-16 09:47:39 -07:00
parent 861e41a792
commit f781afe5cf
7 changed files with 193 additions and 127 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
include/cppkafka/utils/buffered_producer.h
View File

@@ -17,6 +17,11 @@ namespace cppkafka {
template <typename BufferType>
class BufferedProducer {
public:
/**
* Concrete builder
*/
using Builder = ConcreteMessageBuilder<BufferType>;
/**
* \brief Constructs a buffered producer using the provided configuration
*
@@ -54,12 +59,12 @@ private:
using IndexType = std::conditional<sizeof(void*) == 8, uint64_t, uint32_t>::type;
const Topic& get_topic(const std::string& topic);
void produce_message(IndexType index, MessageBuilder& message);
void produce_message(IndexType index, Builder& message);
Configuration prepare_configuration(Configuration config);
void on_delivery_report(const Message& message);
Producer producer_;
std::map<IndexType, MessageBuilder> messages_;
std::map<IndexType, Builder> messages_;
std::vector<IndexType> failed_indexes_;
IndexType current_index_{0};
std::vector<Topic> topics_;
@@ -74,7 +79,7 @@ BufferedProducer<BufferType>::BufferedProducer(Configuration config)
template <typename BufferType>
void BufferedProducer<BufferType>::add_message(const MessageBuilder& builder) {
MessageBuilder local_builder(get_topic(builder.topic().get_name()));
Builder local_builder(get_topic(builder.topic().get_name()));
local_builder.partition(builder.partition());
local_builder.key(builder.key());
local_builder.payload(builder.payload());
@@ -122,13 +127,16 @@ const Topic& BufferedProducer<BufferType>::get_topic(const std::string& topic) {
}
template <typename BufferType>
void BufferedProducer<BufferType>::produce_message(IndexType index,
MessageBuilder& builder) {
void BufferedProducer<BufferType>::produce_message(IndexType index, Builder& builder) {
bool sent = false;
builder.user_data(reinterpret_cast<void*>(index));
MessageBuilder local_builder(builder.topic());
local_builder.partition(builder.partition())
.key(builder.key())
.payload(builder.payload())
.user_data(reinterpret_cast<void*>(index));
while (!sent) {
try {
producer_.produce(builder);
producer_.produce(local_builder);
sent = true;
}
catch (const HandleException& ex) {