python

超轻量级php框架startmvc

kafka-python批量发送数据的实例

更新时间:2020-06-17 23:36:01 作者:startmvc
如下所示:fromkafkaimportKafkaClientfromkafka.producerimportSimpleProducerdefsend_data_2_kafka(datas):'''向kafka

如下所示:


from kafka import KafkaClient
from kafka.producer import SimpleProducer

def send_data_2_kafka(datas):
 '''
 向kafka解析队列发送数据
 '''
 client = KafkaClient(hosts=KAFKABROKER.split(","), timeout=30)
 producer = SimpleProducer(client, async=False)
 
 curcount = len(datas)/PARTNUM
 for i in range(0, PARTNUM):
 start = i*curcount
 if i != PARTNUM - 1:
 end = (i+1)*curcount
 curdata = datas[start:end]
 producer.send_messages(TOPICNAME, *curdata)
 else:
 curdata = datas[start:]
 producer.send_messages(TOPICNAME, *curdata)
 
 producer.stop()
 client.close()

其中PARTNUM为topic的partition的数目,这样保证批量发送的数据均匀的落在kafka的partition中。

以上这篇kafka-python批量发送数据的实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。

kafka python 发送数据