add index creation in odm template

ca-x · Jul 17, 2023 · 8a1fd04 · 8a1fd04
1 parent 81b43a1
commit 8a1fd04
Show file tree

Hide file tree

Showing 3 changed files with 11 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.15",
+		Version:              "0.1.16",
 		EnableBashCompletion: true,
 		Commands: []*cli.Command{
 			{

diff --git a/cmd/kilot/kratos/mongo/internal/templates/odm_biz_layer_logic.tpl b/cmd/kilot/kratos/mongo/internal/templates/odm_biz_layer_logic.tpl
@@ -11,7 +11,9 @@ type {{.ModelName}}Provider interface {
     Aggregate{{.ModelName}}(ctx context.Context, pipeline interface{}, opts ...*options.AggregateOptions) (*mongo.Cursor, error)
     Create{{.ModelName}}(ctx context.Context, data *{{.ModelName}}, opts ...*options.InsertOneOptions) error
 	CreateOne{{.ModelName}}(ctx context.Context, data *{{.ModelName}}, opts ...*options.InsertOneOptions) (result *mongo.InsertOneResult, err error)
+	CreateOne{{.ModelName}}Index(ctx context.Context, index mongo.IndexModel, opts ...*options.CreateIndexesOptions) (result string, err error)
 	CreateMany{{.ModelName}}(ctx context.Context, data []interface{}, opts ...*options.InsertManyOptions) (result *mongo.InsertManyResult, err error)
+	CreateMany{{.ModelName}}Index(ctx context.Context, index []mongo.IndexModel, opts ...*options.CreateIndexesOptions) (result []string, err error)
 	FindOne{{.ModelName}}(ctx context.Context, filter interface{}, opts ...*options.FindOneOptions) (result *{{.ModelName}}, err error)
 	FindOne{{.ModelName}}ById(ctx context.Context, id string, opts ...*options.FindOneOptions) (result *{{.ModelName}}, err error)
 	Find{{.ModelName}}(ctx context.Context, filter interface{}, opts ...*options.FindOptions) (result []*{{.ModelName}}, err error)

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
@@ -30,10 +30,18 @@ func ({{.ModelIdentifier}} {{.ModelNameLowCase}}DataRepo) CreateOne{{.ModelName}
 	return mgm.Coll({{.ModelIdentifier}}.model).InsertOne(ctx, data, opts...)
 }
 
+func ({{.ModelIdentifier}} {{.ModelNameLowCase}}DataRepo) CreateOne{{.ModelName}}Index(ctx context.Context, index mongo.IndexModel, opts ...*options.CreateIndexesOptions) (result string, err error) {
+	return mgm.Coll({{.ModelIdentifier}}.model).Indexes().CreateOne(ctx, index, opts...)
+}
+
 func ({{.ModelIdentifier}} {{.ModelNameLowCase}}DataRepo) CreateMany{{.ModelName}}(ctx context.Context, data []interface{}, opts ...*options.InsertManyOptions) (result *mongo.InsertManyResult, err error) {
 	return mgm.Coll({{.ModelIdentifier}}.model).InsertMany(ctx, data, opts...)
 }
 
+func ({{.ModelIdentifier}} {{.ModelNameLowCase}}DataRepo) Create{{.ModelName}}SessionIndex(ctx context.Context, indexes []mongo.IndexModel, opts ...*options.CreateIndexesOptions) (result []string, err error) {
+	return mgm.Coll({{.ModelIdentifier}}.model).Indexes().CreateMany(ctx, indexes, opts...)
+}
+
 func ({{.ModelIdentifier}} {{.ModelNameLowCase}}DataRepo) FindOne{{.ModelName}}(ctx context.Context, filter interface{}, opts ...*options.FindOneOptions) (result *{{.BizPkg}}.{{.ModelName}}, err error) {
 	result = &{{.BizPkg}}.{{.ModelName}}{}
 	if err = mgm.Coll({{.ModelIdentifier}}.model).FindOne(ctx, filter, opts...).Decode(result); err != nil {