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