mirror of
https://github.com/rancher/norman.git
synced 2025-06-28 08:17:14 +00:00
109 lines
2.1 KiB
Go
109 lines
2.1 KiB
Go
package parse
|
|
|
|
import (
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/rancher/norman/types"
|
|
)
|
|
|
|
var (
|
|
defaultLimit = int64(1000)
|
|
maxLimit = int64(3000)
|
|
)
|
|
|
|
func QueryOptions(apiContext *types.APIContext, schema *types.Schema) types.QueryOptions {
|
|
req := apiContext.Request
|
|
if req.Method != http.MethodGet {
|
|
return types.QueryOptions{}
|
|
}
|
|
|
|
result := &types.QueryOptions{}
|
|
|
|
result.Sort = parseSort(schema, req)
|
|
result.Pagination = parsePagination(req)
|
|
result.Conditions = parseFilters(schema, req)
|
|
|
|
return *result
|
|
}
|
|
|
|
func parseOrder(req *http.Request) types.SortOrder {
|
|
order := req.URL.Query().Get("order")
|
|
if types.SortOrder(order) == types.DESC {
|
|
return types.DESC
|
|
}
|
|
return types.ASC
|
|
}
|
|
|
|
func parseSort(schema *types.Schema, req *http.Request) types.Sort {
|
|
sortField := req.URL.Query().Get("sort")
|
|
if _, ok := schema.CollectionFilters[sortField]; !ok {
|
|
sortField = ""
|
|
}
|
|
return types.Sort{
|
|
Order: parseOrder(req),
|
|
Name: sortField,
|
|
}
|
|
}
|
|
|
|
func parsePagination(req *http.Request) *types.Pagination {
|
|
q := req.URL.Query()
|
|
limit := q.Get("limit")
|
|
marker := q.Get("marker")
|
|
|
|
result := &types.Pagination{
|
|
Limit: &defaultLimit,
|
|
Marker: marker,
|
|
}
|
|
|
|
if limit != "" {
|
|
limitInt, err := strconv.ParseInt(limit, 10, 64)
|
|
if err != nil {
|
|
return result
|
|
}
|
|
|
|
if limitInt > maxLimit {
|
|
result.Limit = &maxLimit
|
|
} else if limitInt >= 0 {
|
|
result.Limit = &limitInt
|
|
}
|
|
}
|
|
|
|
return result
|
|
}
|
|
|
|
func parseNameAndOp(value string) (string, types.ModifierType) {
|
|
name := value
|
|
op := "eq"
|
|
|
|
idx := strings.LastIndex(value, "_")
|
|
if idx > 0 {
|
|
op = value[idx+1:]
|
|
name = value[0:idx]
|
|
}
|
|
|
|
return name, types.ModifierType(op)
|
|
}
|
|
|
|
func parseFilters(schema *types.Schema, req *http.Request) []*types.QueryCondition {
|
|
var conditions []*types.QueryCondition
|
|
for key, values := range req.URL.Query() {
|
|
name, op := parseNameAndOp(key)
|
|
filter, ok := schema.CollectionFilters[name]
|
|
if !ok {
|
|
continue
|
|
}
|
|
|
|
for _, mod := range filter.Modifiers {
|
|
if op != mod || !types.ValidMod(op) {
|
|
continue
|
|
}
|
|
|
|
conditions = append(conditions, types.NewConditionFromString(name, mod, values...))
|
|
}
|
|
}
|
|
|
|
return conditions
|
|
}
|