diff --git a/marconi/storage/mongodb/base.py b/marconi/storage/mongodb/base.py index 21161aa29..19dcc3572 100644 --- a/marconi/storage/mongodb/base.py +++ b/marconi/storage/mongodb/base.py @@ -44,7 +44,11 @@ class Driver(storage.DriverBase): mongodb's database. """ if not self._database: - conn = pymongo.MongoClient(cfg.uri) + if cfg.uri and 'replicaSet' in cfg.uri: + conn = pymongo.MongoReplicaSetClient(cfg.uri) + else: + conn = pymongo.MongoClient(cfg.uri) + self._database = conn[cfg.database] return self._database