如何檢查PyMongo的連接是否還活着？

Question

如何檢查PyMongo是否連接仍然活着？有沒有優雅的方式來做到這一點，或只是嘗試一個簡單的查詢和解析器結果？

Answer 1

我不認爲有一種優雅的方式。 Pymongo具有AutoReconnect異常，如果先前建立的連接丟失，將拋出異常。您應該捕獲此異常並重試將導致連接重新建立的操作。

處理我發現AutoReconnects是一個相對簡單的辦法就是DB操作在不同的方法分離，並用類似下面的裝飾包裹其中：

from pymongo.errors import AutoReconnect 

def autoreconnect_retry(fn, retries=3): 
    def db_op_wrapper(*args, **kwargs): 
     tries = 0 

     while tries < retries: 
      try: 
       return fn(*args, **kwargs) 

      except AutoReconnect: 
       tries += 1 

     raise Exception("No luck even after %d retries" % retries) 

用法：

@autoreconnect_retry 
def insert_foo_record(foo): 
    # Perform db operation 
    pass 

@autoreconnect_retry(20) 
def get_foo_record(id): 
    # Perform db operation 
    pass 

Answer 2

在pymongo2.2連接url中可以選擇添加附加參數：

mongodb://<user>:<password>@<host>:<port>/<db>?options 

支持的選項之一是connectTimeoutMS即

mongodb://<user>:<password>@<host>:<port>/<db>?connectTimeoutMS=300 

，以連接超時擴展到pymongo爲300ms

更高版本具有相同的選項

Answer 3

的相似的變化有了給你會有像錯誤的解決方案'NoneType'對象不可調用，或* autoreconnect_retry（）至少需要1個參數（1給定）*。在這裏，你已經測試和運行的代碼來裝點基於kristaps其執行pymongo操作所有的方法，回答：

from pymongo.errors import AutoReconnect 

def autoreconnect_retry(retries=3): 
""" 
Decorator for connection retrial in mongoDB 
""" 
def autoreconnect_retry_decorator(fn): 
    def db_op_wrapper(*args, **kwargs): 
     tries = 0 

     while tries < retries: 
      try: 
       return fn(*args, **kwargs) 

      except AutoReconnect: 
       tries += 1 

     raise Exception("No luck even after %d retries" % retries) 

    return db_op_wrapper 
return autoreconnect_retry_decorator 

然後你可以裝飾用參數的方法如前所述：

@autoreconnect_retry 
def some_pymongo_operation(): 
    pass 

@autoreconnect_retry(20) 
def some_pymongo_operation(): 
    pass