traefik/old/middlewares/replace_path.go

29 lines
728 B
Go
Raw Normal View History

2017-04-25 18:13:39 +00:00
package middlewares
import (
"context"
2017-04-25 18:13:39 +00:00
"net/http"
)
const (
// ReplacePathKey is the key within the request context used to
// store the replaced path
ReplacePathKey key = "ReplacePath"
// ReplacedPathHeader is the default header to set the old path to
ReplacedPathHeader = "X-Replaced-Path"
)
2017-04-25 18:13:39 +00:00
// ReplacePath is a middleware used to replace the path of a URL request
type ReplacePath struct {
Handler http.Handler
Path string
}
func (s *ReplacePath) ServeHTTP(w http.ResponseWriter, r *http.Request) {
r = r.WithContext(context.WithValue(r.Context(), ReplacePathKey, r.URL.Path))
2017-04-25 18:13:39 +00:00
r.Header.Add(ReplacedPathHeader, r.URL.Path)
r.URL.Path = s.Path
r.RequestURI = r.URL.RequestURI()
2017-04-25 18:13:39 +00:00
s.Handler.ServeHTTP(w, r)
}