Merge branch 'pr/125'

Change-Id: I570e53e4bf714877e4f2d8e831d6f876c98d312d

Author: skriptble

mongo/collection.go

@@ -509,6 +509,7 @@ func (coll *Collection) updateOrReplaceOne(ctx context.Context, filter,
 	res := &UpdateResult{
 		MatchedCount:  r.MatchedCount,
 		ModifiedCount: r.ModifiedCount,
+		UpsertedCount: int64(len(r.Upserted)),
 	}
 	if len(r.Upserted) > 0 {
 		res.UpsertedID = r.Upserted[0].ID
@@ -620,6 +621,7 @@ func (coll *Collection) UpdateMany(ctx context.Context, filter interface{}, upda
 	res := &UpdateResult{
 		MatchedCount:  r.MatchedCount,
 		ModifiedCount: r.ModifiedCount,
+		UpsertedCount: int64(len(r.Upserted)),
 	}
 	// TODO(skriptble): Is this correct? Do we only return the first upserted ID for an UpdateMany?
 	if len(r.Upserted) > 0 {

mongo/results.go

@@ -78,6 +78,8 @@ type UpdateResult struct {
 	MatchedCount int64
 	// The number of documents that were modified.
 	ModifiedCount int64
+	// The number of documents that were upserted.
+	UpsertedCount int64
 	// The identifier of the inserted document if an upsert took place.
 	UpsertedID interface{}
 }