网络编程
位置:首页>> 网络编程>> Python编程>> kafka-python批量发送数据的实例

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

作者:rongyongfeikai2  发布时间:2023-04-24 04:40:23 

标签: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中。

来源:https://blog.csdn.net/rongyongfeikai2/article/details/54576340

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com