全局中间件
所有的请求都经过此中间件
// 所有请求经过此中间件
package main
import (
"fmt"
"time"
"github.com/gin-gonic/gin"
)
//定义中间件
func MiddleWare() gin.HandlerFunc {
return func(ctx *gin.Context) {
t := time.Now()
fmt.Println("中间件开始执行了")
//设置变量到Context的key中,可以通过Get()取
ctx.Set("request", "中间件")
status := ctx.Writer.Status()
fmt.Println("中间件执行完毕", status)
t2 := time.Since(t)
fmt.Println("time:", t2)
}
}
func main() {
r := gin.Default()
r.Use(MiddleWare())
//{}为了规范代码
{
r.GET("/ce", func(ctx *gin.Context) {
//取值
req, _ := ctx.Get("request")
fmt.Println("request:", req)
//页面接收
ctx.JSON(200, gin.H{
"request": req,
})
})
}
r.Run()
}
测试结果
Next()方法
package main
import (
"fmt"
"time"
"github.com/gin-gonic/gin"
)
// 定义中间
func MiddleWare() gin.HandlerFunc {
return func(c *gin.Context) {
t := time.Now()
fmt.Println("中间件开始执行了")
// 设置变量到Context的key中,可以通过Get()取
c.Set("request", "中间件")
// 执行函数
c.Next()
// 中间件执行完后续的一些事情
status := c.Writer.Status()
fmt.Println("中间件执行完毕", status)
t2 := time.Since(t)
fmt.Println("time:", t2)
}
}
func main() {
// 1.创建路由
// 默认使用了2个中间件Logger(), Recovery()
r := gin.Default()
// 注册中间件
r.Use(MiddleWare())
// {}为了代码规范
{
r.GET("/ce", func(c *gin.Context) {
// 取值
req, _ := c.Get("request")
fmt.Println("request:", req)
// 页面接收
c.JSON(200, gin.H{"request": req})
})
}
r.Run()
}
局部中间件
package main
import (
"fmt"
"time"
"github.com/gin-gonic/gin"
)
// 定义中间
func MiddleWare() gin.HandlerFunc {
return func(c *gin.Context) {
t := time.Now()
fmt.Println("中间件开始执行了")
// 设置变量到Context的key中,可以通过Get()取
c.Set("request", "中间件")
// 执行函数
c.Next()
// 中间件执行完后续的一些事情
status := c.Writer.Status()
fmt.Println("中间件执行完毕", status)
t2 := time.Since(t)
fmt.Println("time:", t2)
}
}
func main() {
// 1.创建路由
// 默认使用了2个中间件Logger(), Recovery()
r := gin.Default()
//局部中间键使用
r.GET("/ce", MiddleWare(), func(c *gin.Context) {
// 取值
req, _ := c.Get("request")
fmt.Println("request:", req)
// 页面接收
c.JSON(200, gin.H{"request": req})
})
r.Run()
}
中间件练习
package main
import (
"fmt"
"time"
"github.com/gin-gonic/gin"
)
// 定义中间
func myTime(c *gin.Context) {
start := time.Now()
c.Next()
// 统计时间
since := time.Since(start)
fmt.Println("程序用时:", since)
}
// 定义处理函数
func ShopIndexHandler(ctx *gin.Context) {
time.Sleep(5 * time.Second)
}
func ShopHomeHandler(ctx *gin.Context) {
time.Sleep(3 * time.Second)
}
func main() {
r := gin.Default()
r.Use(myTime)
ShoppingGroup := r.Group("/shopping")
{
ShoppingGroup.GET("/index", ShopIndexHandler)
ShoppingGroup.GET("/home", ShopHomeHandler)
}
r.Run()
}