-
Notifications
You must be signed in to change notification settings - Fork 0
/
logger.go
51 lines (43 loc) · 1.15 KB
/
logger.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
package stdserver
import (
"os"
"sync"
"time"
"github.com/gofiber/fiber/v2"
"github.com/sirupsen/logrus"
)
func newLoggerMiddleware(cfg *Settings) fiber.Handler {
log := cfg.Logger.WithField("module", "fiber")
pid := os.Getpid()
var once sync.Once
var errHandler fiber.ErrorHandler
return func(c *fiber.Ctx) error {
once.Do(func() {
errHandler = c.App().Config().ErrorHandler
})
start := time.Now()
chainErr := c.Next()
if chainErr != nil {
if err := errHandler(c, chainErr); err != nil {
_ = c.SendStatus(fiber.StatusInternalServerError)
}
}
stop := time.Now()
fields := make(logrus.Fields)
fields["pid"] = pid
fields["remote"] = c.IP()
if user := c.Locals("user"); user != nil {
fields["user"] = user
}
fields["status"] = c.Response().StatusCode()
fields["length"] = len(c.Response().Body())
fields["latency"] = stop.Sub(start).Milliseconds()
fields["userAgent"] = c.Get(fiber.HeaderUserAgent)
if chainErr != nil {
log.WithFields(fields).WithError(chainErr).Errorf("%s %s", c.Method(), c.OriginalURL())
} else {
log.WithFields(fields).Infof("%s %s", c.Method(), c.OriginalURL())
}
return nil
}
}