Merge pull request #2145 from denizzzka/mongo_pool_size

MongoDB: adds max pool size setting
vibe-d · Apr 26, 2018 · b06f50e · b06f50e
2 parents 94dbfb3 + fb098d5
commit b06f50e
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 8 deletions.
diff --git a/mongodb/vibe/db/mongo/client.d b/mongodb/vibe/db/mongo/client.d
@@ -60,10 +60,12 @@ final class MongoClient {
 		if(!goodUrl) throw new Exception("Unable to parse mongodb URL: " ~ url);
 
 		m_connections = new ConnectionPool!MongoConnection(() @safe {
-			auto ret = new MongoConnection(settings);
-			ret.connect();
-			return ret;
-		});
+				auto ret = new MongoConnection(settings);
+				ret.connect();
+				return ret;
+			},
+			settings.maxConnections
+		);
 
 		// force a connection to cause an exception for wrong URLs
 		lockConnection();
@@ -72,10 +74,12 @@ final class MongoClient {
 	package this(MongoClientSettings settings)
 	{
 		m_connections = new ConnectionPool!MongoConnection({
-			auto ret = new MongoConnection(settings);
-			ret.connect();
-			return ret;
-		});
+				auto ret = new MongoConnection(settings);
+				ret.connect();
+				return ret;
+			},
+			settings.maxConnections
+		);
 
 		// force a connection to cause an exception for wrong URLs
 		lockConnection();

diff --git a/mongodb/vibe/db/mongo/settings.d b/mongodb/vibe/db/mongo/settings.d
@@ -321,6 +321,7 @@ class MongoClientSettings
 
 	string username;
 	string digest;
+	uint maxConnections = uint.max;
 	MongoHost[] hosts;
 	string database;
 	QueryFlags defQueryFlags = QueryFlags.None;