• 重试次数装饰器

重试次数装饰器

前言, 最近在使用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

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

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