import json import time import requests from pykafka.common import OffsetType from fatcat_client import ReleaseEntity from fatcat_tools import * from .worker_common import FatcatWorker class ElasticsearchReleaseWorker(FatcatWorker): """ Consumes from release-updates topic and pushes into (presumably local) elasticsearch. Uses a consumer group to manage offset. """ def __init__(self, kafka_hosts, consume_topic, poll_interval=10.0, offset=None, elasticsearch_backend="http://localhost:9200", elasticsearch_index="fatcat"): super().__init__(kafka_hosts=kafka_hosts, consume_topic=consume_topic, api_host_url=None) self.consumer_group = "elasticsearch-updates" self.elasticsearch_backend = elasticsearch_backend self.elasticsearch_index = elasticsearch_index def run(self): consume_topic = self.kafka.topics[self.consume_topic] consumer = consume_topic.get_balanced_consumer( consumer_group=self.consumer_group, managed=True, fetch_message_max_bytes=4000000, # up to ~4MBytes auto_commit_enable=True, auto_commit_interval_ms=30000, # 30 seconds compacted_topic=True, ) for msg in consumer: json_str = msg.value.decode('utf-8') release = entity_from_json(json_str, ReleaseEntity) #print(release) elasticsearch_endpoint = "{}/{}/release/{}".format( self.elasticsearch_backend, self.elasticsearch_index, release.ident) print("Updating document: {}".format(elasticsearch_endpoint)) resp = requests.post(elasticsearch_endpoint, json=release_to_elasticsearch(release)) assert resp.status_code in (200, 201) #consumer.commit_offsets()