-
Notifications
You must be signed in to change notification settings - Fork 0
/
gin_gorm.go
186 lines (144 loc) · 5.14 KB
/
gin_gorm.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
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
package golog
import (
"bytes"
"context"
"github.com/gin-gonic/gin"
"go.dtapp.net/gojson"
"go.dtapp.net/gorequest"
"go.dtapp.net/gotime"
"go.dtapp.net/gourl"
"go.opentelemetry.io/otel/attribute"
"io"
"net/http"
"time"
)
// GinLogFunc Gin框架日志函数
type GinLogFunc func(ctx context.Context, response *GormGinLogModel)
// GinGorm 框架日志
type GinGorm struct {
ginLogFunc GinLogFunc // Gin框架日志函数
}
// GinGormFun *GinGorm 框架日志驱动
type GinGormFun func() *GinGorm
// NewGinGorm 创建Gin框架实例
func NewGinGorm(ctx context.Context) (*GinGorm, error) {
gg := &GinGorm{}
return gg, nil
}
// 定义一个自定义的 ResponseWriter
type ginGormBodyWriter struct {
gin.ResponseWriter
body *bytes.Buffer
}
// 实现 http.ResponseWriter 的 Write 方法
func (w ginGormBodyWriter) Write(b []byte) (int, error) {
w.body.Write(b)
return w.ResponseWriter.Write(b)
}
// WriteString 实现 http.ResponseWriter 的 WriteString 方法
func (w ginGormBodyWriter) WriteString(s string) (int, error) {
w.body.WriteString(s)
return w.ResponseWriter.WriteString(s)
}
// WriteHeader 实现 http.ResponseWriter 的 WriteHeader 方法
func (w ginGormBodyWriter) WriteHeader(statusCode int) {
w.ResponseWriter.WriteHeader(statusCode)
}
// Header 实现 http.ResponseWriter 的 Header 方法
func (w ginGormBodyWriter) Header() http.Header {
return w.ResponseWriter.Header()
}
// Middleware 中间件
func (gg *GinGorm) Middleware() gin.HandlerFunc {
return func(g *gin.Context) {
// OpenTelemetry链路追踪
//g.Request = g.Request.WithContext(gg.TraceStartSpan(g))
ctx, span := TraceStartSpan(g.Request.Context(), "gin")
// 开始时间
start := time.Now().UTC()
// 模型
var log = GormGinLogModel{}
// 请求时间
log.RequestTime = gotime.Current().Time
// Read the Body content
var bodyBytes []byte
if g.Request.Body != nil {
bodyBytes, _ = io.ReadAll(g.Request.Body)
}
// 将io.ReadCloser恢复到其原始状态
g.Request.Body = io.NopCloser(bytes.NewBuffer(bodyBytes))
// 创建自定义的 ResponseWriter 并替换原有的
blw := &ginGormBodyWriter{
ResponseWriter: g.Writer,
body: bytes.NewBufferString(""),
}
g.Writer = blw
// 处理请求
g.Next()
// 结束时间
end := time.Now().UTC()
// 请求消耗时长
log.RequestCostTime = end.Sub(start).Milliseconds()
// 响应时间
log.ResponseTime = gotime.Current().Time
// 跟踪编号
log.TraceID = gorequest.TraceSpanGetTraceID(span)
// 请求编号
log.RequestID = gorequest.GetRequestIDContext(ctx)
// 请求主机
log.RequestHost = g.Request.Host
// 请求地址
log.RequestPath = gourl.UriFilterExcludeQueryString(g.Request.RequestURI)
// 请求参数
log.RequestQuery = gojson.JsonEncodeNoError(gojson.ParseQueryString(g.Request.RequestURI))
// 请求方式
log.RequestMethod = g.Request.Method
// 请求协议
log.RequestScheme = g.Request.Proto
// 请求类型
log.RequestContentType = g.ContentType()
// 请求IP
log.RequestClientIP = g.ClientIP()
// 请求UA
log.RequestUserAgent = g.Request.UserAgent()
// 请求头
log.RequestHeader = gojson.JsonEncodeNoError(g.Request.Header)
// 响应头
log.ResponseHeader = gojson.JsonEncodeNoError(blw.Header())
// 响应状态
log.ResponseStatusCode = g.Writer.Status()
// 响应内容
if gojson.IsValidJSON(blw.body.String()) {
log.ResponseBody = gojson.JsonEncodeNoError(gojson.JsonDecodeNoError(blw.body.String()))
} else {
log.ResponseBody = blw.body.String()
}
// OpenTelemetry链路追踪
span.SetAttributes(attribute.String("request.id", log.RequestID))
span.SetAttributes(attribute.String("request.time", log.RequestTime.Format(gotime.DateTimeFormat)))
span.SetAttributes(attribute.String("request.host", log.RequestHost))
span.SetAttributes(attribute.String("request.path", log.RequestPath))
span.SetAttributes(attribute.String("request.query", log.RequestQuery))
span.SetAttributes(attribute.String("request.method", log.RequestMethod))
span.SetAttributes(attribute.String("request.scheme", log.RequestScheme))
span.SetAttributes(attribute.String("request.content_type", log.RequestContentType))
span.SetAttributes(attribute.String("request.body", log.RequestBody))
span.SetAttributes(attribute.String("request.client_ip", log.RequestClientIP))
span.SetAttributes(attribute.String("request.user_agent", log.RequestUserAgent))
span.SetAttributes(attribute.String("request.header", log.RequestHeader))
span.SetAttributes(attribute.Int64("request.cost_time", log.RequestCostTime))
span.SetAttributes(attribute.String("response.time", log.ResponseTime.Format(gotime.DateTimeFormat)))
span.SetAttributes(attribute.String("response.header", log.ResponseHeader))
span.SetAttributes(attribute.Int("response.status_code", log.ResponseStatusCode))
span.SetAttributes(attribute.String("response.body", log.ResponseBody))
// 调用Gin框架日志函数
if gg.ginLogFunc != nil {
gg.ginLogFunc(ctx, &log)
}
span.End() // 结束OpenTelemetry链路追踪
}
}
// SetLogFunc 设置日志记录方法
func (gg *GinGorm) SetLogFunc(ginLogFunc GinLogFunc) {
gg.ginLogFunc = ginLogFunc
}