Python的异常重试方法

迈不过友情╰ 2023-02-20 02:29 2阅读 0赞

# 背景 #

项目msb服务不稳定，通过Python建立websocket总是会有问题，很不稳定，但是一般来说重新建立连接就能成功，多尝试几次就好了。

# 问题处理 #

既然有了相应的需求，就要考虑如何去解决这个websocket建立异常重试的问题  
原来的代码只建立了一次websocket连接：

ws.connect(url, header=header)

方法一：  
使用循环重试的方式：

import  time
       for i in range(4):
           try:
               ws.connect(url, header=header)
               break
           except Exception as e:
               atlog.info("let's retry")
               time.sleep(2)

该方式需要先导入time模块，此处是考虑到重试中间加点间隔时间会好点，密集重试效果不好。  
重试后建立链接成功就退出循环。

方法二：  
使用retrying包的方式建立链接：

from retrying import retry
    @classmethod
    	#stop_max_attempt_number最大重试次数, 两次， stop_max_delay是两次重试间隔多少毫秒，这里我写的间隔1秒
        @retry(stop_max_attempt_number=3,stop_max_delay=1000)
        def do_connect(cls, ws, url, header):
            atlog.info("let's retry..")
            ws.connect(url, hearder=header)

retrying是一个Python的重试包，可以用来自动重试一些可能运行失败的程序段。retryiing提供一个装饰器函数retry,被装饰的函数就会在运行失败的条件下重新执行，默认只要一直报错就会不断重试。  
retry还可以接受一些参数，这个从源码中Retrying类的初始化函数可以看到可选的参数：

*  stop\_max\_attempt\_number：用来设定最大的尝试次数，超过该次数就停止重试
 *  stop\_max\_delay：比如设置成10000，那么从被装饰的函数开始执行的时间点开始，到函数成功运行结束或者失败报错中止的时间点，只要这段时间超过10秒，函数就不会再执行了
 *  wait\_fixed：设置在两次retrying之间的停留时间
 *  wait\_random\_min和wait\_random\_max：用随机的方式产生两次retrying之间的停留时间
 *  wait\_exponential\_multiplier和wait\_exponential\_max：以指数的形式产生两次retrying之间的停留时间，产生的值为2^previous\_attempt\_number \* wait\_exponential\_multiplier，previous\_attempt\_number是前面已经retry的次数，如果产生的这个值超过了wait\_exponential\_max的大小，那么之后两个retrying之间的停留值都为wait\_exponential\_max。这个设计迎合了exponential backoff算法，可以减轻阻塞的情况。

另外，我们可以指定要在出现哪些异常的时候再去retry,这个要用retry\_on\_exception传入一个函数对象：

def retry_if_io_error(exception):
     return isinstance(exception, IOError)
     
    @retry(retry_on_exception=retry_if_io_error)
    def read_a_file():
     with open("file", "r") as f:
      return f.read()

还可以指定要在得到哪些结果的时候去retry,这个要用retry\_on\_result传入一个函数对象：

def retry_if_result_none(result):
     return result is None
     
    @retry(retry_on_result=retry_if_result_none)
    def get_result():
     return None