在编程的时候,我们难免会遇到一些不可靠的情况,比如网络请求失败,数据库连接超时等等。这些不确定性会让我们的程序容易出现各种错误和异常。那么如何来增加程序的容错性和健壮性呢?
可能大多数人会想到使用 try except 来进行异常捕捉进行失败重试 (Retry)。虽然 try-escept 一个非常常见和有效的方式来增强程序稳定性,但是可能一不小心就会造成栈溢出。
所以接下来我就来介绍一个另外的一个专门用于失败重试的库:retrying。
在 Python 生态中,retrying 库提供了非常便捷的装饰器和函数来帮助我们轻松添加失败重试机制。它可以自定义重试策略、停止条件、等待间隔等,对各种异常进行捕捉处理。使用 retrying 可以大大减少我们重复编写失败重试轮询的代码量。
更多内容可以学习《测试工程师 Python 工具开发实战》书籍、《大话性能测试 JMeter 实战》书籍
pip install retrying
我们可以直接在函数上使用装饰器@retry来进行失败重试
import retrying
@retry
def func():
for item in range(0,100):
result=item / 0
print(result)
return result
func()
但是这种方式并不建议使用,就像上面的代码,我们都知道 0 作为除数就会报错,在上面的 func 函数中,因为加了@retry装饰器进行失败重试,这样就就会进入一个死循环一直失败一直重试。
所以我们在进行失败重试的时候最好是需要加上一些参数来限制失败重试。
(1)stop_max_attempt_number
在 retry 中传入 stop_max_attempt_number 参数后可以指定失败重试的次数
@retry(stop_max_attempt_number=2)
def func():
print(f"记录失败重试")
for item in range(0,100):
result=item / 0
print(result)
return result
func()
因为这里我们指定了失败后进行两次重试,如果重试执行两次后还是报错则结束重试,将错误信息抛出来。
(2)wait_fixed 传入 wati_fixed 后,可以指定重试的时间
from retrying import retry
import time
# 设置三秒重试一次
@retry(wait_fixed=3000)
def func():
print(f"记录失败重试:",time.strftime("%Y-%m-%d %H:%M:%S"))
result=1 / 0
print(result)
return result
func()
配置重试间隔时间后,成语遇到执行失败或者报错后,就会根据设置的重试时间去进行重试执行
(3)wait_random_min 和 wait_random_max
通常 wait_random_min 和 wait_random_max 是一起搭配使用的,可以设置一个重试等待的时间,然后会在设置的时间区间内随机取一个等待时间进行重试
from retrying import retry
import time
@retry(wait_random_min=1000,wait_random_max=9000)
def func():
print(f"记录失败重试:",time.strftime("%Y-%m-%d %H:%M:%S"))
result=1 / 0
print(result)
return result
func()
(4)wait_exponential_multiplier 和 wait_exponential_max
官方解释为:以指数的形式产生两次 retrying 之间的停留时间, 产生的值为 2previous_attempt_number * wait_exponential_multiplier, previous_attempt_number 是前面已经 retry 的次数, 如果产生的这个值超过了 wait_exponential_max 的大小, 那么之后两个 retrying 之间的停留值都为 wait_exponential_max
通俗来点讲就是每次重试的时间以 wait_exponential_multiplier 设置的值 2,如果重试后还是失败则继续 2,直到最后的值等于或则超过 wait_exponential_max 设置的值后,后面的每一次重试等待时间都是 wait_exponential_max 设置的值
from retrying import retry
import time
@retry(wait_exponential_multiplier=1000,wait_exponential_max=10000)
def func():
print(f"记录失败重试:",time.strftime("%Y-%m-%d %H:%M:%S"))
result=1 / 0
print(result)
return result
func()
(5) wait_func
在前面介绍的参数都是如何配置失败冲重试的等待时间或者重试次数之类的,但是我们不能时时刻刻盯着程序,在程序代码发生错误时我们应该要进行发送短信或者邮件之类的提醒才行
在这里就可以使用到 wait_func 参数,它接收一个可执行函数,返回一个具体的间隔时间数值,单位 ms。接收的函数须接收两个参数:attempt_number 当前运行次数,delay_since_first_attempt_ms 当前重试机制运行时间 (单位 ms)
from retrying import retry
import time
def func_demo(attempt_number,delay_since_first_attempt_ms):
print("函数运行失败后运行该函数")
if attempt_number == 5:
print("已经重试失败五次了,开始准备发送提醒")
if attempt_number == 10:
print("已经重试失败超10次了,发送邮件给相关人员紧急处理")
if attempt_number >10:
print("重试时间过长,做一些其他临时方案进行补救")
# return一个重试的时间
return 2000
@retry(wait_func=func_demo)
def func():
print(f"记录失败重试:",time.strftime("%Y-%m-%d %H:%M:%S"))
result=1 / 0
return result
func()
使用 wait_func 通过调用其他可执行的函数,我们可以借助它来做一些临时的补救措施,避免程序一直无法运行而产生的影响。
(6)其他参数
在 retry 中还存在有很多参数,有兴趣的小伙伴可以去详细了解下