2014-08-29 21:49:50 +04: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-18 03:42:34 +04:00
|
|
|
package gin
|
|
|
|
|
|
|
|
import (
|
|
|
|
"log"
|
2014-07-04 02:25:12 +04:00
|
|
|
"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-06-30 06:04:45 +04:00
|
|
|
|
2014-07-08 02:16:41 +04:00
|
|
|
errs := c.Errors.ByType(typ)
|
|
|
|
if len(errs) > 0 {
|
2014-06-30 06:04:45 +04:00
|
|
|
// -1 status code = do not change current one
|
|
|
|
c.JSON(-1, c.Errors)
|
|
|
|
}
|
2014-06-30 05:59:21 +04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-07-04 02:25:12 +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
|
2014-07-04 02:25:12 +04:00
|
|
|
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
|
2014-07-08 18:57:04 +04:00
|
|
|
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
|
2014-08-18 06:52:01 +04:00
|
|
|
if len(requester) == 0 {
|
2014-07-08 18:57:04 +04:00
|
|
|
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
|
2014-08-18 06:52:01 +04:00
|
|
|
if len(requester) == 0 {
|
2014-07-08 18:57:04 +04:00
|
|
|
requester = c.Request.RemoteAddr
|
2014-07-06 20:26:40 +04:00
|
|
|
}
|
|
|
|
|
2014-07-04 02:25:12 +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-13 02:18:33 +04:00
|
|
|
stdlogger.Printf("[GIN] %v |%s %3d %s| %12v | %s %4s %s\n%s",
|
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,
|
2014-07-04 02:25:12 +04:00
|
|
|
latency,
|
2014-07-06 20:26:40 +04:00
|
|
|
requester,
|
2014-07-08 18:57:04 +04:00
|
|
|
c.Request.Method, c.Request.URL.Path,
|
2014-07-13 02:18:33 +04:00
|
|
|
c.Errors.String(),
|
2014-07-04 02:25:12 +04:00
|
|
|
)
|
2014-06-18 03:42:34 +04:00
|
|
|
}
|
|
|
|
}
|