Merge branch 'master' into wallet-store-interface-account

[bytom/vapor.git] / vendor / github.com / gin-gonic / gin / logger.go

diff --git a/vendor/github.com/gin-gonic/gin/logger.go b/vendor/github.com/gin-gonic/gin/logger.go
new file mode 100644 (file)
index 0000000..1a8df60
--- /dev/null
+++ b/vendor/github.com/gin-gonic/gin/logger.go
@@ -0,0 +1,152 @@
+// 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.
+
+package gin
+
+import (
+       "fmt"
+       "io"
+       "net/http"
+       "os"
+       "time"
+
+       "github.com/mattn/go-isatty"
+)
+
+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
+)
+
+// DisableConsoleColor disables color output in the console.
+func DisableConsoleColor() {
+       disableColor = true
+}
+
+// ErrorLogger returns a handlerfunc for any error type.
+func ErrorLogger() HandlerFunc {
+       return ErrorLoggerT(ErrorTypeAny)
+}
+
+// ErrorLoggerT returns a handlerfunc for a given error type.
+func ErrorLoggerT(typ ErrorType) HandlerFunc {
+       return func(c *Context) {
+               c.Next()
+               errors := c.Errors.ByType(typ)
+               if len(errors) > 0 {
+                       c.JSON(-1, errors)
+               }
+       }
+}
+
+// Logger instances a Logger middleware that will write the logs to gin.DefaultWriter.
+// By default gin.DefaultWriter = os.Stdout.
+func Logger() HandlerFunc {
+       return LoggerWithWriter(DefaultWriter)
+}
+
+// LoggerWithWriter instance a Logger middleware with the specified writter buffer.
+// 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 ||
+               (os.Getenv("TERM") == "dumb" || (!isatty.IsTerminal(w.Fd()) && !isatty.IsCygwinTerminal(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{}{}
+               }
+       }
+
+       return func(c *Context) {
+               // Start timer
+               start := time.Now()
+               path := c.Request.URL.Path
+               raw := c.Request.URL.RawQuery
+
+               // 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, resetColor string
+                       if isTerm {
+                               statusColor = colorForStatus(statusCode)
+                               methodColor = colorForMethod(method)
+                               resetColor = reset
+                       }
+                       comment := c.Errors.ByType(ErrorTypePrivate).String()
+
+                       if raw != "" {
+                               path = path + "?" + raw
+                       }
+
+                       fmt.Fprintf(out, "[GIN] %v |%s %3d %s| %13v | %15s |%s %-7s %s %s\n%s",
+                               end.Format("2006/01/02 - 15:04:05"),
+                               statusColor, statusCode, resetColor,
+                               latency,
+                               clientIP,
+                               methodColor, method, resetColor,
+                               path,
+                               comment,
+                       )
+               }
+       }
+}
+
+func colorForStatus(code int) string {
+       switch {
+       case code >= http.StatusOK && code < http.StatusMultipleChoices:
+               return green
+       case code >= http.StatusMultipleChoices && code < http.StatusBadRequest:
+               return white
+       case code >= http.StatusBadRequest && code < http.StatusInternalServerError:
+               return yellow
+       default:
+               return red
+       }
+}
+
+func colorForMethod(method string) string {
+       switch method {
+       case "GET":
+               return blue
+       case "POST":
+               return cyan
+       case "PUT":
+               return yellow
+       case "DELETE":
+               return red
+       case "PATCH":
+               return green
+       case "HEAD":
+               return magenta
+       case "OPTIONS":
+               return white
+       default:
+               return reset
+       }
+}