python装饰器实现重试_python重试次数装饰器

重试次数装饰器

前言, 最近在使用tornado框架写Restful API时遇到很多的问题. 有框架的问题, 有异步的问题. 虽然tornado 被公认为当前python语言最好的 异步框架, 但是实际资料并没有像 Django , flask框架那么丰富, 所以遇到很多的坑, 时很多! 这一篇主要是为了介绍一下其中在踩坑中所不得不使用的一些技术 .

问题: 个别API逻辑中存在很多的调用系统层面,或者调用第三方接口阻塞超时的操作, 为了确保 API在某一步是否最终调用成功(先不考虑API接口性能问题) 之前的做法是 遇到所有超时操作,都进行 for循环, 或者while循环, 直到 某一步 最终执行才会进行后续的逻辑. 但是 在测试过程中,发现

使用for循环并不知道到底需要执行几次

使用while循环很容易造成 死循环, 最终导致程序的内存泄露, 程序崩溃.

针对不同的重试, 此时想到了 用装饰器, 直接上代码 如下

#!/usr/bin/env python

# ~*~ coding: utf-8 ~*~

def retry_func(retry_times=3, sleep_time=8):

"""函数重复执行次数"""

def retry_decorator(func):

@wraps(func)

def wrapper_func(*args, **kwargs):

flag = 0

while flag < retry_times:

res = func(*args, **kwargs)

if not res:

flag += 1

logger.warning("{0} execute {1} times".format((func.__name__), flag))

time.sleep(sleep_time)

continue

else:

return res

return wrapper_func

return retry_decorator

装饰器的概念及基本知识这里不再做详解: 参考 装饰器入门到进阶

找到 循环的条件及终止条件