教程 > Gin 教程 > Gin 基础 阅读:81

Gin 中间件

Gin框架允许开发者在处理请求的过程中,加入钩子函数,这个钩子函数就叫中间件。中间件适合处理一些公共的业务逻辑,比如登陆认证,权限校验,记录日志等。具体使用方法如下

package main

import (
    "fmt"
    "net/http"
    "time"
    "github.com/gin-gonic/gin"
)

//定义一个中间键m1统计请求处理函数耗时
func m1(c *gin.Context) {
    fmt.Println("m1 in...")
    start := time.Now()
    // c.Next() //调用后续的处理函数
    c.Abort()//阻止调用后续的处理函数
    cost := time.Since(start)
    fmt.Printf("cost:%v\n", cost)
}

func index(c *gin.Context) {
    c.JSON(http.StatusOK, gin.H{
        "msg": "ok",
    })
}

func main() {
    r := gin.Default()
    r.GET("/", m1, index)
    r.Run()
}

在中间件中使用Goroutine

当在中间件或 handler 中启动新的 Goroutine 时,不能使用原始的上下文,必须使用只读副本。

package main

import (
    "github.com/gin-gonic/gin"
    "log"
    "time"
)

func main() {
    r := gin.Default()

    r.GET("/long_async", func(c *gin.Context) {
        // 创建在 goroutine 中使用的副本
        tmp := c.Copy()
        go func() {
            // 用 time.Sleep() 模拟一个长任务。
            time.Sleep(5 * time.Second)

            // 请注意您使用的是复制的上下文 "tmp",这一点很重要
            log.Println("Done! in path " + tmp.Request.URL.Path)
        }()
    })

    r.GET("/long_sync", func(c *gin.Context) {
        // 用 time.Sleep() 模拟一个长任务。
        time.Sleep(5 * time.Second)

        // 因为没有使用 goroutine,不需要拷贝上下文
        log.Println("Done! in path " + c.Request.URL.Path)
    })
    r.Run()
}

查看笔记

扫码一下
查看教程更方便