Add functionality to sort in find operation

This commit is contained in:
sunboyy 2021-02-26 10:27:00 +07:00
parent 6c00fa860b
commit 962779ce40
10 changed files with 460 additions and 31 deletions

View file

@ -131,7 +131,7 @@ FindByCity(ctx context.Context, city string) ([]*Model, error)
FindAll(ctx context.Context) ([]*Model, error) FindAll(ctx context.Context) ([]*Model, error)
``` ```
Repogen determines a single-entity operation or a multiple-entity by checking the first return value. If it is a pointer of a model, the method will be single-entity operation. If it is a slice of pointers of a model, the method will be multiple-entity operation. Repogen determines a single-entity or a multiple-entity operation by checking the first return value. If it is a pointer of a model, the method will be single-entity operation. If it is a slice of pointers of a model, the method will be multiple-entity operation.
The requirement of the `Find` operation method is that there must be only two return values, the second return value must be of type `error` and the first method parameter must be of type `context.Context`. The requirement of number of method parameters depends on the query which will be described in the query specification section. The requirement of the `Find` operation method is that there must be only two return values, the second return value must be of type `error` and the first method parameter must be of type `context.Context`. The requirement of number of method parameters depends on the query which will be described in the query specification section.

View file

@ -77,7 +77,7 @@ func TestGenerateMongoRepository(t *testing.T) {
}, },
}, },
{ {
Name: "FindByAgeLessThanEqual", Name: "FindByAgeLessThanEqualOrderByAge",
Params: []code.Param{ Params: []code.Param{
{Name: "ctx", Type: code.ExternalType{PackageAlias: "context", Name: "Context"}}, {Name: "ctx", Type: code.ExternalType{PackageAlias: "context", Name: "Context"}},
{Name: "age", Type: code.SimpleType("int")}, {Name: "age", Type: code.SimpleType("int")},
@ -93,10 +93,13 @@ func TestGenerateMongoRepository(t *testing.T) {
{Field: "Age", Comparator: spec.ComparatorLessThanEqual, ParamIndex: 1}, {Field: "Age", Comparator: spec.ComparatorLessThanEqual, ParamIndex: 1},
}, },
}, },
Sorts: []spec.Sort{
{FieldName: "Age", Ordering: spec.OrderingAscending},
},
}, },
}, },
{ {
Name: "FindByAgeGreaterThan", Name: "FindByAgeGreaterThanOrderByAgeAsc",
Params: []code.Param{ Params: []code.Param{
{Name: "ctx", Type: code.ExternalType{PackageAlias: "context", Name: "Context"}}, {Name: "ctx", Type: code.ExternalType{PackageAlias: "context", Name: "Context"}},
{Name: "age", Type: code.SimpleType("int")}, {Name: "age", Type: code.SimpleType("int")},
@ -112,10 +115,13 @@ func TestGenerateMongoRepository(t *testing.T) {
{Field: "Age", Comparator: spec.ComparatorGreaterThan, ParamIndex: 1}, {Field: "Age", Comparator: spec.ComparatorGreaterThan, ParamIndex: 1},
}, },
}, },
Sorts: []spec.Sort{
{FieldName: "Age", Ordering: spec.OrderingAscending},
},
}, },
}, },
{ {
Name: "FindByAgeGreaterThanEqual", Name: "FindByAgeGreaterThanEqualOrderByAgeDesc",
Params: []code.Param{ Params: []code.Param{
{Name: "ctx", Type: code.ExternalType{PackageAlias: "context", Name: "Context"}}, {Name: "ctx", Type: code.ExternalType{PackageAlias: "context", Name: "Context"}},
{Name: "age", Type: code.SimpleType("int")}, {Name: "age", Type: code.SimpleType("int")},
@ -131,6 +137,9 @@ func TestGenerateMongoRepository(t *testing.T) {
{Field: "Age", Comparator: spec.ComparatorGreaterThanEqual, ParamIndex: 1}, {Field: "Age", Comparator: spec.ComparatorGreaterThanEqual, ParamIndex: 1},
}, },
}, },
Sorts: []spec.Sort{
{FieldName: "Age", Ordering: spec.OrderingDescending},
},
}, },
}, },
{ {
@ -191,6 +200,7 @@ import (
"go.mongodb.org/mongo-driver/bson" "go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive" "go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo" "go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
) )
func NewUserRepository(collection *mongo.Collection) UserRepository { func NewUserRepository(collection *mongo.Collection) UserRepository {
@ -207,7 +217,7 @@ func (r *UserRepositoryMongo) FindByID(arg0 context.Context, arg1 primitive.Obje
var entity UserModel var entity UserModel
if err := r.collection.FindOne(arg0, bson.M{ if err := r.collection.FindOne(arg0, bson.M{
"_id": arg1, "_id": arg1,
}).Decode(&entity); err != nil { }, options.FindOne().SetSort(bson.M{})).Decode(&entity); err != nil {
return nil, err return nil, err
} }
return &entity, nil return &entity, nil
@ -219,7 +229,7 @@ func (r *UserRepositoryMongo) FindByGenderNotAndAgeLessThan(arg0 context.Context
{"gender": bson.M{"$ne": arg1}}, {"gender": bson.M{"$ne": arg1}},
{"age": bson.M{"$lt": arg2}}, {"age": bson.M{"$lt": arg2}},
}, },
}) }, options.Find().SetSort(bson.M{}))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -230,10 +240,12 @@ func (r *UserRepositoryMongo) FindByGenderNotAndAgeLessThan(arg0 context.Context
return entities, nil return entities, nil
} }
func (r *UserRepositoryMongo) FindByAgeLessThanEqual(arg0 context.Context, arg1 int) ([]*UserModel, error) { func (r *UserRepositoryMongo) FindByAgeLessThanEqualOrderByAge(arg0 context.Context, arg1 int) ([]*UserModel, error) {
cursor, err := r.collection.Find(arg0, bson.M{ cursor, err := r.collection.Find(arg0, bson.M{
"age": bson.M{"$lte": arg1}, "age": bson.M{"$lte": arg1},
}) }, options.Find().SetSort(bson.M{
"age": 1,
}))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -244,10 +256,12 @@ func (r *UserRepositoryMongo) FindByAgeLessThanEqual(arg0 context.Context, arg1
return entities, nil return entities, nil
} }
func (r *UserRepositoryMongo) FindByAgeGreaterThan(arg0 context.Context, arg1 int) ([]*UserModel, error) { func (r *UserRepositoryMongo) FindByAgeGreaterThanOrderByAgeAsc(arg0 context.Context, arg1 int) ([]*UserModel, error) {
cursor, err := r.collection.Find(arg0, bson.M{ cursor, err := r.collection.Find(arg0, bson.M{
"age": bson.M{"$gt": arg1}, "age": bson.M{"$gt": arg1},
}) }, options.Find().SetSort(bson.M{
"age": 1,
}))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -258,10 +272,12 @@ func (r *UserRepositoryMongo) FindByAgeGreaterThan(arg0 context.Context, arg1 in
return entities, nil return entities, nil
} }
func (r *UserRepositoryMongo) FindByAgeGreaterThanEqual(arg0 context.Context, arg1 int) ([]*UserModel, error) { func (r *UserRepositoryMongo) FindByAgeGreaterThanEqualOrderByAgeDesc(arg0 context.Context, arg1 int) ([]*UserModel, error) {
cursor, err := r.collection.Find(arg0, bson.M{ cursor, err := r.collection.Find(arg0, bson.M{
"age": bson.M{"$gte": arg1}, "age": bson.M{"$gte": arg1},
}) }, options.Find().SetSort(bson.M{
"age": -1,
}))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -275,7 +291,7 @@ func (r *UserRepositoryMongo) FindByAgeGreaterThanEqual(arg0 context.Context, ar
func (r *UserRepositoryMongo) FindByAgeBetween(arg0 context.Context, arg1 int, arg2 int) ([]*UserModel, error) { func (r *UserRepositoryMongo) FindByAgeBetween(arg0 context.Context, arg1 int, arg2 int) ([]*UserModel, error) {
cursor, err := r.collection.Find(arg0, bson.M{ cursor, err := r.collection.Find(arg0, bson.M{
"age": bson.M{"$gte": arg1, "$lte": arg2}, "age": bson.M{"$gte": arg1, "$lte": arg2},
}) }, options.Find().SetSort(bson.M{}))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -292,7 +308,7 @@ func (r *UserRepositoryMongo) FindByGenderOrAge(arg0 context.Context, arg1 Gende
{"gender": arg1}, {"gender": arg1},
{"age": arg2}, {"age": arg2},
}, },
}) }, options.Find().SetSort(bson.M{}))
if err != nil { if err != nil {
return nil, err return nil, err
} }

View file

@ -105,9 +105,15 @@ func (g RepositoryGenerator) generateFindImplementation(operation spec.FindOpera
return "", err return "", err
} }
sorts, err := g.mongoSorts(operation.Sorts)
if err != nil {
return "", err
}
tmplData := mongoFindTemplateData{ tmplData := mongoFindTemplateData{
EntityType: g.StructModel.Name, EntityType: g.StructModel.Name,
QuerySpec: querySpec, QuerySpec: querySpec,
Sorts: sorts,
} }
if operation.Mode == spec.QueryModeOne { if operation.Mode == spec.QueryModeOne {
@ -116,6 +122,24 @@ func (g RepositoryGenerator) generateFindImplementation(operation spec.FindOpera
return generateFromTemplate("mongo_repository_findmany", findManyTemplate, tmplData) return generateFromTemplate("mongo_repository_findmany", findManyTemplate, tmplData)
} }
func (g RepositoryGenerator) mongoSorts(sortSpec []spec.Sort) ([]sort, error) {
var sorts []sort
for _, s := range sortSpec {
bsonTag, err := g.bsonTagFromFieldName(s.FieldName)
if err != nil {
return nil, err
}
sorts = append(sorts, sort{
BsonTag: bsonTag,
Ordering: s.Ordering,
})
}
return sorts, nil
}
func (g RepositoryGenerator) generateUpdateImplementation(operation spec.UpdateOperation) (string, error) { func (g RepositoryGenerator) generateUpdateImplementation(operation spec.UpdateOperation) (string, error) {
update, err := g.getMongoUpdate(operation.Update) update, err := g.getMongoUpdate(operation.Update)
if err != nil { if err != nil {

View file

@ -52,6 +52,7 @@ import (
"go.mongodb.org/mongo-driver/bson" "go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive" "go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo" "go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
) )
func NewUserRepository(collection *mongo.Collection) UserRepository { func NewUserRepository(collection *mongo.Collection) UserRepository {
@ -192,7 +193,8 @@ func (r *UserRepositoryMongo) FindByID(arg0 context.Context, arg1 primitive.Obje
var entity UserModel var entity UserModel
if err := r.collection.FindOne(arg0, bson.M{ if err := r.collection.FindOne(arg0, bson.M{
"_id": arg1, "_id": arg1,
}).Decode(&entity); err != nil { }, options.FindOne().SetSort(bson.M{
})).Decode(&entity); err != nil {
return nil, err return nil, err
} }
return &entity, nil return &entity, nil
@ -224,7 +226,8 @@ func (r *UserRepositoryMongo) FindByID(arg0 context.Context, arg1 primitive.Obje
func (r *UserRepositoryMongo) FindByGender(arg0 context.Context, arg1 Gender) ([]*UserModel, error) { func (r *UserRepositoryMongo) FindByGender(arg0 context.Context, arg1 Gender) ([]*UserModel, error) {
cursor, err := r.collection.Find(arg0, bson.M{ cursor, err := r.collection.Find(arg0, bson.M{
"gender": arg1, "gender": arg1,
}) }, options.Find().SetSort(bson.M{
}))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -267,7 +270,8 @@ func (r *UserRepositoryMongo) FindByGenderAndAge(arg0 context.Context, arg1 Gend
{"gender": arg1}, {"gender": arg1},
{"age": arg2}, {"age": arg2},
}, },
}) }, options.Find().SetSort(bson.M{
}))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -310,7 +314,8 @@ func (r *UserRepositoryMongo) FindByGenderOrAge(arg0 context.Context, arg1 Gende
{"gender": arg1}, {"gender": arg1},
{"age": arg2}, {"age": arg2},
}, },
}) }, options.Find().SetSort(bson.M{
}))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -347,7 +352,8 @@ func (r *UserRepositoryMongo) FindByGenderOrAge(arg0 context.Context, arg1 Gende
func (r *UserRepositoryMongo) FindByGenderNot(arg0 context.Context, arg1 Gender) ([]*UserModel, error) { func (r *UserRepositoryMongo) FindByGenderNot(arg0 context.Context, arg1 Gender) ([]*UserModel, error) {
cursor, err := r.collection.Find(arg0, bson.M{ cursor, err := r.collection.Find(arg0, bson.M{
"gender": bson.M{"$ne": arg1}, "gender": bson.M{"$ne": arg1},
}) }, options.Find().SetSort(bson.M{
}))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -384,7 +390,8 @@ func (r *UserRepositoryMongo) FindByGenderNot(arg0 context.Context, arg1 Gender)
func (r *UserRepositoryMongo) FindByAgeLessThan(arg0 context.Context, arg1 int) ([]*UserModel, error) { func (r *UserRepositoryMongo) FindByAgeLessThan(arg0 context.Context, arg1 int) ([]*UserModel, error) {
cursor, err := r.collection.Find(arg0, bson.M{ cursor, err := r.collection.Find(arg0, bson.M{
"age": bson.M{"$lt": arg1}, "age": bson.M{"$lt": arg1},
}) }, options.Find().SetSort(bson.M{
}))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -421,7 +428,8 @@ func (r *UserRepositoryMongo) FindByAgeLessThan(arg0 context.Context, arg1 int)
func (r *UserRepositoryMongo) FindByAgeLessThanEqual(arg0 context.Context, arg1 int) ([]*UserModel, error) { func (r *UserRepositoryMongo) FindByAgeLessThanEqual(arg0 context.Context, arg1 int) ([]*UserModel, error) {
cursor, err := r.collection.Find(arg0, bson.M{ cursor, err := r.collection.Find(arg0, bson.M{
"age": bson.M{"$lte": arg1}, "age": bson.M{"$lte": arg1},
}) }, options.Find().SetSort(bson.M{
}))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -458,7 +466,8 @@ func (r *UserRepositoryMongo) FindByAgeLessThanEqual(arg0 context.Context, arg1
func (r *UserRepositoryMongo) FindByAgeGreaterThan(arg0 context.Context, arg1 int) ([]*UserModel, error) { func (r *UserRepositoryMongo) FindByAgeGreaterThan(arg0 context.Context, arg1 int) ([]*UserModel, error) {
cursor, err := r.collection.Find(arg0, bson.M{ cursor, err := r.collection.Find(arg0, bson.M{
"age": bson.M{"$gt": arg1}, "age": bson.M{"$gt": arg1},
}) }, options.Find().SetSort(bson.M{
}))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -495,7 +504,8 @@ func (r *UserRepositoryMongo) FindByAgeGreaterThan(arg0 context.Context, arg1 in
func (r *UserRepositoryMongo) FindByAgeGreaterThanEqual(arg0 context.Context, arg1 int) ([]*UserModel, error) { func (r *UserRepositoryMongo) FindByAgeGreaterThanEqual(arg0 context.Context, arg1 int) ([]*UserModel, error) {
cursor, err := r.collection.Find(arg0, bson.M{ cursor, err := r.collection.Find(arg0, bson.M{
"age": bson.M{"$gte": arg1}, "age": bson.M{"$gte": arg1},
}) }, options.Find().SetSort(bson.M{
}))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -533,7 +543,8 @@ func (r *UserRepositoryMongo) FindByAgeGreaterThanEqual(arg0 context.Context, ar
func (r *UserRepositoryMongo) FindByAgeBetween(arg0 context.Context, arg1 int, arg2 int) ([]*UserModel, error) { func (r *UserRepositoryMongo) FindByAgeBetween(arg0 context.Context, arg1 int, arg2 int) ([]*UserModel, error) {
cursor, err := r.collection.Find(arg0, bson.M{ cursor, err := r.collection.Find(arg0, bson.M{
"age": bson.M{"$gte": arg1, "$lte": arg2}, "age": bson.M{"$gte": arg1, "$lte": arg2},
}) }, options.Find().SetSort(bson.M{
}))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -570,7 +581,8 @@ func (r *UserRepositoryMongo) FindByAgeBetween(arg0 context.Context, arg1 int, a
func (r *UserRepositoryMongo) FindByGenderIn(arg0 context.Context, arg1 []Gender) ([]*UserModel, error) { func (r *UserRepositoryMongo) FindByGenderIn(arg0 context.Context, arg1 []Gender) ([]*UserModel, error) {
cursor, err := r.collection.Find(arg0, bson.M{ cursor, err := r.collection.Find(arg0, bson.M{
"gender": bson.M{"$in": arg1}, "gender": bson.M{"$in": arg1},
}) }, options.Find().SetSort(bson.M{
}))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -607,7 +619,8 @@ func (r *UserRepositoryMongo) FindByGenderIn(arg0 context.Context, arg1 []Gender
func (r *UserRepositoryMongo) FindByGenderNotIn(arg0 context.Context, arg1 []Gender) ([]*UserModel, error) { func (r *UserRepositoryMongo) FindByGenderNotIn(arg0 context.Context, arg1 []Gender) ([]*UserModel, error) {
cursor, err := r.collection.Find(arg0, bson.M{ cursor, err := r.collection.Find(arg0, bson.M{
"gender": bson.M{"$nin": arg1}, "gender": bson.M{"$nin": arg1},
}) }, options.Find().SetSort(bson.M{
}))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -643,7 +656,8 @@ func (r *UserRepositoryMongo) FindByGenderNotIn(arg0 context.Context, arg1 []Gen
func (r *UserRepositoryMongo) FindByEnabledTrue(arg0 context.Context) ([]*UserModel, error) { func (r *UserRepositoryMongo) FindByEnabledTrue(arg0 context.Context) ([]*UserModel, error) {
cursor, err := r.collection.Find(arg0, bson.M{ cursor, err := r.collection.Find(arg0, bson.M{
"enabled": true, "enabled": true,
}) }, options.Find().SetSort(bson.M{
}))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -679,7 +693,118 @@ func (r *UserRepositoryMongo) FindByEnabledTrue(arg0 context.Context) ([]*UserMo
func (r *UserRepositoryMongo) FindByEnabledFalse(arg0 context.Context) ([]*UserModel, error) { func (r *UserRepositoryMongo) FindByEnabledFalse(arg0 context.Context) ([]*UserModel, error) {
cursor, err := r.collection.Find(arg0, bson.M{ cursor, err := r.collection.Find(arg0, bson.M{
"enabled": false, "enabled": false,
}) }, options.Find().SetSort(bson.M{
}))
if err != nil {
return nil, err
}
var entities []*UserModel
if err := cursor.All(arg0, &entities); err != nil {
return nil, err
}
return entities, nil
}
`,
},
{
Name: "find with sort ascending",
MethodSpec: spec.MethodSpec{
Name: "FindAllOrderByAge",
Params: []code.Param{
{Name: "ctx", Type: code.ExternalType{PackageAlias: "context", Name: "Context"}},
},
Returns: []code.Type{
code.ArrayType{ContainedType: code.PointerType{ContainedType: code.SimpleType("UserModel")}},
code.SimpleType("error"),
},
Operation: spec.FindOperation{
Mode: spec.QueryModeMany,
Sorts: []spec.Sort{
{FieldName: "Age", Ordering: spec.OrderingAscending},
},
},
},
ExpectedCode: `
func (r *UserRepositoryMongo) FindAllOrderByAge(arg0 context.Context) ([]*UserModel, error) {
cursor, err := r.collection.Find(arg0, bson.M{
}, options.Find().SetSort(bson.M{
"age": 1,
}))
if err != nil {
return nil, err
}
var entities []*UserModel
if err := cursor.All(arg0, &entities); err != nil {
return nil, err
}
return entities, nil
}
`,
},
{
Name: "find with sort descending",
MethodSpec: spec.MethodSpec{
Name: "FindAllOrderByAge",
Params: []code.Param{
{Name: "ctx", Type: code.ExternalType{PackageAlias: "context", Name: "Context"}},
},
Returns: []code.Type{
code.ArrayType{ContainedType: code.PointerType{ContainedType: code.SimpleType("UserModel")}},
code.SimpleType("error"),
},
Operation: spec.FindOperation{
Mode: spec.QueryModeMany,
Sorts: []spec.Sort{
{FieldName: "Age", Ordering: spec.OrderingDescending},
},
},
},
ExpectedCode: `
func (r *UserRepositoryMongo) FindAllOrderByAge(arg0 context.Context) ([]*UserModel, error) {
cursor, err := r.collection.Find(arg0, bson.M{
}, options.Find().SetSort(bson.M{
"age": -1,
}))
if err != nil {
return nil, err
}
var entities []*UserModel
if err := cursor.All(arg0, &entities); err != nil {
return nil, err
}
return entities, nil
}
`,
},
{
Name: "find with multiple sorts",
MethodSpec: spec.MethodSpec{
Name: "FindAllOrderByGenderAndAgeDesc",
Params: []code.Param{
{Name: "ctx", Type: code.ExternalType{PackageAlias: "context", Name: "Context"}},
},
Returns: []code.Type{
code.ArrayType{ContainedType: code.PointerType{ContainedType: code.SimpleType("UserModel")}},
code.SimpleType("error"),
},
Operation: spec.FindOperation{
Mode: spec.QueryModeMany,
Sorts: []spec.Sort{
{FieldName: "Gender", Ordering: spec.OrderingAscending},
{FieldName: "Age", Ordering: spec.OrderingDescending},
},
},
},
ExpectedCode: `
func (r *UserRepositoryMongo) FindAllOrderByGenderAndAgeDesc(arg0 context.Context) ([]*UserModel, error) {
cursor, err := r.collection.Find(arg0, bson.M{
}, options.Find().SetSort(bson.M{
"gender": 1,
"age": -1,
}))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -1652,6 +1777,26 @@ func TestGenerateMethod_Invalid(t *testing.T) {
}, },
ExpectedError: mongo.NewBsonTagNotFoundError("AccessToken"), ExpectedError: mongo.NewBsonTagNotFoundError("AccessToken"),
}, },
{
Name: "bson tag not found in sort",
Method: spec.MethodSpec{
Name: "FindAllOrderByAccessToken",
Params: []code.Param{
{Type: code.ExternalType{PackageAlias: "context", Name: "Context"}},
},
Returns: []code.Type{
code.ArrayType{ContainedType: code.PointerType{ContainedType: code.SimpleType("UserModel")}},
code.SimpleType("error"),
},
Operation: spec.FindOperation{
Mode: spec.QueryModeOne,
Sorts: []spec.Sort{
{FieldName: "AccessToken", Ordering: spec.OrderingAscending},
},
},
},
ExpectedError: mongo.NewBsonTagNotFoundError("AccessToken"),
},
{ {
Name: "bson tag not found in update field", Name: "bson tag not found in update field",
Method: spec.MethodSpec{ Method: spec.MethodSpec{

View file

@ -5,6 +5,7 @@ import (
"strings" "strings"
"github.com/sunboyy/repogen/internal/code" "github.com/sunboyy/repogen/internal/code"
"github.com/sunboyy/repogen/internal/spec"
) )
const constructorTemplate = ` const constructorTemplate = `
@ -14,6 +15,7 @@ import (
"go.mongodb.org/mongo-driver/bson" "go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive" "go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo" "go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
) )
func New{{.InterfaceName}}(collection *mongo.Collection) {{.InterfaceName}} { func New{{.InterfaceName}}(collection *mongo.Collection) {{.InterfaceName}} {
@ -89,19 +91,36 @@ const insertManyTemplate = ` var entities []interface{}
type mongoFindTemplateData struct { type mongoFindTemplateData struct {
EntityType string EntityType string
QuerySpec querySpec QuerySpec querySpec
Sorts []sort
}
type sort struct {
BsonTag string
Ordering spec.Ordering
}
func (s sort) OrderNum() int {
if s.Ordering == spec.OrderingAscending {
return 1
}
return -1
} }
const findOneTemplate = ` var entity {{.EntityType}} const findOneTemplate = ` var entity {{.EntityType}}
if err := r.collection.FindOne(arg0, bson.M{ if err := r.collection.FindOne(arg0, bson.M{
{{.QuerySpec.Code}} {{.QuerySpec.Code}}
}).Decode(&entity); err != nil { }, options.FindOne().SetSort(bson.M{
{{range $index, $element := .Sorts}} "{{$element.BsonTag}}": {{$element.OrderNum}},
{{end}} })).Decode(&entity); err != nil {
return nil, err return nil, err
} }
return &entity, nil` return &entity, nil`
const findManyTemplate = ` cursor, err := r.collection.Find(arg0, bson.M{ const findManyTemplate = ` cursor, err := r.collection.Find(arg0, bson.M{
{{.QuerySpec.Code}} {{.QuerySpec.Code}}
}) }, options.Find().SetSort(bson.M{
{{range $index, $element := .Sorts}} "{{$element.BsonTag}}": {{$element.OrderNum}},
{{end}} }))
if err != nil { if err != nil {
return nil, err return nil, err
} }

View file

@ -48,6 +48,19 @@ func (err invalidQueryError) Error() string {
return fmt.Sprintf("invalid query '%s'", err.QueryString) return fmt.Sprintf("invalid query '%s'", err.QueryString)
} }
// NewInvalidSortError creates invalidSortError
func NewInvalidSortError(sortTokens []string) error {
return invalidSortError{SortString: strings.Join(sortTokens, "")}
}
type invalidSortError struct {
SortString string
}
func (err invalidSortError) Error() string {
return fmt.Sprintf("invalid sort '%s'", err.SortString)
}
// NewUnknownOperationError creates unknownOperationError // NewUnknownOperationError creates unknownOperationError
func NewUnknownOperationError(operationName string) error { func NewUnknownOperationError(operationName string) error {
return unknownOperationError{OperationName: operationName} return unknownOperationError{OperationName: operationName}

View file

@ -38,6 +38,11 @@ func TestError(t *testing.T) {
}), }),
ExpectedString: "cannot use comparator EQUAL_TRUE with struct field 'Age' of type 'int'", ExpectedString: "cannot use comparator EQUAL_TRUE with struct field 'Age' of type 'int'",
}, },
{
Name: "InvalidSortError",
Error: spec.NewInvalidSortError([]string{"Order", "By"}),
ExpectedString: "invalid sort 'OrderBy'",
},
} }
for _, testCase := range testTable { for _, testCase := range testTable {

View file

@ -40,6 +40,7 @@ func (o InsertOperation) Name() string {
type FindOperation struct { type FindOperation struct {
Mode QueryMode Mode QueryMode
Query QuerySpec Query QuerySpec
Sorts []Sort
} }
// Name returns "Find" operation name // Name returns "Find" operation name
@ -47,6 +48,21 @@ func (o FindOperation) Name() string {
return "Find" return "Find"
} }
// Sort is a detail of sorting find result
type Sort struct {
FieldName string
Ordering Ordering
}
// Ordering is a sort order
type Ordering string
// Ordering constants
const (
OrderingAscending = "ASC"
OrderingDescending = "DESC"
)
// UpdateOperation is a method specification for update operations // UpdateOperation is a method specification for update operations
type UpdateOperation struct { type UpdateOperation struct {
Update Update Update Update

View file

@ -1,6 +1,8 @@
package spec package spec
import ( import (
"strings"
"github.com/fatih/camelcase" "github.com/fatih/camelcase"
"github.com/sunboyy/repogen/internal/code" "github.com/sunboyy/repogen/internal/code"
) )
@ -111,7 +113,14 @@ func (p interfaceMethodParser) parseFindOperation(tokens []string) (Operation, e
return nil, err return nil, err
} }
querySpec, err := parseQuery(tokens, 1) queryTokens, sortTokens := p.splitQueryAndSortTokens(tokens)
querySpec, err := parseQuery(queryTokens, 1)
if err != nil {
return nil, err
}
sorts, err := p.parseSort(sortTokens)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -127,9 +136,65 @@ func (p interfaceMethodParser) parseFindOperation(tokens []string) (Operation, e
return FindOperation{ return FindOperation{
Mode: mode, Mode: mode,
Query: querySpec, Query: querySpec,
Sorts: sorts,
}, nil }, nil
} }
func (p interfaceMethodParser) parseSort(rawTokens []string) ([]Sort, error) {
if len(rawTokens) == 0 {
return nil, nil
}
sortTokens := rawTokens[2:]
var sorts []Sort
var aggregatedToken sortToken
for _, token := range sortTokens {
if token != "And" {
aggregatedToken = append(aggregatedToken, token)
} else if len(aggregatedToken) == 0 {
return nil, NewInvalidSortError(rawTokens)
} else {
sorts = append(sorts, aggregatedToken.ToSort())
aggregatedToken = sortToken{}
}
}
if len(aggregatedToken) == 0 {
return nil, NewInvalidSortError(rawTokens)
}
sorts = append(sorts, aggregatedToken.ToSort())
return sorts, nil
}
type sortToken []string
func (t sortToken) ToSort() Sort {
if len(t) > 1 && t[len(t)-1] == "Asc" {
return Sort{FieldName: strings.Join(t[:len(t)-1], ""), Ordering: OrderingAscending}
}
if len(t) > 1 && t[len(t)-1] == "Desc" {
return Sort{FieldName: strings.Join(t[:len(t)-1], ""), Ordering: OrderingDescending}
}
return Sort{FieldName: strings.Join(t, ""), Ordering: OrderingAscending}
}
func (p interfaceMethodParser) splitQueryAndSortTokens(tokens []string) ([]string, []string) {
var queryTokens []string
var sortTokens []string
for i, token := range tokens {
if len(tokens) > i && token == "Order" && tokens[i+1] == "By" {
sortTokens = tokens[i:]
break
} else {
queryTokens = append(queryTokens, token)
}
}
return queryTokens, sortTokens
}
func (p interfaceMethodParser) extractModelOrSliceReturns(returns []code.Type) (QueryMode, error) { func (p interfaceMethodParser) extractModelOrSliceReturns(returns []code.Type) (QueryMode, error) {
if len(returns) != 2 { if len(returns) != 2 {
return "", UnsupportedReturnError return "", UnsupportedReturnError

View file

@ -429,6 +429,99 @@ func TestParseInterfaceMethod_Find(t *testing.T) {
}}, }},
}, },
}, },
{
Name: "FindByArgOrderByArg method",
Method: code.Method{
Name: "FindByCityOrderByAge",
Params: []code.Param{
{Type: code.ExternalType{PackageAlias: "context", Name: "Context"}},
{Type: code.SimpleType("string")},
},
Returns: []code.Type{
code.ArrayType{ContainedType: code.PointerType{ContainedType: code.SimpleType("UserModel")}},
code.SimpleType("error"),
},
},
ExpectedOperation: spec.FindOperation{
Mode: spec.QueryModeMany,
Query: spec.QuerySpec{Predicates: []spec.Predicate{
{Field: "City", Comparator: spec.ComparatorEqual, ParamIndex: 1},
}},
Sorts: []spec.Sort{
{FieldName: "Age", Ordering: spec.OrderingAscending},
},
},
},
{
Name: "FindByArgOrderByArgAsc method",
Method: code.Method{
Name: "FindByCityOrderByAgeAsc",
Params: []code.Param{
{Type: code.ExternalType{PackageAlias: "context", Name: "Context"}},
{Type: code.SimpleType("string")},
},
Returns: []code.Type{
code.ArrayType{ContainedType: code.PointerType{ContainedType: code.SimpleType("UserModel")}},
code.SimpleType("error"),
},
},
ExpectedOperation: spec.FindOperation{
Mode: spec.QueryModeMany,
Query: spec.QuerySpec{Predicates: []spec.Predicate{
{Field: "City", Comparator: spec.ComparatorEqual, ParamIndex: 1},
}},
Sorts: []spec.Sort{
{FieldName: "Age", Ordering: spec.OrderingAscending},
},
},
},
{
Name: "FindByArgOrderByArgDesc method",
Method: code.Method{
Name: "FindByCityOrderByAgeDesc",
Params: []code.Param{
{Type: code.ExternalType{PackageAlias: "context", Name: "Context"}},
{Type: code.SimpleType("string")},
},
Returns: []code.Type{
code.ArrayType{ContainedType: code.PointerType{ContainedType: code.SimpleType("UserModel")}},
code.SimpleType("error"),
},
},
ExpectedOperation: spec.FindOperation{
Mode: spec.QueryModeMany,
Query: spec.QuerySpec{Predicates: []spec.Predicate{
{Field: "City", Comparator: spec.ComparatorEqual, ParamIndex: 1},
}},
Sorts: []spec.Sort{
{FieldName: "Age", Ordering: spec.OrderingDescending},
},
},
},
{
Name: "FindByArgOrderByArgAndArg method",
Method: code.Method{
Name: "FindByCityOrderByCityAndAgeDesc",
Params: []code.Param{
{Type: code.ExternalType{PackageAlias: "context", Name: "Context"}},
{Type: code.SimpleType("string")},
},
Returns: []code.Type{
code.ArrayType{ContainedType: code.PointerType{ContainedType: code.SimpleType("UserModel")}},
code.SimpleType("error"),
},
},
ExpectedOperation: spec.FindOperation{
Mode: spec.QueryModeMany,
Query: spec.QuerySpec{Predicates: []spec.Predicate{
{Field: "City", Comparator: spec.ComparatorEqual, ParamIndex: 1},
}},
Sorts: []spec.Sort{
{FieldName: "City", Ordering: spec.OrderingAscending},
{FieldName: "Age", Ordering: spec.OrderingDescending},
},
},
},
} }
for _, testCase := range testTable { for _, testCase := range testTable {
@ -1248,6 +1341,39 @@ func TestParseInterfaceMethod_Find_Invalid(t *testing.T) {
}, },
ExpectedError: spec.InvalidParamError, ExpectedError: spec.InvalidParamError,
}, },
{
Name: "misplaced operator token (leftmost)",
Method: code.Method{
Name: "FindAllOrderByAndAge",
Returns: []code.Type{
code.ArrayType{ContainedType: code.PointerType{ContainedType: code.SimpleType("UserModel")}},
code.SimpleType("error"),
},
},
ExpectedError: spec.NewInvalidSortError([]string{"Order", "By", "And", "Age"}),
},
{
Name: "misplaced operator token (rightmost)",
Method: code.Method{
Name: "FindAllOrderByAgeAnd",
Returns: []code.Type{
code.ArrayType{ContainedType: code.PointerType{ContainedType: code.SimpleType("UserModel")}},
code.SimpleType("error"),
},
},
ExpectedError: spec.NewInvalidSortError([]string{"Order", "By", "Age", "And"}),
},
{
Name: "misplaced operator token (double operator)",
Method: code.Method{
Name: "FindAllOrderByAgeAndAndGender",
Returns: []code.Type{
code.ArrayType{ContainedType: code.PointerType{ContainedType: code.SimpleType("UserModel")}},
code.SimpleType("error"),
},
},
ExpectedError: spec.NewInvalidSortError([]string{"Order", "By", "Age", "And", "And", "Gender"}),
},
} }
for _, testCase := range testTable { for _, testCase := range testTable {