mainnika/mongox-go-driver
1
0
Fork 0
mirror of https://github.com/mainnika/mongox-go-driver.git synced 2026-07-04 01:32:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Protected save

Browse Source
This commit is contained in:
Nikita Tokarchuk
2019-02-07 23:38:06 +01:00
parent ac9b808a3e
commit 3a6eda141f
1 changed files with 18 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mongox/common/saveone.go
+18 -1
View File
@@ -5,7 +5,9 @@ import (
"github.com/mainnika/mongox-go-driver/mongox/base" "github.com/mainnika/mongox-go-driver/mongox/base"
"github.com/mainnika/mongox-go-driver/mongox/errors" "github.com/mainnika/mongox-go-driver/mongox/errors"
"github.com/mongodb/mongo-go-driver/bson" "github.com/mongodb/mongo-go-driver/bson"
"github.com/mongodb/mongo-go-driver/bson/primitive"
"github.com/mongodb/mongo-go-driver/mongo/options" "github.com/mongodb/mongo-go-driver/mongo/options"
"time"
) )
// SaveOne saves a single source document to the database // SaveOne saves a single source document to the database
@@ -14,11 +16,26 @@ func SaveOne(db *mongox.Database, source interface{}) error {
collection := db.GetCollectionOf(source) collection := db.GetCollectionOf(source)
opts := options.FindOneAndReplace() opts := options.FindOneAndReplace()
id := base.GetID(source) id := base.GetID(source)
protected := base.GetProtection(source)
filter := bson.M{"_id": id}
opts.SetUpsert(true) opts.SetUpsert(true)
opts.SetReturnDocument(options.After) opts.SetReturnDocument(options.After)
result := collection.FindOneAndReplace(db.Context(), bson.M{"_id": id}, source, opts) if protected != nil {
if protected.X.IsZero() {
filter["_x"] = primitive.M{"$exists": false}
filter["_v"] = primitive.M{"$exists": false}
} else {
filter["_x"] = protected.X
filter["_v"] = protected.V
}
protected.X = primitive.NewObjectID()
protected.V = time.Now().Unix()
}
result := collection.FindOneAndReplace(db.Context(), filter, source, opts)
if result.Err() != nil { if result.Err() != nil {
return errors.NotFoundErrorf("%s", result.Err()) return errors.NotFoundErrorf("%s", result.Err())
} }