-
Notifications
You must be signed in to change notification settings - Fork 0
/
http_logger.go
106 lines (94 loc) · 2.39 KB
/
http_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
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
package transport
import (
"net/http"
"time"
"github.com/sirupsen/logrus"
"github.com/y-miyazaki/go-common/pkg/logger"
"go.uber.org/zap"
)
// HTTPLoggerType defines the transport type.
type HTTPLoggerType string
const (
// HTTPLoggerTypeExternal defines the external type.
HTTPLoggerTypeExternal HTTPLoggerType = "external"
// HTTPLoggerTypeInternal defines the internal type.
HTTPLoggerTypeInternal HTTPLoggerType = "internal"
)
// HTTPLogger struct.
type HTTPLogger struct {
http.RoundTripper
logger *logger.Logger
Type HTTPLoggerType
}
// HTTPZapLogger struct.
type HTTPZapLogger struct {
http.RoundTripper
logger *logger.ZapLogger
Type HTTPLoggerType
}
// NewTransportHTTPLogger get http.RoundTripper.
func NewTransportHTTPLogger(
l *logger.Logger,
transportType HTTPLoggerType,
) http.RoundTripper {
return &HTTPLogger{
http.DefaultTransport,
l,
transportType,
}
}
// NewTransportHTTPZapLogger get http.RoundTripper.
func NewTransportHTTPZapLogger(
l *logger.ZapLogger,
transportType HTTPLoggerType,
) http.RoundTripper {
return &HTTPZapLogger{
http.DefaultTransport,
l,
transportType,
}
}
// RoundTrip logs transparently.
func (t HTTPLogger) RoundTrip(req *http.Request) (*http.Response, error) {
timeBefore := time.Now()
response, err := t.RoundTripper.RoundTrip(req)
timeAfter := time.Now()
log := t.logger.WithFields(logrus.Fields{
"url": req.URL.String(),
"method": req.Method,
"protocol": req.Proto,
"duration": timeAfter.Sub(timeBefore).String(),
"transportType": t.Type,
})
if response != nil {
log = log.WithField("status", response.StatusCode)
}
if err != nil || response.StatusCode/100 >= 4 {
log.WithError(err).Error()
} else {
log.Info()
}
return response, err
}
// RoundTrip logs transparently.
func (t HTTPZapLogger) RoundTrip(req *http.Request) (*http.Response, error) {
timeBefore := time.Now()
response, err := t.RoundTripper.RoundTrip(req)
timeAfter := time.Now()
log := t.logger.With(
zap.String("url", req.URL.String()),
zap.String("method", req.Method),
zap.String("protocol", req.Proto),
zap.String("duration", timeAfter.Sub(timeBefore).String()),
zap.Any("transportType", t.Type),
)
if response != nil {
log = log.With(zap.Int("status", response.StatusCode))
}
if err != nil || response.StatusCode/100 >= 4 {
log.WithError(err).Error("")
} else {
log.Info("")
}
return response, err
}