New {request} placeholder to log entire requests (sans body) (#871)

Add a {request} placeholder to the replacer.

Closes #858.
This commit is contained in:
Gustavo Chaín 2016-06-07 19:06:24 +02:00 committed by Matt Holt
parent 01e05afa0c
commit 6c847d0723
2 changed files with 20 additions and 0 deletions

View File

@ -3,6 +3,7 @@ package httpserver
import ( import (
"net" "net"
"net/http" "net/http"
"net/http/httputil"
"net/url" "net/url"
"os" "os"
"path" "path"
@ -11,6 +12,14 @@ import (
"time" "time"
) )
// requestReplacer is a strings.Replacer which is used to
// encode literal \r and \n characters and keep everything
// on one line
var requestReplacer = strings.NewReplacer(
"\r", "\\r",
"\n", "\\n",
)
// Replacer is a type which can replace placeholder // Replacer is a type which can replace placeholder
// substrings in a string with actual values from a // substrings in a string with actual values from a
// http.Request and ResponseRecorder. Always use // http.Request and ResponseRecorder. Always use
@ -95,6 +104,14 @@ func NewReplacer(r *http.Request, rr *ResponseRecorder, emptyValue string) Repla
dir, _ := path.Split(r.URL.Path) dir, _ := path.Split(r.URL.Path)
return dir return dir
}(), }(),
"{request}": func() string {
dump, err := httputil.DumpRequest(r, false)
if err != nil {
return ""
}
return requestReplacer.Replace(string(dump))
}(),
}, },
emptyValue: emptyValue, emptyValue: emptyValue,
} }

View File

@ -74,6 +74,9 @@ func TestReplace(t *testing.T) {
if expected, actual := "The Custom header is foobarbaz.", repl.Replace("The Custom header is {>Custom}."); expected != actual { if expected, actual := "The Custom header is foobarbaz.", repl.Replace("The Custom header is {>Custom}."); expected != actual {
t.Errorf("{>Custom} replacement: expected '%s', got '%s'", expected, actual) t.Errorf("{>Custom} replacement: expected '%s', got '%s'", expected, actual)
} }
if expected, actual := "The request is POST / HTTP/1.1\\r\\nHost: localhost\\r\\nCustom: foobarbaz\\r\\nShorterval: 1\\r\\n\\r\\n.", repl.Replace("The request is {request}."); expected != actual {
t.Errorf("{request} replacement: expected '%s', got '%s'", expected, actual)
}
// Test header case-insensitivity // Test header case-insensitivity
if expected, actual := "The cUsToM header is foobarbaz...", repl.Replace("The cUsToM header is {>cUsToM}..."); expected != actual { if expected, actual := "The cUsToM header is foobarbaz...", repl.Replace("The cUsToM header is {>cUsToM}..."); expected != actual {