gin/logger.go

106 lines
2.3 KiB
Go
Raw Normal View History

2014-08-29 17:49:50 +00:00
// Copyright 2014 Manu Martinez-Almeida. All rights reserved.
// Use of this source code is governed by a MIT style
// license that can be found in the LICENSE file.
2014-06-17 23:42:34 +00:00
package gin
import (
2015-03-08 13:19:50 +00:00
"github.com/mattn/go-colorable"
2014-06-17 23:42:34 +00:00
"log"
"time"
)
2014-10-08 23:40:42 +00: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})
blue = string([]byte{27, 91, 57, 55, 59, 52, 52, 109})
magenta = string([]byte{27, 91, 57, 55, 59, 52, 53, 109})
cyan = string([]byte{27, 91, 57, 55, 59, 52, 54, 109})
reset = string([]byte{27, 91, 48, 109})
)
2014-06-30 01:59:21 +00:00
func ErrorLogger() HandlerFunc {
2014-07-07 22:16:41 +00:00
return ErrorLoggerT(ErrorTypeAll)
}
func ErrorLoggerT(typ uint32) HandlerFunc {
2014-06-30 01:59:21 +00:00
return func(c *Context) {
c.Next()
2014-07-07 22:16:41 +00: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 01:59:21 +00:00
}
}
2014-06-17 23:42:34 +00:00
func Logger() HandlerFunc {
2015-03-04 04:14:10 +00:00
stdlogger := log.New(colorable.NewColorableStdout(), "", 0)
2014-07-06 16:26:40 +00:00
//errlogger := log.New(os.Stderr, "", 0)
2014-06-17 23:42:34 +00:00
return func(c *Context) {
// Start timer
start := time.Now()
2014-06-17 23:42:34 +00:00
// Process request
c.Next()
2014-10-08 23:40:42 +00:00
// Stop timer
2014-07-07 01:04:06 +00:00
end := time.Now()
latency := end.Sub(start)
2014-10-08 23:40:42 +00:00
clientIP := c.ClientIP()
method := c.Request.Method
statusCode := c.Writer.Status()
statusColor := colorForStatus(statusCode)
methodColor := colorForMethod(method)
stdlogger.Printf("[GIN] %v |%s %3d %s| %12v | %s |%s %s %-7s %s\n%s",
2014-07-07 01:04:06 +00:00
end.Format("2006/01/02 - 15:04:05"),
2014-10-08 23:40:42 +00:00
statusColor, statusCode, reset,
latency,
2014-10-08 23:40:42 +00:00
clientIP,
methodColor, reset, method,
c.Request.URL.Path,
c.Errors.String(),
)
2014-06-17 23:42:34 +00:00
}
}
2014-10-08 23:40:42 +00:00
func colorForStatus(code int) string {
switch {
case code >= 200 && code <= 299:
return green
case code >= 300 && code <= 399:
return white
case code >= 400 && code <= 499:
return yellow
default:
return red
}
}
func colorForMethod(method string) string {
switch {
case method == "GET":
return blue
case method == "POST":
return cyan
case method == "PUT":
return yellow
case method == "DELETE":
return red
case method == "PATCH":
return green
case method == "HEAD":
return magenta
case method == "OPTIONS":
return white
default:
return reset
}
}