简单记录下gin中使用中间件记录操作日志

水痕012023-09-07 10:57

1、直接定义中间件

go 复制代码

package middleware

import (
	"bytes"
	"encoding/json"
	"fmt"
	"github.com/gin-gonic/gin"
	"go.uber.org/zap"
	"io"
	"strconv"
	"strings"
)

func LoggerMiddleWare() gin.HandlerFunc {
	return func(ctx *gin.Context) {
		//请求方式
		method := ctx.Request.Method
		//请求路由
		reqUrlList := strings.Split(ctx.Request.URL.String(), "?")
		//状态码
		statusCode := ctx.Writer.Status()
		//请求ip
		clientIP := ctx.ClientIP() 
		// 获取请求体数据
		var data map[string]interface{}
		body, err := io.ReadAll(ctx.Request.Body)
		// 等于拷贝一份往下传递,否则下接口的方法中拿不到请求体数据
		ctx.Request.Body = io.NopCloser(bytes.NewBuffer(body))
		if err != nil {
			fmt.Println(err, "????")
		}
		err = json.Unmarshal(body, &data)
		// 记录到数据库中,排除登录接口
		if reqUrlList[0] != "/api/v1/admin/login" {
			message := ""
			if method == "GET" {
				if len(reqUrlList) == 2 && reqUrlList[1] != "" {
					message = reqUrlList[1]
				}
			} else {
				if utils.MapToJson(data) != "null" {
					message = utils.MapToJson(data)
				}
			}
			// TODO 插入到数据库中
		}

		// 打印日志
		//loggerMap := map[string]interface{} {
		//	"status_code":statusCode,
		//	"client_ip": clientIP,
		//	"req_method":method,
		//	"req_uri": reqUrl,
		//}
		//marshal, _ := json.Marshal(loggerMap)
		loggerStr := fmt.Sprintf("status_code:%s,client_ip:%s,req_method:%s,req_uri:%s", strconv.Itoa(statusCode), clientIP, method, reqUrlList[0])
		global.Logger.Info("中间件本次请求", zap.String("http", loggerStr))
		ctx.Next()
	}
}

2、在需要使用的地方直接使用就可以，自动会收集日志到数据库中