python 消費(fèi) kafka 數(shù)據(jù)教程
1.安裝python模塊
pip install --user kafka-python==1.4.3
如果報(bào)錯(cuò)壓縮相關(guān)的錯(cuò)嘗試安裝下面的依賴
yum install snappy-devel yum install lz4-devel pip install python-snappy pip install lz4
2.生產(chǎn)者
#!/usr/bin/env python # coding : utf-8 from kafka import KafkaProducer import json def kafkaProducer(): producer = KafkaProducer(bootstrap_servers='ip:9092',value_serializer=lambda v: json.dumps(v).encode('utf-8')) producer.send('world', {'key1': 'value1'}) if __name__ == '__main__': kafkaProducer()
2.消費(fèi)者
from kafka import KafkaConsumer from kafka.structs import TopicPartition import time import click import ConfigParser import json import threading import datetime import sched config = ConfigParser.ConfigParser() config.read("amon.ini") @click.group() def cli(): pass @cli.command() @click.option('--topic',type=str) @click.option('--offset', type=click.Choice(['smallest', 'earliest', 'largest'])) @click.option("--group",type=str) def client(topic,offset,group): click.echo(topic) consumer = KafkaConsumer(topic, bootstrap_servers=config.get("KAFKA", "Broker_Servers").split(","), group_id=group, auto_offset_reset=offset) for message in consumer: click.echo(message.value) # click.echo("%d:%d: key=%s value=%s" % (message.partition, # message.offset, message.key, # message.value)) if __name__ == '__main__': cli()
3.多線程消費(fèi)
#coding:utf-8 import threading import os import sys from kafka import KafkaConsumer, TopicPartition, OffsetAndMetadata from collections import OrderedDict threads = [] class MyThread(threading.Thread): def __init__(self, thread_name, topic, partition): threading.Thread.__init__(self) self.thread_name = thread_name self.partition = partition self.topic = topic def run(self): print("Starting " + self.name) Consumer(self.thread_name, self.topic, self.partition) def stop(self): sys.exit() def Consumer(thread_name, topic, partition): broker_list = 'ip1:9092,ip2:9092' ''' fetch_min_bytes(int) - 服務(wù)器為獲取請求而返回的最小數(shù)據(jù)量,否則請等待 fetch_max_wait_ms(int) - 如果沒有足夠的數(shù)據(jù)立即滿足fetch_min_bytes給出的要求,服務(wù)器在回應(yīng)提取請求之前將阻塞的最大時(shí)間量(以毫秒為單位) fetch_max_bytes(int) - 服務(wù)器應(yīng)為獲取請求返回的最大數(shù)據(jù)量。這不是絕對最大值,如果獲取的第一個(gè)非空分區(qū)中的第一條消息大于此值, 則仍將返回消息以確保消費(fèi)者可以取得進(jìn)展。注意:使用者并行執(zhí)行對多個(gè)代理的提取,因此內(nèi)存使用將取決于包含該主題分區(qū)的代理的數(shù)量。 支持的Kafka版本> = 0.10.1.0。默認(rèn)值:52428800(50 MB)。 enable_auto_commit(bool) - 如果為True,則消費(fèi)者的偏移量將在后臺(tái)定期提交。默認(rèn)值:True。 max_poll_records(int) - 單次調(diào)用中返回的最大記錄數(shù)poll()。默認(rèn)值:500 max_poll_interval_ms(int) - poll()使用使用者組管理時(shí)的調(diào)用之間的最大延遲 。這為消費(fèi)者在獲取更多記錄之前可以閑置的時(shí)間量設(shè)置了上限。 如果 poll()在此超時(shí)到期之前未調(diào)用,則認(rèn)為使用者失敗,并且該組將重新平衡以便將分區(qū)重新分配給另一個(gè)成員。默認(rèn)300000 ''' consumer = KafkaConsumer(bootstrap_servers=broker_list, group_id="test000001", client_id=thread_name, enable_auto_commit=False, fetch_min_bytes=1024 * 1024, # 1M # fetch_max_bytes=1024 * 1024 * 1024 * 10, fetch_max_wait_ms=60000, # 30s request_timeout_ms=305000, # consumer_timeout_ms=1, # max_poll_records=5000, ) # 設(shè)置topic partition tp = TopicPartition(topic, partition) # 分配該消費(fèi)者的TopicPartition,也就是topic和partition,根據(jù)參數(shù),每個(gè)線程消費(fèi)者消費(fèi)一個(gè)分區(qū) consumer.assign([tp]) #獲取上次消費(fèi)的最大偏移量 offset = consumer.end_offsets([tp])[tp] print(thread_name, tp, offset) # 設(shè)置消費(fèi)的偏移量 consumer.seek(tp, offset) print u"程序首次運(yùn)行\(zhòng)t線程:", thread_name, u"分區(qū):", partition, u"偏移量:", offset, u"\t開始消費(fèi)..." num = 0 # 記錄該消費(fèi)者消費(fèi)次數(shù) while True: msg = consumer.poll(timeout_ms=60000) end_offset = consumer.end_offsets([tp])[tp] '''可以自己記錄控制消費(fèi)''' print u'已保存的偏移量', consumer.committed(tp), u'最新偏移量,', end_offset if len(msg) > 0: print u"線程:", thread_name, u"分區(qū):", partition, u"最大偏移量:", end_offset, u"有無數(shù)據(jù),", len(msg) lines = 0 for data in msg.values(): for line in data: print line lines += 1 ''' do something ''' # 線程此批次消息條數(shù) print(thread_name, "lines", lines) if True: # 可以自己保存在各topic, partition的偏移量 # 手動(dòng)提交偏移量 offsets格式:{TopicPartition:OffsetAndMetadata(offset_num,None)} consumer.commit(offsets={tp: (OffsetAndMetadata(end_offset, None))}) if True == 0: # 系統(tǒng)退出?這個(gè)還沒試 os.exit() ''' sys.exit() 只能退出該線程,也就是說其它兩個(gè)線程正常運(yùn)行,主程序不退出 ''' else: os.exit() else: print thread_name, '沒有數(shù)據(jù)' num += 1 print thread_name, "第", num, "次" if __name__ == '__main__': try: t1 = MyThread("Thread-0", "test", 0) threads.append(t1) t2 = MyThread("Thread-1", "test", 1) threads.append(t2) t3 = MyThread("Thread-2", "test", 2) threads.append(t3) for t in threads: t.start() for t in threads: t.join() print("exit program with 0") except: print("Error: failed to run consumer program")
參考:https://kafka-python.readthedocs.io/en/master/index.html
http://www.dbjr.com.cn/article/176911.htm
以上這篇python 消費(fèi) kafka 數(shù)據(jù)教程就是小編分享給大家的全部內(nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
Python中將圖像轉(zhuǎn)換為PDF的方法實(shí)現(xiàn)
本文主要介紹了Python中將圖像轉(zhuǎn)換為PDF的方法實(shí)現(xiàn),主要使用img2pdf和PyPDF2軟件包,具有一定的參考價(jià)值,感興趣的可以了解一下2023-08-08OpenCV計(jì)算平均值cv::mean實(shí)例代碼
函數(shù)cv::mean計(jì)算數(shù)組元素的平均值M,每個(gè)通道都是獨(dú)立的,并返回這個(gè)平均值,這篇文章主要給大家介紹了關(guān)于OpenCV計(jì)算平均值cv::mean的相關(guān)資料,需要的朋友可以參考下2021-08-08Pytorch出現(xiàn)錯(cuò)誤Attribute?Error:module?‘torch‘?has?no?attrib
這篇文章主要給大家介紹了關(guān)于Pytorch出現(xiàn)錯(cuò)誤Attribute?Error:module?‘torch‘?has?no?attribute?'_six'解決的相關(guān)資料,文中通過圖文介紹的非常詳細(xì),需要的朋友可以參考下2023-11-11windows系統(tǒng)多個(gè)python中更改默認(rèn)python版本
這篇文章主要給大家介紹了關(guān)于windows系統(tǒng)多個(gè)python中更改默認(rèn)python版本的相關(guān)資料,在Python開發(fā)中,不同的項(xiàng)目往往需要使用不同的Python版本,需要的朋友可以參考下2023-09-09