elasticsearch异步查询

from elasticsearch import AsyncElasticsearch
class AIOElasticsearchUtils:
    def __init__(self):
        self.es = AsyncElasticsearch(["http://host:port@user:password"])
        self.timeout ='2s'
    async def search(self, dsl, index_name):
        for i in range(3):
            try:
                response = await self.es.search(index=index_name,body=dsl,timeout='10s')
                if str(response.get("timed_out", "")).lower() == "true":
                    raise Exception("Es Timeout.")
                return response
            except Exception as e:
                if str(e).find("Timeout") > 0:
                    continue
                raise e
        raise Exception("ES search timeout.")