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

相关推荐
眠りたいです4 天前
基于脚手架微服务的视频点播系统-脚手架开发部分-brpc中间件介绍与使用及二次封装
c++·微服务·中间件·rpc·架构·brpc
眠りたいです4 天前
基于脚手架微服务的视频点播系统-脚手架开发部分-jsoncpp,protobuf,Cpp-httplib与WebSocketpp中间件介绍与使用
c++·websocket·微服务·中间件·json·protobuf·cpp-httplib
深蓝电商API5 天前
Scrapy 中间件详解:自定义下载器与爬虫的 “拦截器”
爬虫·scrapy·中间件
知行合一。。。5 天前
SOFA 架构--02--核心中间件与工具
中间件·架构
谢尔登6 天前
【Nest】日志记录
javascript·中间件·node.js
SirLancelot17 天前
MinIO-基本介绍(一)基本概念、特点、适用场景
后端·云原生·中间件·容器·aws·对象存储·minio
RunningShare8 天前
云原生时代的数据流高速公路:深入解剖Apache Pulsar的架构设计哲学
大数据·中间件·apache·pulsar
野熊佩骑9 天前
一文读懂Redis之数据持久化
linux·运维·数据库·redis·缓存·中间件·centos
武子康9 天前
AI-调查研究-90-具身智能 机器人数据采集与通信中间件全面解析:ROS/ROS2、LCM 与工业总线对比
人工智能·ai·中间件·机器人·职场发展·个人开发·具身智能
十五年专注C++开发9 天前
通信中间件 Fast DDS(三) :fastddsgen的安装与使用
linux·c++·windows·中间件·跨平台