fix es sync.py

This commit is contained in:
Jack Robison 2021-09-13 19:10:52 -04:00 committed by Victor Shyba
parent 589a6588da
commit c4126edb79

View file

@ -57,8 +57,9 @@ async def run_sync(index_name='claims', db=None, clients=32):
logging.info("ES sync host: %s:%i", env.elastic_host, env.elastic_port) logging.info("ES sync host: %s:%i", env.elastic_host, env.elastic_port)
es = AsyncElasticsearch([{'host': env.elastic_host, 'port': env.elastic_port}]) es = AsyncElasticsearch([{'host': env.elastic_host, 'port': env.elastic_port}])
claim_generator = get_all_claims(index_name=index_name, db=db) claim_generator = get_all_claims(index_name=index_name, db=db)
try: try:
await asyncio.gather(*(async_bulk(es, claim_generator, request_timeout=600) for _ in range(clients))) await async_bulk(es, claim_generator, request_timeout=600)
await es.indices.refresh(index=index_name) await es.indices.refresh(index=index_name)
finally: finally:
await es.close() await es.close()