fix es sync.py

This commit is contained in:
Jack Robison 2021-09-13 19:10:52 -04:00
parent a567326853
commit 1ee1a5f2a1
No known key found for this signature in database
GPG key ID: DF25C68FE0239BB2

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()