feat:add replaceOne

ca-x · Oct 11, 2023 · 37f9451 · 37f9451
1 parent ce5d920
commit 37f9451
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 1 deletion.
diff --git a/cmd/kilot/kilot.go b/cmd/kilot/kilot.go
@@ -22,7 +22,7 @@ func main() {
 			},
 		},
 		Name:                 "kilot",
-		Version:              "0.1.17",
+		Version:              "0.1.18",
 		EnableBashCompletion: true,
 		Commands: []*cli.Command{
 			{

diff --git a/cmd/kilot/kratos/mongo/internal/templates/normal_data_layer_logic.tpl b/cmd/kilot/kratos/mongo/internal/templates/normal_data_layer_logic.tpl
@@ -58,6 +58,7 @@ func ({{.ModelIdentifier}} {{.ModelNameLowCase}}DataRepo) Find{{.ModelName}}(ctx
 	}
 	return result, nil
 }
+//TODO:implement ReplaceOne{{.ModelName}} and ReplaceOne{{.ModelName}}ById
 
 func ({{.ModelIdentifier}} {{.ModelNameLowCase}}DataRepo) Estimated{{.ModelName}}Count(ctx context.Context, opts ...*options.EstimatedDocumentCountOptions) (int64, error) {
 	return {{.ModelIdentifier}}.coll.EstimatedDocumentCount(ctx, opts...)

diff --git a/cmd/kilot/kratos/mongo/internal/templates/odm_data_layer_logic.tpl b/cmd/kilot/kratos/mongo/internal/templates/odm_data_layer_logic.tpl
@@ -70,6 +70,24 @@ func ({{.ModelIdentifier}} {{.ModelNameLowCase}}DataRepo) Find{{.ModelName}}(ctx
 	return result, nil
 }
 
+func ({{.ModelIdentifier}} {{.ModelNameLowCase}}DataRepo) ReplaceOne{{.ModelName}}(ctx context.Context, filter interface{}, data *{{.BizPkg}}.{{.ModelName}}, opts ...*options.ReplaceOptions) (result *mongo.UpdateResult, err error) {
+	if data.CreatedAt.IsZero() {
+		data.CreatedAt = time.Now()
+	}
+	return mgm.Coll({{.ModelIdentifier}}.model).ReplaceOne(ctx, filter, data, opts...)
+}
+
+func ({{.ModelIdentifier}} {{.ModelNameLowCase}}DataRepo) ReplaceOne{{.ModelName}}ById(ctx context.Context, id string, data *{{.BizPkg}}.{{.ModelName}}, opts ...*options.ReplaceOptions) (result *mongo.UpdateResult, err error) {
+	recordId, err := t.model.PrepareID(id)
+	if err != nil {
+		return nil, err
+	}
+	if data.CreatedAt.IsZero() {
+		data.CreatedAt = time.Now()
+	}
+	return mgm.Coll({{.ModelIdentifier}}.model).ReplaceOne(ctx, bson.M{"_id": recordId}, data, opts...)
+}
+
 func ({{.ModelIdentifier}} {{.ModelNameLowCase}}DataRepo) Estimated{{.ModelName}}Count(ctx context.Context, opts ...*options.EstimatedDocumentCountOptions) (int64, error) {
 	return mgm.Coll({{.ModelIdentifier}}.model).EstimatedDocumentCount(ctx, opts...)
 }