from aioelasticsearch import Elasticsearch

from asyncworker import App, Options
from asyncworker.rabbitmq import RabbitMQMessage


app = App(host="localhost", user="guest", password="guest", prefetch_count=512)
elasticsearch = Elasticsearch()


@app.route(["words_to_index"], vhost="/", options={Options.BULK_SIZE: 256})
async def drain_handler(messages: RabbitMQMessage):
    await elasticsearch.bulk([message.body for message in messages])


app.run()