support raw order

This commit is contained in:
Anker Jam 2020-10-26 00:01:15 +08:00
parent 56fa213a6e
commit 7d4e88c1b9
3 changed files with 34 additions and 16 deletions

View File

@ -432,21 +432,24 @@ func (t *dbTables) getOrderSQL(orders []*clauses.Order) (orderSQL string) {
orderSqls := make([]string, 0, len(orders)) orderSqls := make([]string, 0, len(orders))
for _, order := range orders { for _, order := range orders {
column := order.GetColumn() column := order.GetColumn()
var sort string clause := strings.Split(column, clauses.ExprDot)
switch order.GetSort() {
case clauses.SortAscending:
sort = "ASC"
case clauses.SortDescending:
sort = "DESC"
}
clause := strings.Split(column, ExprSep)
index, _, fi, suc := t.parseExprs(t.mi, clause) if order.IsRaw() {
if !suc { if len(clause) == 2 {
panic(fmt.Errorf("unknown field/column name `%s`", strings.Join(clause, ExprSep))) orderSqls = append(orderSqls, fmt.Sprintf("%s.%s%s%s %s", clause[0], Q, clause[1], Q, order.SortString()))
} } else if len(clause) == 1 {
orderSqls = append(orderSqls, fmt.Sprintf("%s%s%s %s", Q, clause[0], Q, order.SortString()))
} else {
panic(fmt.Errorf("unknown field/column name `%s`", strings.Join(clause, ExprSep)))
}
} else {
index, _, fi, suc := t.parseExprs(t.mi, clause)
if !suc {
panic(fmt.Errorf("unknown field/column name `%s`", strings.Join(clause, ExprSep)))
}
orderSqls = append(orderSqls, fmt.Sprintf("%s.%s%s%s %s", index, Q, fi.column, Q, sort)) orderSqls = append(orderSqls, fmt.Sprintf("%s.%s%s%s %s", index, Q, fi.column, Q, order.SortString()))
}
} }
orderSQL = fmt.Sprintf("ORDER BY %s ", strings.Join(orderSqls, ", ")) orderSQL = fmt.Sprintf("ORDER BY %s ", strings.Join(orderSqls, ", "))

View File

@ -2,4 +2,5 @@ package clauses
const ( const (
ExprSep = "__" ExprSep = "__"
ExprDot = "."
) )

View File

@ -1,5 +1,7 @@
package clauses package clauses
import "strings"
type Sort int8 type Sort int8
const ( const (
@ -33,6 +35,18 @@ func (o *Order) GetSort() Sort {
return o.sort return o.sort
} }
func (o *Order) SortString() string {
switch o.GetSort() {
case SortAscending:
return "ASC"
case SortDescending:
return "DESC"
}
return ``
}
func (o *Order) IsRaw() bool { func (o *Order) IsRaw() bool {
return o.isRaw return o.isRaw
} }
@ -41,10 +55,10 @@ func ParseOrder(expressions ...string) []*Order {
var orders []*Order var orders []*Order
for _, expression := range expressions { for _, expression := range expressions {
sort := SortAscending sort := SortAscending
column := expression column := strings.ReplaceAll(expression, ExprSep, ExprDot)
if expression[0] == '-' { if column[0] == '-' {
sort = SortDescending sort = SortDescending
column = expression[1:] column = column[1:]
} }
orders = append(orders, &Order{ orders = append(orders, &Order{