]>
Commit | Line | Data |
---|---|---|
7a9e5112 | 1 | package api |
2 | ||
3 | import ( | |
372ed740 | 4 | "fmt" |
372ed740 | 5 | "time" |
6 | ||
7 | "github.com/Sirupsen/logrus" | |
7a9e5112 | 8 | "github.com/gin-gonic/gin" |
9 | "github.com/jloup/utils" | |
10 | ) | |
11 | ||
12 | var log = utils.StandardL().WithField("module", "api") | |
13 | ||
14 | func Logger() gin.HandlerFunc { | |
15 | return func(c *gin.Context) { | |
16 | path := c.Request.URL.Path | |
372ed740 | 17 | start := time.Now() |
7a9e5112 | 18 | rawQuery := c.Request.URL.RawQuery |
19 | ||
20 | c.Next() | |
21 | ||
372ed740 | 22 | latency := time.Now().Sub(start).Round(10 * time.Microsecond) |
23 | code := c.Writer.Status() | |
24 | ||
25 | l := log | |
26 | l = l.WithField("latency", latency) | |
27 | l = l.WithField("client_ip", c.ClientIP()) | |
28 | l = l.WithField("method", c.Request.Method) | |
29 | l = l.WithField("status_code", code) | |
30 | ||
31 | if _, exists := c.Get("user"); exists { | |
32 | l = l.WithField("user_id", GetUser(c).Id) | |
33 | } | |
34 | ||
35 | if rawQuery != "" { | |
36 | path = fmt.Sprintf("%s?%s", path, rawQuery) | |
37 | } | |
38 | ||
39 | l = l.WithField("path", path) | |
7a9e5112 | 40 | |
372ed740 | 41 | msgLog := fmt.Sprintf("[%v] %d %s '%s'", latency, code, c.Request.Method, path) |
7a9e5112 | 42 | |
372ed740 | 43 | var level logrus.Level |
44 | switch { | |
45 | case code >= 200 && code < 400: | |
4059b21b | 46 | level = logrus.DebugLevel |
372ed740 | 47 | case code >= 400 && code < 500: |
4059b21b | 48 | level = logrus.InfoLevel |
372ed740 | 49 | default: |
50 | level = logrus.ErrorLevel | |
7a9e5112 | 51 | } |
372ed740 | 52 | |
ecb0fb69 | 53 | errors := c.Errors.ByType(gin.ErrorTypePrivate) |
4495b984 | 54 | |
55 | for _, err := range errors { | |
4059b21b | 56 | l.WithField("description", err.Err).Logf(level, "%s: %s", msgLog, err.Err) |
372ed740 | 57 | } |
58 | ||
4495b984 | 59 | if errors == nil { |
60 | l.Logf(level, msgLog) | |
61 | } | |
7a9e5112 | 62 | } |
63 | } |