cc/server/middleware/operation.go

package middleware

import (
	"bytes"
	"gin-vue-admin/global"
	"gin-vue-admin/model"
	"gin-vue-admin/service"
	"github.com/gin-gonic/gin"
	"io/ioutil"
	"net/http"
	"strconv"
	"strings"
)

var body []byte

func RecordRequestBody() gin.HandlerFunc {
	return func(c *gin.Context) {
		if c.Request.Method != http.MethodGet {
			var err error
			body, err = ioutil.ReadAll(c.Request.Body)
			if err != nil {
				global.GVA_LOG.Error(err)
			}
			c.Request.Body = ioutil.NopCloser(bytes.NewBuffer(body))
		} else {
			body = nil
		}
	}
}

func OperationRecord() gin.HandlerFunc {
	return gin.LoggerWithConfig(gin.LoggerConfig{
		Formatter: func(param gin.LogFormatterParams) string {
			// 防止加载查询参数，再次过滤
			for _, v := range global.GVA_CONFIG.Operation.SkipPaths {
				if strings.Contains(param.Path, v) {
					return ""
				}
			}
			userId, err := strconv.Atoi(param.Request.Header.Get("x-user-id"))
			if err != nil {
				userId = 0
			}
			err = service.CreateSysOperationRecord(model.SysOperationRecord{
				Ip:           param.ClientIP,
				Method:       param.Method,
				Path:         param.Path,
				Status:       param.StatusCode,
				Latency:      param.Latency,
				Agent:        param.Request.UserAgent(),
				ErrorMessage: param.ErrorMessage,
				Body:         string(body),
				UserId:       userId,
			})
			if err != nil {
				global.GVA_LOG.Error(err)
			}
			return ""
		},
		Output:    nil,
		SkipPaths: global.GVA_CONFIG.Operation.SkipPaths,
	})
}