40 lines
808 B
Go
40 lines
808 B
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
)
|
|
|
|
type loggableResponseWriter struct {
|
|
StatusCode int
|
|
|
|
next http.ResponseWriter
|
|
}
|
|
|
|
func (w *loggableResponseWriter) Header() http.Header {
|
|
return w.next.Header()
|
|
}
|
|
|
|
func (w *loggableResponseWriter) Write(b []byte) (int, error) {
|
|
return w.next.Write(b)
|
|
}
|
|
|
|
func (w *loggableResponseWriter) WriteHeader(statusCode int) {
|
|
w.StatusCode = statusCode
|
|
w.next.WriteHeader(statusCode)
|
|
}
|
|
|
|
func (w *loggableResponseWriter) Flush() {
|
|
if flusher, ok := w.next.(http.Flusher); ok {
|
|
flusher.Flush()
|
|
}
|
|
}
|
|
|
|
func logHandler(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
lw := &loggableResponseWriter{next: w}
|
|
next.ServeHTTP(lw, r)
|
|
log.Printf("%s %s %s %d", r.RemoteAddr, r.Method, r.URL, lw.StatusCode)
|
|
})
|
|
}
|