Integration of MongoDB storage driver.

livebridge/components.py

@@ -15,7 +15,7 @@
 # limitations under the License.
 import logging
 from livebridge import config
-from livebridge.storages import DynamoClient, SQLStorage
+from livebridge.storages import DynamoClient, SQLStorage, MongoStorage
 
 
 logger = logging.getLogger(__name__)
@@ -78,9 +78,13 @@ def add_target(cls):
 
 
 def get_db_client(**kwargs):
-    if kwargs.get("dsn") or config.DB.get("dsn"):
+    dsn = kwargs.get("dsn") or config.DB.get("dsn")
+    if dsn:
         params = config.DB if not kwargs else kwargs
-        return SQLStorage(**params)
+        if dsn.startswith("mongodb://"):
+            return MongoStorage(**params)
+        else:
+            return SQLStorage(**params)
     # default dynamodb
     params = config.AWS if not kwargs else kwargs
     return DynamoClient(**params)

livebridge/storages/__init__.py

@@ -16,3 +16,4 @@
 from livebridge import config
 from livebridge.storages.dynamo import DynamoClient
 from livebridge.storages.sql import SQLStorage
+from livebridge.storages.mongo import MongoStorage

requirements.txt

@@ -2,6 +2,8 @@ aiobotocore==0.3.3
 aiohttp==2.1.0
 asynctest==0.10.0
 bleach==2.0.0
+dsnparse==0.1.4
+motor==1.1
 pytest==3.1.2
 pytest-cov==2.5.1
 python-dateutil==2.6.0