update custom and setOnInsert

func25 · Oct 24, 2023 · af5fc42 · af5fc42
1 parent 0b257f9
commit af5fc42
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 2 deletions.
diff --git a/internal/mopertest/seed.go b/internal/mopertest/seed.go
@@ -41,7 +41,7 @@ func (Hero) CollName() string {
 
 // %s:%s/?w=majority&retryWrites=false
 func init() {
-	ctx, _ := context.WithTimeout(context.Background(), 3*time.Second)
+	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)
 	err := mocom.Connect(ctx, "mongodb://localhost:27017/?w=majority&retryWrites=false", "defaultdb")
 	if err != nil {
 		log.Fatal("cannot connect mongo", err)

diff --git a/mocom/find.go b/mocom/find.go
@@ -8,7 +8,7 @@ import (
 
 // Find finds documents from from model's collection
 // model should be implement `CollName() string`
-func FindT[T Modeler](ctx context.Context, filter interface{}, opts ...*options.FindOptions) (res []T, err error) {
+func Find[T Modeler](ctx context.Context, filter interface{}, opts ...*options.FindOptions) (res []T, err error) {
 	var t T
 	cur, err := CollRead(t.CollName()).Find(ctx, filter, opts...)
 	if err != nil {

diff --git a/moper/update.go b/moper/update.go
@@ -7,6 +7,8 @@ const (
 	_set   = "$set"
 	_push  = "$push"
 	_unset = "$unset"
+
+	_setOnInsert = "$setOnInsert"
 )
 
 // SET
@@ -22,6 +24,14 @@ func (d D) SetD(pairs D) D {
 	return append(d, bson.E{Key: _set, Value: pairs})
 }
 
+func (d D) SetOnInsert(key string, value interface{}) D {
+	return append(d, bson.E{Key: _setOnInsert, Value: bson.D{{Key: key, Value: value}}})
+}
+
+func (d D) SetManyOnInsert(pairs ...P) D {
+	return append(d, bson.E{Key: _setOnInsert, Value: toPair(pairs)})
+}
+
 // UNSET
 func (d D) Unset(keys ...string) D {
 	res := D{}
@@ -73,3 +83,8 @@ func toPair(pairs []P) bson.D {
 
 	return updated
 }
+
+// CUSTOM
+func (d D) CustomMany(cmd string, pairs ...P) D {
+	return append(d, bson.E{Key: cmd, Value: toPair(pairs)})
+}