Add TrimTrailingSlashes option

This commit is contained in:
Nise Void 2020-05-19 17:04:00 +02:00
parent 5b5a102c71
commit 66ae2a435c
Signed by: NiseVoid
GPG Key ID: FBA14AC83EA602F3
1 changed files with 14 additions and 1 deletions

View File

@ -38,6 +38,7 @@ type Router struct {
NotFoundHandler Handle
MethodNotAllowedHandler Handle
ErrorHandler ErrorHandle
TrimTrailingSlashes bool
server *http.Server
}
@ -127,7 +128,7 @@ func (r *Router) Stop() error {
return err
}
func (r *Router) getHttpr() *httprouter.Router {
func (r *Router) getHttpr() http.Handler {
httpr := httprouter.New()
for _, v := range r.routes {
@ -156,6 +157,18 @@ func (r *Router) getHttpr() *httprouter.Router {
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
}