mirror of
https://github.com/mainnika/mongox-go-driver.git
synced 2026-05-23 00:03:36 +00:00
Export getId into function
This commit is contained in:
@@ -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())
|
||||
|
||||
Reference in New Issue
Block a user