zhangdeman/gopkg
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

增加gin middleware

Browse Source
This commit is contained in:
白茶清欢 2022-01-13 21:53:05 +08:00
parent 0af7fe03da
commit 409120f77d
1 changed files with 27 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
gin/middleware/init_request.go
View File

@ -19,12 +19,36 @@ import (
// Author : go_developer@163.com<白茶清欢>
//
// Date : 5:53 下午 2021/3/9
func InitRequest() func(ctx *gin.Context) {
func InitRequest(startTimeField string, traceIDField string) gin.HandlerFunc {
return func(ctx *gin.Context) {
// 设置请求开始时间
ctx.Set("start_time", time.Now().Unix())
if len(startTimeField) == 0 {
startTimeField = "start_time"
}
if len(traceIDField) == 0 {
traceIDField = "trace_id"
}
ctx.Set(startTimeField, time.Now().Unix())
// 设置请求trace_id
ctx.Set("trace_id", time.Now().Format("20060102150405")+"+"+commonUtil.GetHostIP()+"-"+commonUtil.Md5(commonUtil.GenRandomString("", 16)))
ctx.Set(traceIDField, commonUtil.GetHostIP()+"-"+ctx.ClientIP()+"-"+time.Now().Format("20060102150405")+"-"+commonUtil.Md5(commonUtil.GenRandomString("", 16)))
ctx.Next()
}
}
// SupportCross 支持跨域
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 9:40 PM 2022/1/13
func SupportCross(header map[string]string) gin.HandlerFunc {
return func(ctx *gin.Context) {
ctx.Writer.Header().Set("Access-Control-Allow-Origin", "*")
ctx.Header("Access-Control-Allow-Headers", "Content-Type,AccessToken,X-CSRF-Token, Authorization, Token, Admin-User-Token, admin-user-token")
ctx.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE,UPDATE") //服务器支持的所有跨域请求的方
ctx.Header("Access-Control-Expose-Headers", "admin-user-token, Admin-User-Token, Content-Length, Access-Control-Allow-Origin, Access-Control-Allow-Headers, Content-Type")
for key, val := range header {
ctx.Header(key, val)
}
ctx.Next()
}
}