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

如下所示：

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批量发送数据的实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。