gin/logger.go

141 lines
3.1 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-04-07 16:37:17 +00:00
"fmt"
"io"
"os"
2014-06-17 23:42:34 +00:00
"time"
"github.com/mattn/go-isatty"
2014-06-17 23:42:34 +00:00
)
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})
disableColor = false
2014-10-08 23:40:42 +00:00
)
func DisableConsoleColor() {
disableColor = true
}
2014-06-30 01:59:21 +00:00
func ErrorLogger() HandlerFunc {
2015-05-12 13:22:13 +00:00
return ErrorLoggerT(ErrorTypeAny)
2014-07-07 22:16:41 +00:00
}
2015-05-22 23:59:36 +00:00
func ErrorLoggerT(typ ErrorType) HandlerFunc {
2014-06-30 01:59:21 +00:00
return func(c *Context) {
c.Next()
errors := c.Errors.ByType(typ)
if len(errors) > 0 {
c.JSON(-1, errors)
}
2014-06-30 01:59:21 +00:00
}
}
2016-04-14 23:16:46 +00:00
// Logger instances a Logger middleware that will write the logs to gin.DefaultWriter
2015-05-29 19:03:41 +00:00
// By default gin.DefaultWriter = os.Stdout
2014-06-17 23:42:34 +00:00
func Logger() HandlerFunc {
2015-05-12 13:22:13 +00:00
return LoggerWithWriter(DefaultWriter)
2015-04-07 16:37:17 +00:00
}
2014-07-06 16:26:40 +00:00
2016-04-14 23:16:46 +00:00
// LoggerWithWriter instance a Logger middleware with the specified writter buffer.
2015-05-29 19:03:41 +00:00
// Example: os.Stdout, a file opened in write mode, a socket...
func LoggerWithWriter(out io.Writer, notlogged ...string) HandlerFunc {
isTerm := true
if w, ok := out.(*os.File); !ok || !isatty.IsTerminal(w.Fd()) || disableColor {
isTerm = false
}
var skip map[string]struct{}
if length := len(notlogged); length > 0 {
skip = make(map[string]struct{}, length)
for _, path := range notlogged {
skip[path] = struct{}{}
}
}
2014-06-17 23:42:34 +00:00
return func(c *Context) {
// Start timer
start := time.Now()
path := c.Request.URL.Path
2014-06-17 23:42:34 +00:00
// Process request
c.Next()
// Log only when path is not being skipped
if _, ok := skip[path]; !ok {
// Stop timer
end := time.Now()
latency := end.Sub(start)
clientIP := c.ClientIP()
method := c.Request.Method
statusCode := c.Writer.Status()
var statusColor, methodColor string
if isTerm {
statusColor = colorForStatus(statusCode)
methodColor = colorForMethod(method)
}
comment := c.Errors.ByType(ErrorTypePrivate).String()
fmt.Fprintf(out, "[GIN] %v |%s %3d %s| %13v | %s |%s %s %-7s %s\n%s",
end.Format("2006/01/02 - 15:04:05"),
statusColor, statusCode, reset,
latency,
clientIP,
methodColor, method, reset,
path,
comment,
)
}
2014-06-17 23:42:34 +00:00
}
}
2014-10-08 23:40:42 +00:00
func colorForStatus(code int) string {
switch {
2015-05-09 01:35:31 +00:00
case code >= 200 && code < 300:
2014-10-08 23:40:42 +00:00
return green
2015-05-09 01:35:31 +00:00
case code >= 300 && code < 400:
2014-10-08 23:40:42 +00:00
return white
2015-05-09 01:35:31 +00:00
case code >= 400 && code < 500:
2014-10-08 23:40:42 +00:00
return yellow
default:
return red
}
}
func colorForMethod(method string) string {
2015-05-09 01:35:31 +00:00
switch method {
case "GET":
2014-10-08 23:40:42 +00:00
return blue
2015-05-09 01:35:31 +00:00
case "POST":
2014-10-08 23:40:42 +00:00
return cyan
2015-05-09 01:35:31 +00:00
case "PUT":
2014-10-08 23:40:42 +00:00
return yellow
2015-05-09 01:35:31 +00:00
case "DELETE":
2014-10-08 23:40:42 +00:00
return red
2015-05-09 01:35:31 +00:00
case "PATCH":
2014-10-08 23:40:42 +00:00
return green
2015-05-09 01:35:31 +00:00
case "HEAD":
2014-10-08 23:40:42 +00:00
return magenta
2015-05-09 01:35:31 +00:00
case "OPTIONS":
2014-10-08 23:40:42 +00:00
return white
default:
return reset
}
}