From bbbdf1a6ecdd6794ca00c7bac8e5600d84ce6892 Mon Sep 17 00:00:00 2001 From: Nikita Tokarchuk Date: Wed, 26 Dec 2018 23:18:13 +0100 Subject: [PATCH] DeleteOne function --- mongox/common/deleteone.go | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 mongox/common/deleteone.go diff --git a/mongox/common/deleteone.go b/mongox/common/deleteone.go new file mode 100644 index 0000000..d18cef8 --- /dev/null +++ b/mongox/common/deleteone.go @@ -0,0 +1,34 @@ +package common + +import ( + "github.com/mainnika/mongox-go-driver/mongox" + "github.com/mainnika/mongox-go-driver/mongox/errors" + "github.com/mainnika/mongox-go-driver/mongox/query" + "github.com/mongodb/mongo-go-driver/mongo" + "github.com/mongodb/mongo-go-driver/mongo/options" +) + +// DeleteOne removes a document from a database and then returns it into target +func DeleteOne(db *mongox.Database, target interface{}, filters ...interface{}) error { + + collection := db.GetCollectionOf(target) + opts := &options.FindOneAndDeleteOptions{} + composed := query.Compose(filters...) + + opts.Sort = composed.Sorter() + + result := collection.FindOneAndDelete(db.Context(), composed.M(), opts) + if result.Err() != nil { + return errors.InternalErrorf("can't create find one and delete result: %s", result.Err()) + } + + err := result.Decode(target) + if err == mongo.ErrNoDocuments { + return errors.NotFoundErrorf("%s", err) + } + if err != nil { + return errors.InternalErrorf("can't decode result: %s", err) + } + + return nil +}