Gin 中間件

Gin框架允許開(kāi)發(fā)者在處理請(qǐng)求的過(guò)程中，加入鉤子函數(shù)，這個(gè)鉤子函數(shù)就叫中間件。中間件適合處理一些公共的業(yè)務(wù)邏輯，比如登陸認(rèn)證，權(quán)限校驗(yàn)，記錄日志等。具體使用方法如下

package main

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

//定義一個(gè)中間鍵m1統(tǒng)計(jì)請(qǐng)求處理函數(shù)耗時(shí)
func m1(c *gin.Context) {
	fmt.Println("m1 in...")
	start := time.Now()
	// c.Next() //調(diào)用后續(xù)的處理函數(shù)
	c.Abort()//阻止調(diào)用后續(xù)的處理函數(shù)
	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

當(dāng)在中間件或 handler 中啟動(dòng)新的 Goroutine 時(shí)，不能使用原始的上下文，必須使用只讀副本。

package main

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

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

	r.GET("/long_async", func(c *gin.Context) {
		// 創(chuàng)建在 goroutine 中使用的副本
		tmp := c.Copy()
		go func() {
			// 用 time.Sleep() 模擬一個(gè)長(zhǎng)任務(wù)。
			time.Sleep(5 * time.Second)

			// 請(qǐng)注意您使用的是復(fù)制的上下文 "tmp"，這一點(diǎn)很重要
			log.Println("Done! in path " + tmp.Request.URL.Path)
		}()
	})

	r.GET("/long_sync", func(c *gin.Context) {
		// 用 time.Sleep() 模擬一個(gè)長(zhǎng)任務(wù)。
		time.Sleep(5 * time.Second)

		// 因?yàn)闆](méi)有使用 goroutine，不需要拷貝上下文
		log.Println("Done! in path " + c.Request.URL.Path)
	})
	r.Run()
}