feat(update): Implements SetOnInsertAny function and method.(#89)

Author: codepzj

builder/update/bson_build.go

@@ -74,6 +74,10 @@ func SetOnInsert(key string, value any) bson.D {
 	return bson.D{{Key: SetOnInsertOp, Value: bson.D{{Key: key, Value: value}}}}
 }
 
+func SetOnInsertAny(value any) bson.D {
+	return bson.D{{Key: SetOnInsertOp, Value: value}}
+}
+
 func CurrentDate(key string, value any) bson.D {
 	return bson.D{{Key: CurrentDateOp, Value: bson.D{{Key: key, Value: value}}}}
 }

builder/update/bson_build_test.go

@@ -179,6 +179,50 @@ func TestSetOnInsert(t *testing.T) {
 	})
 }
 
+func TestSetOnInsertAny(t *testing.T) {
+	type FieldStruct struct {
+		Name string `bson:"name,omitempty"`
+		Age  int    `bson:"age,omitempty"`
+	}
+
+	testCases := []struct {
+		name  string
+		value any
+		want  bson.D
+	}{
+		{
+			name:  "nil value",
+			value: nil,
+			want:  bson.D{bson.E{Key: "$setOnInsert", Value: nil}},
+		},
+		{
+			name:  "string value",
+			value: "Alice",
+			want:  bson.D{bson.E{Key: "$setOnInsert", Value: "Alice"}},
+		},
+		{
+			name:  "map value",
+			value: map[string]any{"name": "Alice"},
+			want:  bson.D{bson.E{Key: "$setOnInsert", Value: map[string]any{"name": "Alice"}}},
+		},
+		{
+			name:  "struct value",
+			value: FieldStruct{Name: "Alice", Age: 18},
+			want:  bson.D{bson.E{Key: "$setOnInsert", Value: FieldStruct{Name: "Alice", Age: 18}}},
+		},
+		{
+			name:  "bson value",
+			value: bson.D{bson.E{Key: "name", Value: "Alice"}},
+			want:  bson.D{bson.E{Key: "$setOnInsert", Value: bson.D{bson.E{Key: "name", Value: "Alice"}}}},
+		},
+	}
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			assert.Equal(t, tc.want, SetOnInsertAny(tc.value))
+		})
+	}
+}
+
 func TestCurrentDate(t *testing.T) {
 	t.Run("test CurrentDate", func(t *testing.T) {
 		assert.Equal(t, bson.D{bson.E{Key: "$currentDate", Value: bson.D{bson.E{Key: "lastModified", Value: true}}}}, CurrentDate("lastModified", true))

builder/update/field_update_builder.go

@@ -56,6 +56,14 @@ func (b *fieldUpdateBuilder) SetOnInsert(key string, value any) *Builder {
 	return b.parent
 }
 
+func (b *fieldUpdateBuilder) SetOnInsertAny(value any) *Builder {
+	e := bson.E{Key: SetOnInsertOp, Value: value}
+	if !b.parent.tryMergeValue(SetOnInsertOp, e) {
+		b.parent.data = append(b.parent.data, e)
+	}
+	return b.parent
+}
+
 func (b *fieldUpdateBuilder) CurrentDate(key string, value any) *Builder {
 	e := bson.E{Key: key, Value: value}
 	if !b.parent.tryMergeValue(CurrentDateOp, e) {

builder/update/field_update_builder_test.go

@@ -106,6 +106,45 @@ func Test_fieldUpdateBuilder_SetOnInsert(t *testing.T) {
 	})
 }
 
+func Test_fieldUpdateBuilder_SetOnInsertAny(t *testing.T) {
+	testCases := []struct {
+		name  string
+		value any
+		want  bson.D
+	}{
+		{
+			name:  "nil value",
+			value: nil,
+			want:  bson.D{bson.E{Key: "$setOnInsert", Value: nil}},
+		},
+		{
+			name:  "string value",
+			value: "Alice",
+			want:  bson.D{bson.E{Key: "$setOnInsert", Value: "Alice"}},
+		},
+		{
+			name:  "map value",
+			value: map[string]any{"name": "Alice"},
+			want:  bson.D{bson.E{Key: "$setOnInsert", Value: map[string]any{"name": "Alice"}}},
+		},
+		{
+			name:  "struct value",
+			value: struct{ Name string }{Name: "Alice"},
+			want:  bson.D{bson.E{Key: "$setOnInsert", Value: struct{ Name string }{Name: "Alice"}}},
+		},
+		{
+			name:  "bson value",
+			value: bson.D{bson.E{Key: "name", Value: "Alice"}},
+			want:  bson.D{bson.E{Key: "$setOnInsert", Value: bson.D{bson.E{Key: "name", Value: "Alice"}}}},
+		},
+	}
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			assert.Equal(t, tc.want, NewBuilder().SetOnInsertAny(tc.value).Build())
+		})
+	}
+}
+
 func Test_fieldUpdateBuilder_Inc(t *testing.T) {
 	t.Run("single operation", func(t *testing.T) {
 		assert.Equal(t, bson.D{{Key: "$inc", Value: bson.D{bson.E{Key: "orders", Value: 1}}}}, NewBuilder().Inc("orders", 1).Build())