MongoDB: Use database from connection string when set (#347)

Will still default to "MiniProfiler" when not set but otherwise will use the database as defined from the connection string.
MiniProfiler · Dec 10, 2018 · 1801371 · 1801371
1 parent 42393f5
commit 1801371
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/src/MiniProfiler.Providers.MongoDB/MongoDbStorage.cs b/src/MiniProfiler.Providers.MongoDB/MongoDbStorage.cs
@@ -26,9 +26,12 @@ public MongoDbStorage(string connectionString)
                 BsonClassMapFields();
             }
 
-            _client = new MongoClient(connectionString);
+            var url = new MongoUrl(connectionString);
+            var databaseName = url.DatabaseName ?? "MiniProfiler";
+
+            _client = new MongoClient(url);
             _collection = _client
-                .GetDatabase("MiniProfiler")
+                .GetDatabase(databaseName)
                 .GetCollection<MiniProfiler>("profilers");
         }
 

diff --git a/tests/MiniProfiler.Tests/Storage/MongoDbStorageTests.cs b/tests/MiniProfiler.Tests/Storage/MongoDbStorageTests.cs
@@ -46,7 +46,8 @@ public static class MongoDbStorageExtensions
         /// <param name="storage">The storage to drop schema for.</param>
         public static void DropDatabase(this MongoDbStorage storage)
         {
-            storage.GetClient().DropDatabase("MiniProfiler");
+            var url = new MongoDB.Driver.MongoUrl(TestConfig.Current.MongoDbConnectionString);
+            storage.GetClient().DropDatabase(url.DatabaseName);
         }
     }
 }