Retry()函数的高级用法：使用自定义重试策略和回调函数

Retry()函数是一个用于实现重试逻辑的装饰器函数，用来在遇到异常情况时重复执行某个函数或者方法。它可以通过指定重试次数、重试间隔时间、自定义的重试策略和回调函数来满足不同的需求。下面将详细介绍Retry()函数的高级用法，并给出相应的使用例子。

1. 自定义重试策略

Retry()函数通过retry参数来指定重试策略，可以是一个自定义的函数。该函数接受一个异常对象作为参数并返回一个布尔值，用于判断是否要进行重试。如果返回True，重试将继续进行；如果返回False，重试将停止。

下面是一个例子，展示如何使用自定义重试策略：

from tenacity import RetryError, retry, stop_after_attempt

@retry(retry=stop_after_attempt(3))
def do_something():
    try:
        # 这里是你需要执行的函数或方法
        ...
    except Exception as e:
        # 自定义重试策略
        if isinstance(e, CustomException):
            return True
        else:
            return False

在上面的例子中，do_something()函数被装饰为重试函数，并使用自定义的重试策略。在捕获异常时，如果异常是CustomException类型，则返回True，表示要进行重试；否则返回False，表示不需要重试。

2. 回调函数

Retry()函数还可以通过在回调函数中指定一些操作，比如记录日志、发送通知等。回调函数可以在每次重试之前、之后或出现错误时被调用。

下面是一个例子，展示如何使用回调函数：

from tenacity import retry, stop_after_attempt, before_log, after_log, retry_error_callback

def log_before_attempt(details):
    print(f"Performing attempt number {details['attempt_number']}")

def log_after_attempt(details):
    print(f"Attempt number {details['attempt_number']} performed successfully")

def log_on_error(retry_state):
    print(f"An error occurred during attempt number {retry_state.attempt_number}: {retry_state.outcome.exception()}")

@retry(retry=stop_after_attempt(3), before=before_log(log_before_attempt), after=after_log(log_after_attempt), retry_error_callback=retry_error_callback(log_on_error))
def do_something():
    try:
        # 这里是你需要执行的函数或方法
        ...
    except Exception as e:
        # 自定义重试策略
        return False

在上面的例子中，do_something()函数被装饰为重试函数，并使用了before_log()和after_log()回调函数来记录每次重试的信息，并且使用retry_error_callback()回调函数在出现错误时打印错误信息。

以上是Retry()函数的高级用法，通过指定自定义重试策略和回调函数，可以实现更灵活、定制化的重试逻辑。使用Retry()函数可以有效地处理并恢复异常情况，提高代码的可靠性和稳定性。