wberube
/
echoip
mirror of https://github.com/mpolden/echoip
1
0
Fork 0
Code Issues Releases Wiki Activity

Only allow a single matcher per route

This commit is contained in:
Martin Polden 2018-03-19 19:44:14 +01:00
parent 055496906d
commit b01bddb63e
1 changed files with 10 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
http/router.go
View File

@ -10,11 +10,11 @@ type router struct {
}
type route struct {
method string
path string
prefix bool
matcherFuncs []func(*http.Request) bool
handler appHandler
method string
path string
prefix bool
handler appHandler
matcherFunc func(*http.Request) bool
}
func NewRouter() *router {
@ -48,15 +48,14 @@ func (r *router) Handler() http.Handler {
})
}
func (r *route) Header(header, value string) *route {
return r.MatcherFunc(func(req *http.Request) bool {
func (r *route) Header(header, value string) {
r.MatcherFunc(func(req *http.Request) bool {
return req.Header.Get(header) == value
})
}
func (r *route) MatcherFunc(f func(*http.Request) bool) *route {
r.matcherFuncs = append(r.matcherFuncs, f)
return r
func (r *route) MatcherFunc(f func(*http.Request) bool) {
r.matcherFunc = f
}
func (r *route) match(req *http.Request) bool {
@ -70,11 +69,5 @@ func (r *route) match(req *http.Request) bool {
} else if r.path != req.URL.Path {
return false
}
match := len(r.matcherFuncs) == 0
for _, f := range r.matcherFuncs {
if match = f(req); match {
break
}
}
return match
return r.matcherFunc == nil || r.matcherFunc(req)
}