-
Notifications
You must be signed in to change notification settings - Fork 350
/
query.go
107 lines (86 loc) · 2.3 KB
/
query.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
/*
Package source implements a custom predicate to match routes
based on the Query Params in URL
It supports checking existence of query params and also checking whether
query params value match to a given regular exp
Examples:
// Checking existence of a query param
// matches http://example.org?bb=a&query=withvalue
example1: QueryParam("query") -> "http://example.org";
// Even a query param without a value
// matches http://example.org?bb=a&query=
example1: QueryParam("query") -> "http://example.org";
// matches with regexp
// matches http://example.org?bb=a&query=example
example1: QueryParam("query", "^example$") -> "http://example.org";
// matches with regexp and multiple values of query param
// matches http://example.org?bb=a&query=testing&query=example
example1: QueryParam("query", "^example$") -> "http://example.org";
*/
package query
import (
"net/http"
"regexp"
"github.com/zalando/skipper/predicates"
"github.com/zalando/skipper/routing"
)
type matchType int
const (
exists matchType = iota + 1
matches
)
type predicate struct {
typ matchType
paramName string
valueExp *regexp.Regexp
}
type spec struct{}
// New creates a new QueryParam predicate specification.
func New() routing.PredicateSpec { return &spec{} }
func (s *spec) Name() string {
return predicates.QueryParamName
}
func (s *spec) Create(args []interface{}) (routing.Predicate, error) {
if len(args) == 0 || len(args) > 2 {
return nil, predicates.ErrInvalidPredicateParameters
}
name, ok1 := args[0].(string)
switch {
case !ok1:
return nil, predicates.ErrInvalidPredicateParameters
case len(args) == 1:
return &predicate{exists, name, nil}, nil
case len(args) == 2:
value, ok2 := args[1].(string)
if !ok2 {
return nil, predicates.ErrInvalidPredicateParameters
}
valueExp, err := regexp.Compile(value)
if err != nil {
return nil, err
}
return &predicate{matches, name, valueExp}, nil
default:
return nil, predicates.ErrInvalidPredicateParameters
}
}
func (p *predicate) Match(r *http.Request) bool {
queryMap := r.URL.Query()
vals, ok := queryMap[p.paramName]
switch p.typ {
case exists:
return ok
case matches:
if !ok {
return false
} else {
for _, v := range vals {
if p.valueExp.MatchString(v) {
return true
}
}
return false
}
}
return false
}