51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
package query_parser
|
|
|
|
import (
|
|
"strings"
|
|
)
|
|
|
|
type OrderByHandler func(field string, desc bool)
|
|
|
|
// ParseOrderByString 解析排序字符串，调用处理函数。
|
|
func ParseOrderByString(orderBy string, handler OrderByHandler) error {
|
|
if orderBy == "" {
|
|
return nil
|
|
}
|
|
|
|
parts := strings.Split(orderBy, ",")
|
|
for _, part := range parts {
|
|
part = strings.TrimSpace(part)
|
|
ParseOrderByField(part, handler)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// ParseOrderByStrings 解析多个排序字符串，调用处理函数。
|
|
func ParseOrderByStrings(orderBys []string, handler OrderByHandler) error {
|
|
for _, v := range orderBys {
|
|
if v == "" {
|
|
continue
|
|
}
|
|
|
|
ParseOrderByField(v, handler)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// ParseOrderByField 解析单个排序字段，调用处理函数。
|
|
func ParseOrderByField(orderBy string, handler OrderByHandler) {
|
|
orderBy = strings.TrimSpace(orderBy)
|
|
if orderBy == "" {
|
|
return // 没有排序条件
|
|
}
|
|
|
|
if strings.HasPrefix(orderBy, "-") {
|
|
handler(orderBy[1:], true) // 降序
|
|
} else if strings.HasPrefix(orderBy, "+") {
|
|
handler(orderBy[1:], false) // 升序
|
|
} else {
|
|
handler(orderBy, false) // 升序
|
|
}
|
|
}
|