forked from Fuyu/router
Add TrimTrailingSlashes option
This commit is contained in:
parent
5b5a102c71
commit
66ae2a435c
15
router.go
15
router.go
@ -38,6 +38,7 @@ type Router struct {
|
|||||||
NotFoundHandler Handle
|
NotFoundHandler Handle
|
||||||
MethodNotAllowedHandler Handle
|
MethodNotAllowedHandler Handle
|
||||||
ErrorHandler ErrorHandle
|
ErrorHandler ErrorHandle
|
||||||
|
TrimTrailingSlashes bool
|
||||||
server *http.Server
|
server *http.Server
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -127,7 +128,7 @@ func (r *Router) Stop() error {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *Router) getHttpr() *httprouter.Router {
|
func (r *Router) getHttpr() http.Handler {
|
||||||
httpr := httprouter.New()
|
httpr := httprouter.New()
|
||||||
|
|
||||||
for _, v := range r.routes {
|
for _, v := range r.routes {
|
||||||
@ -156,6 +157,18 @@ func (r *Router) getHttpr() *httprouter.Router {
|
|||||||
r.ErrorHandler(c, err)
|
r.ErrorHandler(c, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if r.TrimTrailingSlashes {
|
||||||
|
httpr.RedirectTrailingSlash = false
|
||||||
|
|
||||||
|
return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
|
||||||
|
l := len(req.URL.Path)
|
||||||
|
if l > 1 && req.URL.Path[l-1] == '/' {
|
||||||
|
req.URL.Path = req.URL.Path[:l-1]
|
||||||
|
}
|
||||||
|
httpr.ServeHTTP(w, req)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
return httpr
|
return httpr
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user