0

0

Python函数怎样用装饰器实现函数重试机制 Python函数重试装饰器的入门应用技巧​

蓮花仙者

蓮花仙者

发布时间:2025-08-13 16:51:01

|

767人浏览过

|

来源于php中文网

原创

可以通过在重试装饰器中引入异常类型过滤、日志记录和指数退避机制来实现更智能的重试策略。1. 通过设置exceptions参数指定仅对特定异常(如networkerror)进行重试,而对其他异常直接抛出;2. 使用logging模块在重试前记录警告日志,失败后记录错误日志,便于问题排查;3. 采用指数退避算法(如base_delay (2 * retries) + 随机抖动)动态增加重试间隔,避免请求堆积和服务器压力,从而有效防止无限重试并提升系统稳定性。最终方案结合了异常类型判断、日志输出和延迟递增策略,完整实现了高效、可控、可维护的函数重试机制。

Python函数怎样用装饰器实现函数重试机制 Python函数重试装饰器的入门应用技巧​

Python函数重试装饰器通过在函数执行失败时自动重新调用该函数,提高程序的稳定性和容错能力。

解决方案:

函数重试装饰器的核心思想是:当被装饰的函数抛出异常时,装饰器会捕获这个异常,并根据设定的重试次数和延迟时间,自动重新执行该函数。如果超过最大重试次数,仍然失败,则抛出异常。

立即学习Python免费学习笔记(深入)”;

一个简单的Python函数重试装饰器实现如下:

import time
import functools

