欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

python重试次数装饰器

程序员文章站 2022-08-26 15:55:12
[TOC] 重试次数装饰器 前言, 最近在使用tornado框架写Restful API时遇到很多的问题. 有框架的问题, 有异步的问题. 虽然 被公认为当前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
  • 装饰器的概念及基本知识这里不再做详解: 参考 装饰器入门到进阶

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