Export getId into function

This commit is contained in:
Nikita Tokarchuk
2018-12-26 23:14:48 +01:00
parent 557788255e
commit bc595dd024
2 changed files with 46 additions and 17 deletions
+1 -17
View File
@@ -13,27 +13,11 @@ func SaveOne(db *mongox.Database, source interface{}) error {
collection := db.GetCollectionOf(source)
opts := &options.FindOneAndReplaceOptions{}
id := base.GetID(source)
opts.SetUpsert(true)
opts.SetReturnDocument(options.After)
var id interface{}
switch doc := source.(type) {
case mongox.BaseObjectID:
id = doc.GetID()
if id == primitive.NilObjectID {
id = primitive.NewObjectID()
}
case mongox.BaseString:
id = doc.GetID()
if id == "" {
panic(errors.Malformedf("source contains malformed document, %v", source))
}
default:
panic(errors.Malformedf("source contains malformed document, %v", source))
}
result := collection.FindOneAndReplace(db.Context(), bson.M{"_id": id}, source, opts)
if result.Err() != nil {
return errors.NotFoundErrorf("%s", result.Err())