golang常用代码片段--定制化gin中间件

· · 669 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

limiter限流器中间件

  • 作为gin中的一个中间件,起到对接口限流的作用
  • 接口限流的配置组件每个人都不同,使用自己的就可以,本文使用的之前文章中的myConf
func MyLimiter(c *gin.Context) {
    url := c.Request.URL.Path
    if limiter := myConf.GetLimiterByUrl(url); limiter != nil && !limiter.Allow() {
        myLog.GetUidTraceLog(c).Errorf("limit url: [%s]", url)
        c.AbortWithStatus(http.StatusForbidden)
        return
    }
    c.Next()
}

logger记录整体请求信息的中间件,并且传入和uid绑定的logger

func MyLogger(c *gin.Context) {
    start := time.Now()
    traceLog := logrus.WithField("uid", c.Query("uid"))
    c.Set("traceLog", traceLog)
    c.Next()
    rawQuery := ""
    if c.Request.URL.RawQuery != "" {
        rawQuery = "?" + c.Request.URL.RawQuery
    }
    traceLog.WithFields(logrus.Fields{
        "remoteIp": c.ClientIP(),
        "method":   c.Request.Method,
        "url":      c.Request.URL.Path + rawQuery,
        "status":   c.Writer.Status(),
        "cost":     fmt.Sprintf("%dms", time.Since(start).Milliseconds()),
    }).Info("total log")
}

panic恢复中间件

func MyRecover(c *gin.Context) {
    defer func() {
        if err := recover(); err != nil {
            myLog.GetUidTraceLog(c).Errorf("unknown panic: [%s], stacktrace: [%s]", err, debug.Stack())
            c.AbortWithStatus(http.StatusInternalServerError)
        }
    }()
    c.Next()
}

有疑问加站长微信联系(非本文作者)

本文来自:简书

感谢作者:

查看原文:golang常用代码片段--定制化gin中间件

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

669 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传