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

相关推荐
ByteBlossom1 天前
SpringBoot集成Kafka实战应用
spring boot·中间件·kafka
驾驭人生1 天前
基本IP保护 Swagger UI 的中间件
中间件
瀚高PG实验室1 天前
金蝶中间件适配HGDB
中间件·瀚高数据库
武子康1 天前
Java-114 深入浅出 MySQL 开源分布式中间件 ShardingSphere 深度解读
java·数据库·分布式·mysql·中间件·性能优化·开源
万行2 天前
点评项目(Redis中间件)&第二部分Redis基础
java·数据库·redis·spring·中间件
Thuni_soft2 天前
华宇TAS应用中间件与长城科技两款产品完成兼容互认证
科技·中间件·tas
Joker—H2 天前
【Java】Redis(中间件)
java·开发语言·经验分享·redis·中间件
lifallen2 天前
Caffeine TimerWheel时间轮 深度解析:O(1)复杂度增删和触发时间事件
java·数据结构·算法·缓存·中间件
天翼云开发者社区3 天前
分布式系统-秒杀
大数据·安全·中间件
xiao-xiang5 天前
zookeeper-集群扩缩容
linux·分布式·zookeeper·中间件·注册中心·zk