108 lines
2.6 KiB
Go
108 lines
2.6 KiB
Go
|
package middleware
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"bytes"
|
|||
|
|
"gitee.ltd/lxh/logger/log"
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
jsoniter "github.com/json-iterator/go"
|
|||
|
|
"io"
|
|||
|
|
"regexp"
|
|||
|
|
"strings"
|
|||
|
|
"time"
|
|||
|
|
"wireguard-dashboard/model/entity"
|
|||
|
|
"wireguard-dashboard/repository"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// bodyWriter
|
|||
|
|
// @description: 重写ResponseBody
|
|||
|
|
type bodyWriter struct {
|
|||
|
|
gin.ResponseWriter
|
|||
|
|
body *bytes.Buffer
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func SystemLogRequest() gin.HandlerFunc {
|
|||
|
|
return func(c *gin.Context) {
|
|||
|
|
var userId string
|
|||
|
|
if userInfo, ok := c.Get("user"); ok {
|
|||
|
|
userId = userInfo.(*entity.User).Id
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 开始时间
|
|||
|
|
start := time.Now()
|
|||
|
|
host := c.Request.Host // 请求域名
|
|||
|
|
path := c.Request.URL.Path // 接口地址
|
|||
|
|
query := c.Request.URL.RawQuery // 参数
|
|||
|
|
if strings.Contains(path, "/api/dashboard/list") {
|
|||
|
|
c.Next()
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var bodyStr string
|
|||
|
|
if !strings.Contains(path, "/api/login") {
|
|||
|
|
body, err := c.GetRawData() // body参数
|
|||
|
|
if err == nil {
|
|||
|
|
bodyStr = string(body)
|
|||
|
|
c.Request.Body = io.NopCloser(bytes.NewBuffer(body))
|
|||
|
|
reg := regexp.MustCompile("\\s+")
|
|||
|
|
bodyStr = reg.ReplaceAllString(bodyStr, "")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
method := c.Request.Method // 请求方式
|
|||
|
|
ip := c.ClientIP() // 取出IP
|
|||
|
|
// 处理实际客户端IP
|
|||
|
|
if c.Request.Header.Get("U-Real-Ip") != "" {
|
|||
|
|
ip = c.Request.Header.Get("U-Real-Ip") // 这个是网关Nginx自定义的Header头
|
|||
|
|
} else if c.Request.Header.Get("U-Forwarded-For") != "" {
|
|||
|
|
ip = c.Request.Header.Get("U-Forwarded-For") // 这个是网关Nginx自定义的Header头
|
|||
|
|
}
|
|||
|
|
ua := c.Request.UserAgent() // UA
|
|||
|
|
|
|||
|
|
// 重写客户端IP
|
|||
|
|
c.Request.Header.Set("X-Forwarded-For", ip)
|
|||
|
|
c.Request.Header.Set("X-Real-Ip", ip)
|
|||
|
|
|
|||
|
|
// 拦截response
|
|||
|
|
bw := &bodyWriter{body: bytes.NewBufferString(""), ResponseWriter: c.Writer}
|
|||
|
|
c.Writer = bw
|
|||
|
|
|
|||
|
|
header := c.Request.Header
|
|||
|
|
headerStr, _ := jsoniter.MarshalToString(header)
|
|||
|
|
|
|||
|
|
// 执行下一步请求
|
|||
|
|
c.Next()
|
|||
|
|
// 计算耗时
|
|||
|
|
cost := time.Since(start).Milliseconds()
|
|||
|
|
|
|||
|
|
// 组装实体
|
|||
|
|
l := entity.SystemLog{
|
|||
|
|
UserId: userId,
|
|||
|
|
ClientIP: ip,
|
|||
|
|
Host: host,
|
|||
|
|
Method: method,
|
|||
|
|
Uri: path,
|
|||
|
|
Header: headerStr,
|
|||
|
|
Body: bodyStr,
|
|||
|
|
Form: "",
|
|||
|
|
Query: query,
|
|||
|
|
UserAgent: ua,
|
|||
|
|
Cost: cost,
|
|||
|
|
StatusCode: c.Writer.Status(),
|
|||
|
|
Response: "",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 如果不是返回200，把返回值保存一下
|
|||
|
|
if c.Writer.Status() != 200 {
|
|||
|
|
resp := bw.body.String()
|
|||
|
|
l.Response = resp
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
go func() {
|
|||
|
|
if er := repository.SystemLog().SaveLog(&l); er != nil {
|
|||
|
|
log.Debugf("请求日志: %+v", l)
|
|||
|
|
log.Errorf("保存请求日志失败: %v", er)
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
}
|
|||
|
|
}
|