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

  • 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、在需要使用的地方直接使用就可以,自动会收集日志到数据库中

相关推荐
ChinaRainbowSea5 小时前
1. 初始 RabbitMQ 消息队列
java·中间件·rabbitmq·java-rabbitmq
还是鼠鼠19 小时前
Node.js全局生效的中间件
javascript·vscode·中间件·node.js·json·express
能来帮帮蒟蒻吗1 天前
GO语言学习(16)Gin后端框架
开发语言·笔记·学习·golang·gin
Json20113151 天前
Gin、Echo 和 Beego三个 Go 语言 Web 框架的核心区别及各自的优缺点分析,结合其设计目标、功能特性与适用场景
前端·golang·gin·beego
@PHARAOH1 天前
WHAT - 如何理解中间件
中间件
生命是有光的1 天前
【中间件安全计划】锚定Tomcat安全基线
安全·中间件·tomcat
Json20113152 天前
Swoole 的 Hyperf 框架和 Go 的 Gin 框架高并发原理以及技术实现对比分析
网络·php·gin·swoole
赴前尘2 天前
Go+Gin实现安全多文件上传:带MD5校验的完整解决方案
安全·golang·gin
Golinie2 天前
Ollama+Langchaingo+Gin开发本地LLM简单应用
大模型·gin·ollama·langchaingo
yuanlaile3 天前
Gin介绍及Gin环境搭建
golang·gin·gin环境搭建