"github.com/gorilla/mux"
"log"
"net/http"
+ "strings"
)
type LoggingResponseWriter struct {
Status int
Length int
http.ResponseWriter
+ ResponseBody string
}
func (loggingWriter *LoggingResponseWriter) WriteHeader(code int) {
}
func (loggingWriter *LoggingResponseWriter) Write(data []byte) (int, error) {
- loggingWriter.Length = len(data)
+ loggingWriter.Length += len(data)
+ if loggingWriter.Status >= 400 {
+ loggingWriter.ResponseBody += string(data)
+ }
return loggingWriter.ResponseWriter.Write(data)
}
}
func (loggingRouter *LoggingRESTRouter) ServeHTTP(resp http.ResponseWriter, req *http.Request) {
- loggingWriter := LoggingResponseWriter{200, 0, resp}
+ loggingWriter := LoggingResponseWriter{200, 0, resp, ""}
loggingRouter.router.ServeHTTP(&loggingWriter, req)
- if loggingWriter.Status == 200 {
- log.Printf("[%s] %s %s %d %d", req.RemoteAddr, req.Method, req.URL.Path[1:], loggingWriter.Status, loggingWriter.Length)
- } else {
- log.Printf("[%s] %s %s %d", req.RemoteAddr, req.Method, req.URL.Path[1:], loggingWriter.Status)
+ statusText := "OK"
+ if loggingWriter.Status >= 400 {
+ statusText = strings.Replace(loggingWriter.ResponseBody, "\n", "", -1)
}
+ log.Printf("[%s] %s %s %d %d \"%s\"", req.RemoteAddr, req.Method, req.URL.Path[1:], loggingWriter.Status, loggingWriter.Length, statusText)
+
}