You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
2.0 KiB
93 lines
2.0 KiB
package middleware
|
|
|
|
import (
|
|
"bytes"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/zenazn/goji/web"
|
|
)
|
|
|
|
// Logger is a middleware that logs the start and end of each request, along
|
|
// with some useful data about what was requested, what the response status was,
|
|
// and how long it took to return. When standard output is a TTY, Logger will
|
|
// print in color, otherwise it will print in black and white.
|
|
//
|
|
// Logger prints a request ID if one is provided.
|
|
func Logger(c *web.C, h http.Handler) http.Handler {
|
|
fn := func(w http.ResponseWriter, r *http.Request) {
|
|
reqId := GetReqId(*c)
|
|
|
|
printStart(reqId, r)
|
|
|
|
lw := wrapWriter(w)
|
|
|
|
t1 := time.Now()
|
|
h.ServeHTTP(lw, r)
|
|
lw.maybeWriteHeader()
|
|
t2 := time.Now()
|
|
|
|
printEnd(reqId, lw, t2.Sub(t1))
|
|
}
|
|
|
|
return http.HandlerFunc(fn)
|
|
}
|
|
|
|
func printStart(reqId string, r *http.Request) {
|
|
var buf bytes.Buffer
|
|
|
|
if reqId != "" {
|
|
cW(&buf, bBlack, "[%s] ", reqId)
|
|
}
|
|
buf.WriteString("Started ")
|
|
cW(&buf, bMagenta, "%s ", r.Method)
|
|
cW(&buf, nBlue, "%q ", r.URL.String())
|
|
buf.WriteString("from ")
|
|
buf.WriteString(r.RemoteAddr)
|
|
|
|
log.Print(buf.String())
|
|
}
|
|
|
|
func printEnd(reqId string, w writerProxy, dt time.Duration) {
|
|
var buf bytes.Buffer
|
|
|
|
if reqId != "" {
|
|
cW(&buf, bBlack, "[%s] ", reqId)
|
|
}
|
|
buf.WriteString("Returning ")
|
|
if w.status() < 200 {
|
|
cW(&buf, bBlue, "%03d", w.status())
|
|
} else if w.status() < 300 {
|
|
cW(&buf, bGreen, "%03d", w.status())
|
|
} else if w.status() < 400 {
|
|
cW(&buf, bCyan, "%03d", w.status())
|
|
} else if w.status() < 500 {
|
|
cW(&buf, bYellow, "%03d", w.status())
|
|
} else {
|
|
cW(&buf, bRed, "%03d", w.status())
|
|
}
|
|
buf.WriteString(" in ")
|
|
if dt < 500*time.Millisecond {
|
|
cW(&buf, nGreen, "%s", dt)
|
|
} else if dt < 5*time.Second {
|
|
cW(&buf, nYellow, "%s", dt)
|
|
} else {
|
|
cW(&buf, nRed, "%s", dt)
|
|
}
|
|
|
|
log.Print(buf.String())
|
|
}
|
|
|
|
func wrapWriter(w http.ResponseWriter) writerProxy {
|
|
_, cn := w.(http.CloseNotifier)
|
|
_, fl := w.(http.Flusher)
|
|
_, hj := w.(http.Hijacker)
|
|
|
|
bw := basicWriter{ResponseWriter: w}
|
|
if cn && fl && hj {
|
|
return &fancyWriter{bw}
|
|
} else {
|
|
return &bw
|
|
}
|
|
}
|