Gin为框架入门——中间件和自定义上下文

Gin中的中间件和上下文传值：

package main

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

//--------------------------------------------//
type MyContext struct {
    *gin.Context
    userId   int64 
}
//
func do(c *MyContext) {
    fmt.Println(c.userId)//123456
}
//
type HandlerFunc func(* MyContext)
//
func MyHandler(handler HandlerFunc) gin.HandlerFunc {
    return func(c *gin.Context) {
		//
		ctx := new(MyContext)
		ctx.Context = c
		//
        //if userId, ok := c.Keys["keyUserId"]; ok {
		if userId, ok := c.Get("keyUserId"); ok {
			ctx.userId = userId.(int64)
			fmt.Println(c.Get("mykey2"))//m1 true
		}
		//
        handler(ctx)
    }
}
//--------------------------------------------//
//
//--------------------------------------------//
func MyMiddleware(ctx *gin.Context) {
	ctx.Set("mykey", 10)
	ctx.Set("mykey2", "m1")
	ctx.Set("keyUserId", int64(123456))//
}
//--------------------------------------------//

func main() {
	//生产模式
	//gin.SetMode(gin.ReleaseMode)
	//无中间件
	router := gin.New()
	//自定义中间件 + 自定义上下文
	router.GET("/hi", MyMiddleware, MyHandler(do))	
	//运行Gin
    router.Run(":8080")
}

编译运行。

参考：

https://www.cnblogs.com/iiiiiher/p/12202091.html