Python的异常重试方法

背景

项目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