Python Requests 库中的重试策略实践

网络请求由于各种原因可能会失败，例如网络波动、服务暂时不可达等。为了增强客户端的健壮性，实现请求的自动重试是一个常见的做法。在Python中，requests库是处理HTTP请求的标准工具之一。然而，requests本身并不直接提供重试机制，这需要借助urllib3库中的Retry类来实现。

本文将介绍如何在requests中实现请求的自动重试。

1. 重试的必要性

在分布式系统中，服务间的通信可能会由于各种原因失败。而自动重试机制能够提高系统的可靠性和容错能力。合理的重试策略可以减少暂时性故障导致的请求失败。

2. 实现重试的基本原理

在requests中实现重试通常需要以下步骤：

1. 导入必要的模块。
2. 创建一个HTTPAdapter实例。
3. 在HTTPAdapter上配置Retry策略。
4. 将配置好的HTTPAdapter挂载到Session对象上。
5. 使用配置了重试的Session对象发送请求。

3. 使用urllib3实现重试

以下是一个具体的示例，展示如何为requests请求添加重试逻辑。

3.1. 导入模块

首先，需要导入requests库和urllib3的Retry类。

import requests
from requests.adapters import HTTPAdapter
from urllib3.util import Retry

3.2. 配置重试策略

使用Retry类来定义重试策略。这里可以指定重试次数、状态码集合、异常类型等。

retries = Retry(
    total=5,  # 总重试次数
    backoff_factor=1,  # 指数退避因子
    status_forcelist=[500, 502, 503, 504],  # 指定哪些状态码触发重试
    allowed_methods=frozenset(['GET', 'POST']),  # 允许重试的HTTP方法
)

3.3. 创建HTTPAdapter并配置重试

创建一个HTTPAdapter实例，并设置重试策略。

adapter = HTTPAdapter(max_retries=retries)

3.4. 挂载Adapter到Session

创建Session对象，并为HTTP和HTTPS请求挂载上面创建的adapter。

pythonCopy code
session = requests.Session()
session.mount('http://', adapter)
session.mount('https://', adapter)

3.5. 发送请求

使用配置了重试策略的session对象发送请求。

url = "http://httpbin.org/status/500"
response = session.get(url)

4. 示例：请求一个可能返回错误的服务

以下是一个完整的示例，包括错误处理。

import requests
from requests.adapters import HTTPAdapter
from urllib3.util.retry import Retry

def request_with_retry(url, max_retries=5, backoff_factor=1, status_forcelist=None):
    if status_forcelist is None:
        status_forcelist = [500, 502, 503, 504]
        
    session = requests.Session()
    retries = Retry(total=max_retries,
                    backoff_factor=backoff_factor,
                    status_forcelist=status_forcelist,
                    method_whitelist=["GET", "POST"])
    adapter = HTTPAdapter(max_retries=retries)
    session.mount('http://', adapter)
    session.mount('https://', adapter)
    
    try:
        response = session.get(url)
        response.raise_for_status()  # 如果请求返回的是4XX, 5XX响应码，将引发HTTPError异常
        return response
    except requests.exceptions.HTTPError as e:
        print(f"HTTP Error: {e}")
    except requests.exceptions.ConnectionError as e:
        print(f"Connection Error: {e}")
    except requests.exceptions.Timeout as e:
        print(f"Timeout Error: {e}")
    except requests.exceptions.RequestException as e:
        print(f"Request Exception: {e}")

url = "http://httpbin.org/status/500"
response = request_with_retry(url)

if response:
    print(response.text)

在这个示例中，如果服务响应500系列错误，或者连接异常，request_with_retry函数将尝试最多五次的请求。

结论

在Python中使用requests库配合urllib3的Retry类，可以灵活地实现HTTP请求的自动重试机制。这样可以显著提高应用程序处理网络波动的能力。在微服务、API调用等场景下，这种模式尤为重要。需要注意的是，应当谨慎选择重试的次数和策略，以防止过多的重试导致服务负载过重。

---------------------------END---------------------------

题外话

感谢你能看到最后，给大家准备了一些福利！

感兴趣的小伙伴，赠送全套Python学习资料，包含面试题、简历资料等具体看下方。


👉CSDN大礼包🎁：全网最全《Python学习资料》免费赠送🆓！（安全链接，放心点击）

一、Python所有方向的学习路线

Python所有方向的技术点做的整理，形成各个领域的知识点汇总，它的用处就在于，你可以按照下面的知识点去找对应的学习资源，保证自己学得较为全面。

二、Python兼职渠道推荐*

学的同时助你创收，每天花1-2小时兼职，轻松稿定生活费.

三、最新Python学习笔记

当我学到一定基础，有自己的理解能力的时候，会去阅读一些前辈整理的书籍或者手写的笔记资料，这些笔记详细记载了他们对一些技术点的理解，这些理解是比较独到，可以学到不一样的思路。

四、实战案例

纸上得来终觉浅，要学会跟着视频一起敲，要动手实操，才能将自己的所学运用到实际当中去，这时候可以搞点实战案例来学习。

👉CSDN大礼包🎁：全网最全《Python学习资料》免费赠送🆓！（安全链接，放心点击）

若有侵权，请联系删除