0

0

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

蓮花仙者

蓮花仙者

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

|

767人浏览过

|

来源于php中文网

原创

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

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)
用于保留原始函数的元信息,例如函数名和文档字符串。

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

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

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

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

BiLin AI
BiLin AI

免费的多语言AI搜索引擎

下载
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

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

豆包大模型
豆包大模型

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

WorkBuddy
WorkBuddy

腾讯云推出的AI原生桌面智能体工作台

腾讯元宝
腾讯元宝

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

文心一言
文心一言

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

讯飞写作
讯飞写作

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

即梦AI
即梦AI

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

ChatGPT
ChatGPT

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

相关专题

更多
scripterror怎么解决
scripterror怎么解决

scripterror的解决办法有检查语法、文件路径、检查网络连接、浏览器兼容性、使用try-catch语句、使用开发者工具进行调试、更新浏览器和JavaScript库或寻求专业帮助等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

492

2023.10.18

500error怎么解决
500error怎么解决

500error的解决办法有检查服务器日志、检查代码、检查服务器配置、更新软件版本、重新启动服务、调试代码和寻求帮助等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

382

2023.10.25

js 字符串转数组
js 字符串转数组

js字符串转数组的方法:1、使用“split()”方法;2、使用“Array.from()”方法;3、使用for循环遍历;4、使用“Array.split()”方法。本专题为大家提供js字符串转数组的相关的文章、下载、课程内容,供大家免费下载体验。

761

2023.08.03

js截取字符串的方法
js截取字符串的方法

js截取字符串的方法有substring()方法、substr()方法、slice()方法、split()方法和slice()方法。本专题为大家提供字符串相关的文章、下载、课程内容,供大家免费下载体验。

221

2023.09.04

java基础知识汇总
java基础知识汇总

java基础知识有Java的历史和特点、Java的开发环境、Java的基本数据类型、变量和常量、运算符和表达式、控制语句、数组和字符串等等知识点。想要知道更多关于java基础知识的朋友,请阅读本专题下面的的有关文章,欢迎大家来php中文网学习。

1570

2023.10.24

字符串介绍
字符串介绍

字符串是一种数据类型,它可以是任何文本,包括字母、数字、符号等。字符串可以由不同的字符组成,例如空格、标点符号、数字等。在编程中,字符串通常用引号括起来,如单引号、双引号或反引号。想了解更多字符串的相关内容,可以阅读本专题下面的文章。

651

2023.11.24

java读取文件转成字符串的方法
java读取文件转成字符串的方法

Java8引入了新的文件I/O API,使用java.nio.file.Files类读取文件内容更加方便。对于较旧版本的Java,可以使用java.io.FileReader和java.io.BufferedReader来读取文件。在这些方法中,你需要将文件路径替换为你的实际文件路径,并且可能需要处理可能的IOException异常。想了解更多java的相关内容,可以阅读本专题下面的文章。

1228

2024.03.22

php中定义字符串的方式
php中定义字符串的方式

php中定义字符串的方式:单引号;双引号;heredoc语法等等。想了解更多字符串的相关内容,可以阅读本专题下面的文章。

1205

2024.04.29

TypeScript类型系统进阶与大型前端项目实践
TypeScript类型系统进阶与大型前端项目实践

本专题围绕 TypeScript 在大型前端项目中的应用展开,深入讲解类型系统设计与工程化开发方法。内容包括泛型与高级类型、类型推断机制、声明文件编写、模块化结构设计以及代码规范管理。通过真实项目案例分析,帮助开发者构建类型安全、结构清晰、易维护的前端工程体系,提高团队协作效率与代码质量。

49

2026.03.13

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.5万人学习

Django 教程
Django 教程

共28课时 | 5万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.9万人学习

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

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