kafka-python批量发送数据的方法

最新推荐文章于 2025-07-25 17:11:04 发布

原创最新推荐文章于 2025-07-25 17:11:04 发布 · 1.1w 阅读

2 ·

CC 4.0 BY-SA版权

BIGDATA 专栏收录该内容

57 篇文章

订阅专栏

本文介绍了一种将数据均匀分配到Kafka topic分区的方法，通过使用Python的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中。