kafka-python实现数据均匀批量发送至Kafka分区的方法

慢慢来

发布时间：2026-10-13| 6人浏览过

原创

本文介绍使用kafka-python库通过SimpleProducer向Kafka发送批量数据的实现方式。核心逻辑是将数据按Topic分区数（PARTNUM）均分，确保每条数据均匀落入不同Partition。代码展示了如何初始化客户端、计算切片范围并调用send_messages方法，最后关闭连接，适用于需要保证数据负载均衡的场景。

具体实现代码如下所示：

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的分区数量。通过这种方式，可以确保批量发送的数据能够均匀地分布到Kafka的各个Partition中，从而实现负载均衡。

以上关于kafka-python批量发送数据的实例分享完毕，希望能为读者提供有价值的参考。如果您觉得内容有用，欢迎继续支持本站获取更多技术分享。