Add safe collection getter & merge method & insertMany

Author: WebFreak001

source/mongoschema/package.d

@@ -717,10 +717,16 @@ private:
 mixin template MongoSchema()
 {
 	import std.typecons : Nullable;
+	import std.range : isInputRange, ElementType;
 
 	static MongoCollection _schema_collection_;
 	private BsonObjectID _schema_object_id_;
 
+	@property static MongoCollection collection() @safe
+	{
+		return _schema_collection_;
+	}
+
 	/// Returns: the _id value (if set by save or find)
 	@property ref BsonObjectID bsonID() @safe
 	{
@@ -732,14 +738,30 @@ mixin template MongoSchema()
 	{
 		if (_schema_object_id_.valid)
 		{
-			_schema_collection_.update(Bson(["_id" : Bson(_schema_object_id_)]),
+			collection.update(Bson(["_id" : Bson(_schema_object_id_)]),
 					this.toSchemaBson(), UpdateFlags.upsert);
 		}
 		else
 		{
 			_schema_object_id_ = BsonObjectID.generate;
 			auto bson = this.toSchemaBson();
-			_schema_collection_.insert(bson);
+			collection.insert(bson);
+		}
+	}
+
+	/// Inserts or merges into an existing value.
+	void merge()
+	{
+		if (_schema_object_id_.valid)
+		{
+			collection.update(Bson(["_id" : Bson(_schema_object_id_)]),
+					Bson(["$set": this.toSchemaBson()]), UpdateFlags.upsert);
+		}
+		else
+		{
+			_schema_object_id_ = BsonObjectID.generate;
+			auto bson = this.toSchemaBson();
+			collection.insert(bson);
 		}
 	}
 
@@ -748,7 +770,7 @@ mixin template MongoSchema()
 	{
 		if (!_schema_object_id_.valid)
 			return false;
-		_schema_collection_.remove(Bson(["_id" : Bson(_schema_object_id_)]),
+		collection.remove(Bson(["_id" : Bson(_schema_object_id_)]),
 				DeleteFlags.SingleRemove);
 		return true;
 	}
@@ -757,7 +779,7 @@ mixin template MongoSchema()
 	/// Throws: DocumentNotFoundException if not found
 	static auto findOneOrThrow(T)(T query)
 	{
-		Bson found = _schema_collection_.findOne(query);
+		Bson found = collection.findOne(query);
 		if (found.isNull)
 			throw new DocumentNotFoundException("Could not find one " ~ typeof(this).stringof);
 		return found;
@@ -787,7 +809,7 @@ mixin template MongoSchema()
 	/// Tries to find a document by the _id field and returns a Nullable which `isNull` if it could not be found. Otherwise it will be the document wrapped in the nullable.
 	static Nullable!(typeof(this)) tryFindById(BsonObjectID id)
 	{
-		Bson found = _schema_collection_.findOne(Bson(["_id" : Bson(id)]));
+		Bson found = collection.findOne(Bson(["_id" : Bson(id)]));
 		if (found.isNull)
 			return Nullable!(typeof(this)).init;
 		return Nullable!(typeof(this))(fromSchemaBson!(typeof(this))(found));
@@ -802,7 +824,7 @@ mixin template MongoSchema()
 	/// Tries to find a document in this collection. It will return a Nullable which `isNull` if the document could not be found. Otherwise it will be the document wrapped in the nullable.
 	static Nullable!(typeof(this)) tryFindOne(T)(T query)
 	{
-		Bson found = _schema_collection_.findOne(query);
+		Bson found = collection.findOne(query);
 		if (found.isNull)
 			return Nullable!(typeof(this)).init;
 		return Nullable!(typeof(this))(fromSchemaBson!(typeof(this))(found));
@@ -813,7 +835,7 @@ mixin template MongoSchema()
 			int num_skip = 0, int num_docs_per_chunk = 0)
 	{
 		typeof(this)[] values;
-		foreach (entry; _schema_collection_.find(query, null, flags, num_skip, num_docs_per_chunk))
+		foreach (entry; collection.find(query, null, flags, num_skip, num_docs_per_chunk))
 		{
 			values ~= fromSchemaBson!(typeof(this))(entry);
 		}
@@ -824,7 +846,7 @@ mixin template MongoSchema()
 	deprecated("use findAll instead") static typeof(this)[] find()
 	{
 		typeof(this)[] values;
-		foreach (entry; _schema_collection_.find())
+		foreach (entry; collection.find())
 		{
 			values ~= fromSchemaBson!(typeof(this))(entry);
 		}
@@ -835,58 +857,73 @@ mixin template MongoSchema()
 	static DocumentRange!(typeof(this)) findRange(T)(T query,
 			QueryFlags flags = QueryFlags.None, int num_skip = 0, int num_docs_per_chunk = 0)
 	{
-		return DocumentRange!(typeof(this))(_schema_collection_.find(serializeToBson(query),
+		return DocumentRange!(typeof(this))(collection.find(serializeToBson(query),
 				null, flags, num_skip, num_docs_per_chunk));
 	}
 
 	/// Queries all elements from the collection as range.
 	static DocumentRange!(typeof(this)) findAll()
 	{
-		return DocumentRange!(typeof(this))(_schema_collection_.find());
+		return DocumentRange!(typeof(this))(collection.find());
+	}
+
+	/// Inserts many documents at once. The resulting IDs of the symbols will be generated by the server and not known to the caller.
+	static void insertMany(T)(T documents, InsertFlags options = InsertFlags.none)
+		if (isInputRange!T && is(ElementType!T : typeof(this)))
+	{
+		import std.array : array;
+		import std.algorithm : map;
+
+		if (documents.empty)
+			return;
+		collection.insert(documents.map!((a) {
+				a.bsonID = BsonObjectID.init;
+				return a.toSchemaBson;
+			}).array, options); // .array needed because of vibe-d issue #2185
 	}
 
 	/// Updates a document.
 	static void update(T, U)(T query, U update, UpdateFlags options = UpdateFlags.none)
 	{
-		_schema_collection_.update(query, update, options);
+		collection.update(query, update, options);
 	}
 
 	/// Updates a document or inserts it when not existent. Shorthand for `update(..., UpdateFlags.upsert)`
 	static void upsert(T, U)(T query, U update, UpdateFlags options = UpdateFlags.upsert)
 	{
-		_schema_collection_.update(query, update, options);
+		collection.update(query, update, options);
 	}
 
 	/// Deletes one or any amount of documents matching the selector based on the flags.
 	static void remove(T)(T selector, DeleteFlags flags = DeleteFlags.none)
 	{
-		_schema_collection_.remove(selector, flags);
+		collection.remove(selector, flags);
 	}
 
 	/// Removes all documents from this collection.
 	static void removeAll()
 	{
-		_schema_collection_.remove();
+		collection.remove();
 	}
 
 	/// Drops the entire collection and all indices in the database.
 	static void dropTable()
 	{
-		_schema_collection_.drop();
+		collection.drop();
 	}
 
 	/// Returns the count of documents in this collection matching this query.
 	static auto count(T)(T query)
 	{
-		return _schema_collection_.count(query);
+		return collection.count(query);
 	}
 
 	/// Returns the count of documents in this collection.
 	static auto countAll()
 	{
 		import vibe.data.bson : Bson;
 
-		return _schema_collection_.count(Bson.emptyObject);
+		return collection.count(Bson.emptyObject);
 	}
 
 	/// Start of an aggregation call. Returns a pipeline with typesafe functions for modifying the pipeline and running it at the end.
@@ -906,7 +943,7 @@ mixin template MongoSchema()
 	/// --------------------
 	static SchemaPipeline aggregate()
 	{
-		return SchemaPipeline(_schema_collection_);
+		return SchemaPipeline(collection);
 	}
 }
 
@@ -917,8 +954,10 @@ void register(T)(MongoCollection collection) @safe
 
 	static if (hasMember!(T, "_schema_collection_"))
 	{
-		assert(T._schema_collection_.name.length == 0, "Can't register a Schema to 2 collections!");
-		T._schema_collection_ = collection;
+		(() @trusted {
+			assert(T._schema_collection_.name.length == 0, "Can't register a Schema to 2 collections!");
+			T._schema_collection_ = collection;
+		})();
 	}
 
 	foreach (memberName; __traits(allMembers, T))