/
dump.go
72 lines (59 loc) · 1.48 KB
/
dump.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
package echox
import (
"bytes"
"io"
"net/http"
"net/http/httputil"
"time"
"github.com/labstack/echo/v4"
"go.uber.org/zap"
"github.com/xinpianchang/xservice/pkg/log"
)
type bodyDumpResponseWriter struct {
io.Writer
http.ResponseWriter
}
func (w *bodyDumpResponseWriter) Write(b []byte) (int, error) {
return w.Writer.Write(b)
}
func Dump(filename ...string) echo.MiddlewareFunc {
var logger log.Logger
if len(filename) > 0 {
logger, _ = log.NewLogger(filename[0])
}
return func(next echo.HandlerFunc) echo.HandlerFunc {
return func(c echo.Context) error {
var l log.Logger
if logger == nil {
l = log.For(c.Request().Context())
} else {
l = logger.For(c.Request().Context())
}
requestId := c.Response().Header().Get(echo.HeaderXRequestID)
req, _ := httputil.DumpRequest(c.Request(), true)
start := time.Now()
l.Debug(
requestId,
zap.Any("url", c.Request().URL),
zap.String("req", string(req)),
)
resBody := new(bytes.Buffer)
mw := io.MultiWriter(c.Response().Writer, resBody)
writer := &bodyDumpResponseWriter{Writer: mw, ResponseWriter: c.Response().Writer}
c.Response().Writer = writer
err := next(c)
if err != nil {
l.Error(requestId, zap.Error(err))
return err
}
l.Debug(
requestId,
zap.Int64("cost", time.Since(start).Milliseconds()),
zap.Int("status", c.Response().Status),
zap.Any("header", c.Response().Header()),
zap.String("rsp", resBody.String()),
)
return nil
}
}
}