PY: check if connection successful

differently with updated verion of pymongo
UnivaCorporation · Dec 14, 2017 · 4c86c96 · 4c86c96
1 parent 9ab7a95
commit 4c86c96
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/source/python/urb/db/urb_mongo_client.py b/source/python/urb/db/urb_mongo_client.py
@@ -34,7 +34,10 @@ class URBMongoClient(object):
     def __init__(self, db_uri='mongodb://localhost:27017/', db_name=DB_NAME, expire=TTL_MONTHS):
         self.logger = LogManager.get_instance().get_logger(self.__class__.__name__)
         try:
+            # for pymongo >= 3.0 constructor doesn't throw anymore
             self.client = MongoClient(db_uri)
+            # check connection with ismaster command
+            self.client.admin.command('ismaster')
             self.logger.info("Connected to Mongo DB: %s" % db_uri)
             self.db = self.client[db_name]
             self.expire = expire