Export getId into function

mongox/base/getid.go

@@ -0,0 +1,45 @@
+package base
+
+import (
+	"github.com/mainnika/mongox-go-driver/mongox"
+	"github.com/mainnika/mongox-go-driver/mongox/errors"
+	"github.com/mongodb/mongo-go-driver/bson/primitive"
+)
+
+// GetID returns source document id
+func GetID(source interface{}) (id interface{}) {
+
+	switch doc := source.(type) {
+	case mongox.BaseObjectID:
+		return getObjectIdOrGenerate(doc)
+	case mongox.BaseString:
+		return getStringIdOrPanic(doc)
+	default:
+		panic(errors.Malformedf("source contains malformed document, %v", source))
+	}
+
+	return
+}
+
+func getObjectIdOrGenerate(source mongox.BaseObjectID) (id primitive.ObjectID) {
+
+	id = source.GetID()
+	if id != primitive.NilObjectID {
+		return id
+	}
+
+	id = primitive.NewObjectID()
+	source.SetID(id)
+
+	return
+}
+
+func getStringIdOrPanic(source mongox.BaseString) (id string) {
+
+	id = source.GetID()
+	if id != "" {
+		return id
+	}
+
+	panic(errors.Malformedf("victim contains malformed document, %v", source))
+}

mongox/common/saveone.go

@@ -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())