-
Notifications
You must be signed in to change notification settings - Fork 3
/
zap.go
131 lines (102 loc) · 3.17 KB
/
zap.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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
package zap
import (
"net/http"
"strings"
"time"
shttp "github.com/alexfalkowski/go-service/http"
"github.com/alexfalkowski/go-service/meta"
stime "github.com/alexfalkowski/go-service/time"
tm "github.com/alexfalkowski/go-service/transport/meta"
tstrings "github.com/alexfalkowski/go-service/transport/strings"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
)
const (
service = "http"
)
// NewHandler for zap.
func NewHandler(logger *zap.Logger) *Handler {
return &Handler{logger: logger}
}
// Handler for zap.
type Handler struct {
logger *zap.Logger
}
// ServeHTTP or zap.
func (h *Handler) ServeHTTP(resp http.ResponseWriter, req *http.Request, next http.HandlerFunc) {
path, method := req.URL.Path, strings.ToLower(req.Method)
if tstrings.IsHealth(path) {
next(resp, req)
return
}
start := time.Now()
ctx := req.Context()
res := &shttp.ResponseWriter{ResponseWriter: resp, StatusCode: http.StatusOK}
next(res, req)
fields := []zapcore.Field{
zap.Int64(tm.DurationKey, stime.ToMilliseconds(time.Since(start))),
zap.String(tm.StartTimeKey, start.Format(time.RFC3339)),
zap.String(tm.ServiceKey, service),
zap.String(tm.PathKey, path),
zap.String(tm.MethodKey, method),
}
for k, v := range meta.Attributes(ctx) {
fields = append(fields, zap.String(k, v))
}
if d, ok := ctx.Deadline(); ok {
fields = append(fields, zap.String(tm.DeadlineKey, d.Format(time.RFC3339)))
}
fields = append(fields, zap.Int(tm.CodeKey, res.StatusCode))
loggerLevel := codeToLevel(res.StatusCode, h.logger)
loggerLevel("finished call with code "+res.Status(), fields...)
}
// NewRoundTripper for zap.
func NewRoundTripper(logger *zap.Logger, r http.RoundTripper) *RoundTripper {
return &RoundTripper{logger: logger, RoundTripper: r}
}
// RoundTripper for zap.
type RoundTripper struct {
logger *zap.Logger
http.RoundTripper
}
// RoundTrip for zap.
func (r *RoundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
if tstrings.IsHealth(req.URL.String()) {
return r.RoundTripper.RoundTrip(req)
}
path, method := req.URL.Path, strings.ToLower(req.Method)
start := time.Now()
ctx := req.Context()
resp, err := r.RoundTripper.RoundTrip(req)
fields := []zapcore.Field{
zap.Int64(tm.DurationKey, stime.ToMilliseconds(time.Since(start))),
zap.String(tm.StartTimeKey, start.Format(time.RFC3339)),
zap.String(tm.ServiceKey, service),
zap.String(tm.PathKey, path),
zap.String(tm.MethodKey, method),
}
for k, v := range meta.Attributes(ctx) {
fields = append(fields, zap.String(k, v))
}
if d, ok := ctx.Deadline(); ok {
fields = append(fields, zap.String(tm.DeadlineKey, d.Format(time.RFC3339)))
}
if err != nil {
fields = append(fields, zap.Error(err))
r.logger.Error("finished call with error", fields...)
return nil, err
}
fields = append(fields, zap.Int(tm.CodeKey, resp.StatusCode))
loggerLevel := codeToLevel(resp.StatusCode, r.logger)
loggerLevel("finished call with code "+resp.Status, fields...)
return resp, nil
}
func codeToLevel(code int, logger *zap.Logger) func(msg string, fields ...zapcore.Field) {
if code >= 400 && code <= 499 {
return logger.Warn
}
if code >= 500 && code <= 599 {
return logger.Error
}
return logger.Info
}