python消费kafka数据批量插入到es的方法

最新推荐文章于 2024-07-25 10:49:41 发布

python 一语呢喃

最新推荐文章于 2024-07-25 10:49:41 发布

阅读量1.7k

点赞数

分类专栏： python基础教程文章标签： python kafka 大数据

本文链接：https://blog.csdn.net/haoxun02/article/details/104334926

版权

本文介绍了如何使用Python结合pykafka消费Kafka消息，并利用Elasticsearch进行批量数据插入。首先，配置信息存于logging.conf中，安装Elasticsearch依赖。接着，针对Kafka 0.8版本，使用get_simple_consumer消费，每个应用独占一个partition以避免重复消费。通过设置consumer_timeout_ms实现超时退出，并用while True循环保证持续消费。最后，数据积累到一定量后一次性批量插入到Elasticsearch。目前该方法正处于批量压测阶段。

摘要由CSDN通过智能技术生成

今天小编就为大家分享一篇python消费kafka数据批量插入到es的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
1、es的批量插入

这是为了方便后期配置的更改，把配置信息放在logging.conf中

用elasticsearch来实现批量操作，先安装依赖包，sudo pip install Elasticsearch2

from elasticsearch import Elasticsearch 
class ImportEsData:
 
  logging.config.fileConfig("logging.conf")
  logger = logging.getLogger("msg")
 
  def __init__(self,hosts,index,type):
    self.es = Elasticsearch(hosts=hosts.strip(',').split(','), timeout=5000)
    self.index = index
    self.type = type
 
 
  def set_date(self,data): 
    # 批量处理 
    # es.index(index="test-index",doc_type="test-type",id=42,body={"any":"data","timestamp":datetime.now()})
    self.es.index(index=self.index,doc_type=self.index,body=data)

2、使用pykafka消费kafka

1.因为kafka是0.8，pykafka不支持zk，只能用get_simple_