gin/logger.go

84 lines
1.8 KiB
Go
Raw Normal View History

2014-06-18 03:42:34 +04:00
package gin
import (
"log"
"os"
2014-06-18 03:42:34 +04:00
"time"
)
2014-06-30 05:59:21 +04:00
func ErrorLogger() HandlerFunc {
2014-07-08 02:16:41 +04:00
return ErrorLoggerT(ErrorTypeAll)
}
func ErrorLoggerT(typ uint32) HandlerFunc {
2014-06-30 05:59:21 +04:00
return func(c *Context) {
c.Next()
2014-07-08 02:16:41 +04:00
errs := c.Errors.ByType(typ)
if len(errs) > 0 {
// -1 status code = do not change current one
c.JSON(-1, c.Errors)
}
2014-06-30 05:59:21 +04:00
}
}
var (
green = string([]byte{27, 91, 57, 55, 59, 52, 50, 109})
white = string([]byte{27, 91, 57, 48, 59, 52, 55, 109})
yellow = string([]byte{27, 91, 57, 55, 59, 52, 51, 109})
red = string([]byte{27, 91, 57, 55, 59, 52, 49, 109})
reset = string([]byte{27, 91, 48, 109})
)
2014-06-18 03:42:34 +04:00
func Logger() HandlerFunc {
2014-07-06 20:26:40 +04:00
stdlogger := log.New(os.Stdout, "", 0)
//errlogger := log.New(os.Stderr, "", 0)
2014-06-18 03:42:34 +04:00
return func(c *Context) {
// Start timer
start := time.Now()
2014-06-18 03:42:34 +04:00
// Process request
c.Next()
2014-07-06 20:26:40 +04:00
// save the IP of the requester
requester := c.Request.Header.Get("X-Real-IP")
2014-07-06 20:26:40 +04:00
// if the requester-header is empty, check the forwarded-header
if requester == "" {
requester = c.Request.Header.Get("X-Forwarded-For")
2014-07-06 20:26:40 +04:00
}
// if the requester is still empty, use the hard-coded address from the socket
if requester == "" {
requester = c.Request.RemoteAddr
2014-07-06 20:26:40 +04:00
}
var color string
code := c.Writer.Status()
switch {
case code >= 200 && code <= 299:
color = green
case code >= 300 && code <= 399:
color = white
case code >= 400 && code <= 499:
color = yellow
default:
color = red
}
2014-07-07 05:04:06 +04:00
end := time.Now()
latency := end.Sub(start)
2014-07-06 23:09:23 +04:00
stdlogger.Printf("[GIN] %v |%s %3d %s| %12v | %s %4s %s\n",
2014-07-07 05:04:06 +04:00
end.Format("2006/01/02 - 15:04:05"),
2014-07-08 16:07:59 +04:00
color, code, reset,
latency,
2014-07-06 20:26:40 +04:00
requester,
c.Request.Method, c.Request.URL.Path,
)
2014-06-18 03:42:34 +04:00
// Calculate resolution time
if len(c.Errors) > 0 {
2014-07-06 20:26:40 +04:00
stdlogger.Println(c.Errors.String())
}
2014-06-18 03:42:34 +04:00
}
}