def retry(max_retries=3, delay=1):
    """
    重试装饰器。

    Args:
        max_retries: 最大重试次数,默认为3。
        delay: 重试间隔时间,默认为1秒。
    """
    def decorator_retry(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            retries = 0
            while retries < max_retries:
                try:
                    return func(*args, **kwargs)
                except Exception as e:
                    print(f"函数 {func.__name__} 失败,重试中... (第 {retries + 1} 次)")
                    retries += 1
                    time.sleep(delay)
            print(f"函数 {func.__name__} 达到最大重试次数,仍然失败。")
            raise  # 重新抛出异常
        return wrapper
    return decorator_retry

# 示例
@retry(max_retries=5, delay=2)
def unreliable_function():
    """
    一个可能失败的函数。
    """
    import random
    if random.randint(0, 9) < 7:  # 70%的概率失败
        raise Exception("模拟函数执行失败")
    print("函数执行成功!")
    return "成功"

if __name__ == '__main__':
    try:
        result = unreliable_function()
        print(f"函数返回结果: {result}")
    except Exception as e:
        print(f"最终失败: {e}")

这个例子中,

retry
是一个装饰器工厂函数,它接受
max_retries
delay
作为参数,用于配置重试次数和间隔时间。
decorator_retry
是真正的装饰器,它接受一个函数
func
作为参数,并返回一个新的函数
wrapper
wrapper
函数负责执行
func
,并在发生异常时进行重试。
functools.wraps(func)
用于保留原始函数的元信息,例如函数名和文档字符串。

如何根据不同的异常类型进行不同的重试策略?

有时候,我们希望针对不同的异常类型采取不同的重试策略。例如,对于网络连接错误,我们可能希望立即重试,而对于参数错误,我们则不希望重试。

Type Studio
Type Studio

一个视频编辑器,提供自动转录、自动生成字幕、视频翻译等功能

下载

可以修改装饰器,使其接受一个

exceptions
参数,用于指定需要重试的异常类型:

import time
import functools

def retry(exceptions=(Exception,), max_retries=3, delay=1):
    """
    重试装饰器,可以指定需要重试的异常类型。

    Args:
        exceptions: 需要重试的异常类型,默认为Exception。
        max_retries: 最大重试次数,默认为3。
        delay: 重试间隔时间,默认为1秒。
    """
    def decorator_retry(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            retries = 0
            while retries < max_retries:
                try:
                    return func(*args, **kwargs)
                except exceptions as e: # 只捕获指定的异常类型
                    print(f"函数 {func.__name__} 遇到 {type(e).__name__} 异常,重试中... (第 {retries + 1} 次)")
                    retries += 1
                    time.sleep(delay)
                except Exception as e:
                    # 捕获其他异常,直接抛出
                    print(f"函数 {func.__name__} 遇到 {type(e).__name__} 异常,不进行重试。")
                    raise
            print(f"函数 {func.__name__} 达到最大重试次数,仍然失败。")
            raise  # 重新抛出异常
        return wrapper
    return decorator_retry


# 示例
class NetworkError(Exception):
    pass

@retry(exceptions=(NetworkError,), max_retries=5, delay=2)
def unreliable_network_function():
    """
    一个可能因为网络问题失败的函数。
    """
    import random
    if random.randint(0, 9) < 7:
        raise NetworkError("模拟网络连接失败")
    print("网络函数执行成功!")
    return "成功"

if __name__ == '__main__':
    try:
        result = unreliable_network_function()
        print(f"函数返回结果: {result}")
    except Exception as e:
        print(f"最终失败: {e}")

这个例子中,

retry
装饰器现在接受一个
exceptions
参数,用于指定需要重试的异常类型。只有当函数抛出的异常类型在
exceptions
中时,才会进行重试。

如何在重试过程中记录日志?

记录日志可以帮助我们更好地了解函数执行的情况,并在出现问题时进行排查。可以在装饰器中加入日志记录功能:

import time
import functools
import logging

logging.basicConfig(level=logging.INFO)  # 配置日志

def retry(exceptions=(Exception,), max_retries=3, delay=1):
    """
    重试装饰器,可以指定需要重试的异常类型,并记录日志。

    Args:
        exceptions: 需要重试的异常类型,默认为Exception。
        max_retries: 最大重试次数,默认为3。
        delay: 重试间隔时间,默认为1秒。
    """
    def decorator_retry(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            retries = 0
            while retries < max_retries:
                try:
                    return func(*args, **kwargs)
                except exceptions as e:
                    logging.warning(f"函数 {func.__name__} 遇到 {type(e).__name__} 异常,重试中... (第 {retries + 1} 次)")
                    retries += 1
                    time.sleep(delay)
                except Exception as e:
                    logging.error(f"函数 {func.__name__} 遇到 {type(e).__name__} 异常,不进行重试。")
                    raise
            logging.error(f"函数 {func.__name__} 达到最大重试次数,仍然失败。")
            raise  # 重新抛出异常
        return wrapper
    return decorator_retry

# 示例
@retry(max_retries=3, delay=1)
def another_unreliable_function():
    import random
    if random.randint(0, 9) < 5:
        raise ValueError("模拟值错误")
    print("函数执行成功!")
    return "成功"

if __name__ == '__main__':
    try:
        result = another_unreliable_function()
        print(f"函数返回结果: {result}")
    except Exception as e:
        print(f"最终失败: {e}")

这个例子中,我们使用了

logging
模块来记录日志。在每次重试之前,我们使用
logging.warning
记录一条警告信息。如果达到最大重试次数仍然失败,我们使用
logging.error
记录一条错误信息。

如何避免无限重试?

如果函数总是抛出异常,那么重试装饰器可能会陷入无限循环。为了避免这种情况,我们需要设置一个最大重试次数。在上面的例子中,我们已经使用了

max_retries
参数来限制重试次数。但是,如果
max_retries
设置得过大,仍然可能导致程序长时间阻塞。

一种更健壮的方法是使用指数退避算法来调整重试间隔时间。指数退避算法会随着重试次数的增加,逐渐增加重试间隔时间。这样可以避免在高并发情况下,大量重试请求同时到达服务器,导致服务器崩溃。

一个使用指数退避算法的重试装饰器实现如下:

import time
import functools
import random

def retry_with_backoff(exceptions=(Exception,), max_retries=5, base_delay=1):
    """
    使用指数退避算法的重试装饰器。

    Args:
        exceptions: 需要重试的异常类型,默认为Exception。
        max_retries: 最大重试次数,默认为5。
        base_delay: 基础延迟时间,默认为1秒。
    """
    def decorator_retry(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            retries = 0
            while retries < max_retries:
                try:
                    return func(*args, **kwargs)
                except exceptions as e:
                    delay = base_delay * (2 ** retries) + random.uniform(0, 1)  # 指数退避 + 随机抖动
                    print(f"函数 {func.__name__} 遇到 {type(e).__name__} 异常,重试中... (第 {retries + 1} 次),延迟 {delay:.2f} 秒")
                    retries += 1
                    time.sleep(delay)
                except Exception as e:
                    print(f"函数 {func.__name__} 遇到 {type(e).__name__} 异常,不进行重试。")
                    raise
            print(f"函数 {func.__name__} 达到最大重试次数,仍然失败。")
            raise  # 重新抛出异常
        return wrapper
    return decorator_retry

# 示例
@retry_with_backoff(max_retries=4, base_delay=0.5)
def yet_another_unreliable_function():
    import random
    if random.randint(0, 9) < 6:
        raise IOError("模拟IO错误")
    print("函数执行成功!")
    return "成功"

if __name__ == '__main__':
    try:
        result = yet_another_unreliable_function()
        print(f"函数返回结果: {result}")
    except Exception as e:
        print(f"最终失败: {e}")

这个例子中,我们使用

base_delay * (2 ** retries) + random.uniform(0, 1)
来计算重试间隔时间。其中,
base_delay
是基础延迟时间,
2 ** retries
是指数退避因子,
random.uniform(0, 1)
是一个随机抖动,用于避免多个重试请求同时到达服务器。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
python开发工具
python开发工具

php中文网为大家提供各种python开发工具,好的开发工具,可帮助开发者攻克编程学习中的基础障碍,理解每一行源代码在程序执行时在计算机中的过程。php中文网还为大家带来python相关课程以及相关文章等内容,供大家免费下载使用。

778

2023.06.15

python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

686

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

769

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

740

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

1445

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

571

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

581

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

752

2023.08.11

拼多多赚钱的5种方法 拼多多赚钱的5种方法
拼多多赚钱的5种方法 拼多多赚钱的5种方法

在拼多多上赚钱主要可以通过无货源模式一件代发、精细化运营特色店铺、参与官方高流量活动、利用拼团机制社交裂变,以及成为多多进宝推广员这5种方法实现。核心策略在于通过低成本、高效率的供应链管理与营销,利用平台社交电商红利实现盈利。

31

2026.01.26

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Django 教程
Django 教程

共28课时 | 3.5万人学习

Excel 教程
Excel 教程

共162课时 | 13.6万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.3万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号