如何为elasticsearch-py客户端的请求重试添加延迟及退避因子？

我目前通过以下代码实例化Python Elasticsearch客户端：

es = Elasticsearch(
        hosts=ELASTICSEARCH_URL,
        timeout=5,
        ignore_unavailable=True,
        # connection_retries=Retry(
        #     total=5,
        #     backoff_factor=1.1,
        #     status_forcelist=[429, 502, 503, 504],
        #     raise_on_status=False,
        # )
)

我希望在重试之间引入延迟（理想情况下带有退避因子），让执行如下search请求时能按退避因子进行重试：

es.search(index="abc", body={"query": {"match_all": {}}})

请问该如何实现这一需求？

要实现带退避策略的重试，需根据你使用的elasticsearch-py版本，正确配置客户端的重试参数，以下是通用配置方式：

1. 导入重试依赖类 #

首先导入Retry类，不同版本的库导入路径略有差异：

# 低版本elasticsearch-py（7.x及以下）
from elasticsearch import Elasticsearch
from elasticsearch.helpers import Retry

# 高版本elasticsearch-py（8.x+）
from elasticsearch import Elasticsearch
from elasticsearch.transport import Retry

2. 配置带退避的重试规则 #

方式一：简洁参数配置（适用于多数版本） #

直接通过客户端初始化参数配置重试逻辑：

es = Elasticsearch(
    hosts=ELASTICSEARCH_URL,
    timeout=5,
    ignore_unavailable=True,
    max_retries=5,  # 最大重试次数
    retry_backoff_factor=1.1,  # 退避因子，重试间隔按该倍数递增
    retry_on_status=[429, 502, 503, 504],  # 触发重试的HTTP状态码
    retry_on_timeout=True  # 请求超时后也触发重试
)

方式二：显式配置Transport（适用于8.x+高版本） #

通过transport_options传入完整的Retry实例，支持更精细的控制：

es = Elasticsearch(
    hosts=ELASTICSEARCH_URL,
    timeout=5,
    ignore_unavailable=True,
    transport_options={
        "retry": Retry(
            total=5,  # 最大重试次数
            backoff_factor=1.1,  # 退避因子
            status_forcelist=[429, 502, 503, 504],  # 指定重试的状态码
            raise_on_status=False  # 最终失败时不抛出异常，返回响应；设为True则抛出异常
        )
    }
)

3. 重试行为说明 #

配置完成后，当es.search请求遇到指定状态码或超时情况时，客户端会自动执行退避重试：

• 第1次重试间隔：1 * 退避因子 秒
• 第2次重试间隔：2 * 退避因子 秒
• 后续重试间隔按倍数递增，直到达到最大重试次数

注意事项 #

• 不同版本的elasticsearch-py参数命名可能有差异，建议对照你使用版本的官方文档调整参数
• 如果需要在最终重试失败时抛出异常，可将raise_on_status设为True

内容的提问来源于stack exchange，提问作者room13