mirror of
https://github.com/beego/bee.git
synced 2024-11-25 09:50:55 +00:00
addAll now supports: filter(query), fields, sortby, order, limit, offset
This commit is contained in:
parent
7a4d5f2798
commit
7fa01c437c
95
g_models.go
95
g_models.go
@ -626,11 +626,55 @@ func Get{{modelName}}ById(id int) (v *{{modelName}}, err error) {
|
|||||||
|
|
||||||
// GetAll{{modelName}} retrieves all {{modelName}} matches certain condition. Returns empty list if
|
// GetAll{{modelName}} retrieves all {{modelName}} matches certain condition. Returns empty list if
|
||||||
// no records exist
|
// no records exist
|
||||||
func GetAll{{modelName}}() (l []{{modelName}}, err error) {
|
func GetAll{{modelName}}(query map[string]string, fields []string, sortby []string, order []string,
|
||||||
|
offset int64, limit int64) (l []{{modelName}}, err error) {
|
||||||
o := orm.NewOrm()
|
o := orm.NewOrm()
|
||||||
t := new({{modelName}})
|
qs := o.QueryTable(new({{modelName}}))
|
||||||
qs := o.QueryTable(t)
|
// query k=v
|
||||||
if _, err := qs.All(&l); err == nil {
|
for k, v := range query {
|
||||||
|
qs = qs.Filter(k, v)
|
||||||
|
}
|
||||||
|
// order by:
|
||||||
|
var sortFields []string
|
||||||
|
if len(sortby) != 0 {
|
||||||
|
if len(sortby) == len(order) {
|
||||||
|
// 1) for each sort field, there is an associated order
|
||||||
|
for i, v := range sortby {
|
||||||
|
orderby := ""
|
||||||
|
if order[i] == "desc" {
|
||||||
|
orderby = "-" + v
|
||||||
|
} else if order[i] == "asc" {
|
||||||
|
orderby = v
|
||||||
|
} else {
|
||||||
|
return nil, errors.New("Error: Invalid order. Must be either [asc|desc]")
|
||||||
|
}
|
||||||
|
sortFields = append(sortFields, orderby)
|
||||||
|
}
|
||||||
|
qs = qs.OrderBy(sortFields...)
|
||||||
|
} else if len(sortby) != len(order) && len(order) == 1 {
|
||||||
|
// 2) there is exactly one order, all the sorted fields will be sorted by this order
|
||||||
|
for _, v := range sortby {
|
||||||
|
orderby := ""
|
||||||
|
if order[0] == "desc" {
|
||||||
|
orderby = "-" + v
|
||||||
|
} else if order[0] == "asc" {
|
||||||
|
orderby = v
|
||||||
|
} else {
|
||||||
|
return nil, errors.New("Error: Invalid order. Must be either [asc|desc]")
|
||||||
|
}
|
||||||
|
sortFields = append(sortFields, orderby)
|
||||||
|
}
|
||||||
|
} else if len(sortby) != len(order) && len(order) != 1 {
|
||||||
|
return nil, errors.New("Error: 'sortby', 'order' sizes mismatch or 'order' size is not 1")
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if len(order) != 0 {
|
||||||
|
return nil, errors.New("Error: unused 'order' fields")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
qs = qs.OrderBy(sortFields...)
|
||||||
|
if _, err := qs.Limit(limit, offset).All(&l, fields...); err == nil {
|
||||||
return l, nil
|
return l, nil
|
||||||
}
|
}
|
||||||
return nil, err
|
return nil, err
|
||||||
@ -715,7 +759,48 @@ func (this *{{ctrlName}}Controller) GetOne() {
|
|||||||
// @Failure 403
|
// @Failure 403
|
||||||
// @router / [get]
|
// @router / [get]
|
||||||
func (this *{{ctrlName}}Controller) GetAll() {
|
func (this *{{ctrlName}}Controller) GetAll() {
|
||||||
l, err := models.GetAll{{ctrlName}}()
|
var fields []string
|
||||||
|
var sortby []string
|
||||||
|
var order []string
|
||||||
|
var query map[string]string = make(map[string]string)
|
||||||
|
var limit int64 = 10
|
||||||
|
var offset int64 = 0
|
||||||
|
|
||||||
|
// fields: col1,col2,entity.col3
|
||||||
|
if v := this.GetString("fields"); v != "" {
|
||||||
|
fields = strings.Split(v, ",")
|
||||||
|
}
|
||||||
|
// limit: 10 (default is 10)
|
||||||
|
if v, err := this.GetInt("limit"); err == nil {
|
||||||
|
limit = v
|
||||||
|
}
|
||||||
|
// offset: 0 (default is 0)
|
||||||
|
if v, err := this.GetInt("offset"); err == nil {
|
||||||
|
offset = v
|
||||||
|
}
|
||||||
|
// sortby: col1,col2
|
||||||
|
if v := this.GetString("sortby"); v != "" {
|
||||||
|
sortby = strings.Split(v, ",")
|
||||||
|
}
|
||||||
|
// order: desc,asc
|
||||||
|
if v := this.GetString("order"); v != "" {
|
||||||
|
order = strings.Split(v, ",")
|
||||||
|
}
|
||||||
|
// query: k:v,k:v
|
||||||
|
if v := this.GetString("query"); v != "" {
|
||||||
|
for _, cond := range strings.Split(v, ",") {
|
||||||
|
kv := strings.Split(cond, ":")
|
||||||
|
if len(kv) != 2 {
|
||||||
|
this.Data["json"] = errors.New("Error: invalid query key/value pair")
|
||||||
|
this.ServeJson()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
k, v := kv[0], kv[1]
|
||||||
|
query[k] = v
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
l, err := models.GetAll{{ctrlName}}(query, fields, sortby, order, offset, limit)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
this.Data["json"] = err.Error()
|
this.Data["json"] = err.Error()
|
||||||
} else {
|
} else {
|
||||||
|
Loading…
Reference in New Issue
Block a user