first commit

nodes/broker/kafka_utils/KafkaProducer.cpp

@@ -0,0 +1,167 @@
+
+#ifdef VP_WITH_KAFKA
+#include "KafkaProducer.h"
+
+// callbacks
+class ProducerDeliveryReportCb : public RdKafka::DeliveryReportCb {
+public:
+	void dr_cb(RdKafka::Message &message) {
+		if (message.err())
+			std::cerr << "Message delivery failed: " << message.errstr() << std::endl;
+		else {
+			// Message delivered to topic test [0] at offset 135000
+            /*
+			std::cerr << "Message delivered to topic " << message.topic_name()
+				<< " [" << message.partition() << "] at offset "
+				<< message.offset() << std::endl;*/
+		}
+	}
+};
+
+class ProducerEventCb : public RdKafka::EventCb {
+public:
+	void event_cb(RdKafka::Event &event) {
+		switch (event.type()) {
+		case RdKafka::Event::EVENT_ERROR:
+			std::cout << "RdKafka::Event::EVENT_ERROR: " << RdKafka::err2str(event.err()) << std::endl;
+			break;
+		case RdKafka::Event::EVENT_STATS:
+			//std::cout << "RdKafka::Event::EVENT_STATS: " << event.str() << std::endl;
+			break;
+		case RdKafka::Event::EVENT_LOG:
+			//std::cout << "RdKafka::Event::EVENT_LOG " << event.fac() << std::endl;
+			break;
+		case RdKafka::Event::EVENT_THROTTLE:
+			//std::cout << "RdKafka::Event::EVENT_THROTTLE " << event.broker_name() << std::endl;
+			break;
+		}
+	}
+};
+
+
+class HashPartitionerCb : public RdKafka::PartitionerCb {
+public:
+	int32_t partitioner_cb(const RdKafka::Topic *topic, const std::string *key,
+		int32_t partition_cnt, void *msg_opaque) {
+		char msg[128] = { 0 };
+		int32_t partition_id = generate_hash(key->c_str(), key->size()) % partition_cnt;
+		//                          [topic][key][partition_cnt][partition_id] 
+		//                          :[test][6419][2][1]
+		/*sprintf(msg, "HashPartitionerCb:topic:[%s], key:[%s]partition_cnt:[%d], partition_id:[%d]", topic->name().c_str(),
+			key->c_str(), partition_cnt, partition_id);
+		std::cout << msg << std::endl;*/
+		return partition_id;
+	}
+private:
+
+	static inline unsigned int generate_hash(const char *str, size_t len) {
+		unsigned int hash = 5381;
+		for (size_t i = 0; i < len; i++)
+			hash = ((hash << 5) + hash) + str[i];
+		return hash;
+	}
+};
+
+
+KafkaProducer::KafkaProducer(const std::string& brokers, const std::string& topic, int partition) {
+	m_brokers = brokers;
+	m_topicStr = topic;
+	m_partition = partition;
+
+	m_config = RdKafka::Conf::create(RdKafka::Conf::CONF_GLOBAL);
+	if(m_config==NULL)
+		std::cout << "Create RdKafka Conf failed." << std::endl;
+
+	m_topicConfig = RdKafka::Conf::create(RdKafka::Conf::CONF_TOPIC);
+	if (m_topicConfig == NULL)
+		std::cout << "Create RdKafka Topic Conf failed." << std::endl;
+
+	RdKafka::Conf::ConfResult errCode;
+	std::string errorStr;
+	m_dr_cb = new ProducerDeliveryReportCb;
+
+	errCode = m_config->set("dr_cb", m_dr_cb, errorStr);
+	if (errCode != RdKafka::Conf::CONF_OK)
+	{
+		std::cout << "Conf set failed:" << errorStr << std::endl;
+	}
+
+	m_event_cb = new ProducerEventCb;
+	errCode = m_config->set("event_cb", m_event_cb, errorStr);
+	if (errCode != RdKafka::Conf::CONF_OK)
+	{
+		std::cout << "Conf set failed:" << errorStr << std::endl;
+	}
+
+	m_partitioner_cb = new HashPartitionerCb;
+	errCode = m_topicConfig->set("partitioner_cb", m_partitioner_cb, errorStr);
+	if (errCode != RdKafka::Conf::CONF_OK)
+	{
+		std::cout << "Conf set failed:" << errorStr << std::endl;
+	}
+
+	errCode = m_config->set("statistics.interval.ms", "10000", errorStr);
+	if (errCode != RdKafka::Conf::CONF_OK)
+	{
+		std::cout << "Conf set failed:" << errorStr << std::endl;
+	}
+	errCode = m_config->set("message.max.bytes", "10240000", errorStr);
+	if (errCode != RdKafka::Conf::CONF_OK)
+	{
+		std::cout << "Conf set failed:" << errorStr << std::endl;
+	}
+	errCode = m_config->set("bootstrap.servers", m_brokers, errorStr);
+	if (errCode != RdKafka::Conf::CONF_OK)
+	{
+		std::cout << "Conf set failed:" << errorStr << std::endl;
+	}
+
+	m_producer = RdKafka::Producer::create(m_config, errorStr);
+	if (m_producer == NULL)
+	{
+		std::cout << "Create Producer failed:" << errorStr << std::endl;
+	}
+
+	m_topic = RdKafka::Topic::create(m_producer, m_topicStr, m_topicConfig, errorStr);
+	if (m_topic == NULL)
+	{
+		std::cout << "Create Topic failed:" << errorStr << std::endl;
+	}
+}
+
+KafkaProducer::~KafkaProducer() {
+	while (m_producer->outq_len() > 0) {
+		std::cerr << "Waiting for " << m_producer->outq_len() << std::endl;
+		m_producer->flush(5000);
+	}
+	delete m_config;
+	delete m_topicConfig;
+	delete m_topic;
+	delete m_producer;
+	delete m_dr_cb;
+	delete m_event_cb;
+	delete m_partitioner_cb;
+}
+
+
+void KafkaProducer::pushMessage(const std::string& str) {
+	int32_t len = str.length();
+	void* payload = const_cast<void*>(static_cast<const void*>(str.data()));
+	RdKafka::ErrorCode errorCode = m_producer->produce(
+		m_topic,
+		RdKafka::Topic::PARTITION_UA,
+		RdKafka::Producer::RK_MSG_COPY,
+		payload,
+		len,
+		NULL,
+		NULL);
+	m_producer->poll(0);
+	if (errorCode != RdKafka::ERR_NO_ERROR) {
+		std::cerr << "Produce failed: " << RdKafka::err2str(errorCode) << std::endl;
+		if (errorCode == RdKafka::ERR__QUEUE_FULL) {
+			m_producer->poll(100);
+		}
+	}
+}
+
+#endif