Rewrite: Use middleware.Replacer for simple rule.

This commit is contained in:
Abiola Ibrahim 2015-11-24 12:20:39 +01:00
parent 0b51369932
commit 016344bae7
2 changed files with 7 additions and 1 deletions

View file

@ -52,7 +52,11 @@ func (s SimpleRule) Rewrite(r *http.Request) bool {
// take note of this rewrite for internal use by fastcgi
// all we need is the URI, not full URL
r.Header.Set(headerFieldName, r.URL.RequestURI())
r.URL.Path = s.To
// replace variables
to := path.Clean(middleware.NewReplacer(r, nil, "").Replace(s.To))
r.URL.Path = to
return true
}
return false

View file

@ -17,6 +17,7 @@ func TestRewrite(t *testing.T) {
Rules: []Rule{
NewSimpleRule("/from", "/to"),
NewSimpleRule("/a", "/b"),
NewSimpleRule("/b", "/b{uri}"),
},
}
@ -50,6 +51,7 @@ func TestRewrite(t *testing.T) {
}{
{"/from", "/to"},
{"/a", "/b"},
{"/b", "/b/b"},
{"/aa", "/aa"},
{"/", "/"},
{"/a?foo=bar", "/b?foo=bar"},