使用gin中間件進行跨域

當我們使用gin框架時，可以在中間件中進行跨域設置。

// 中間件
func middleware() gin.HandlerFunc {
    return func(c *gin.Context) {
        t := time.Now()
        c.Set("example", "123456")
        // c.Next()后就執行真實的路由函數，路由執行完成之后接着走time.Since(t)
        c.Next()

        // 從time.Now()到目前為止過了多長時間
        latency := time.Since(t)
        log.Print("--", latency)

        // gin設置響應頭，設置跨域
        c.Header("Access-Control-Allow-Origin", "*")
        c.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE")
        c.Header("Access-Control-Allow-Headers", "Action, Module, X-PINGOTHER, Content-Type, Content-Disposition")

        //設置中間件的響應頭，路由的響應頭可以在路由返回中設置，參考/ping
        // c.Writer.WriteHeader(http.StatusMovedPermanently)
        status := c.Writer.Status()
        log.Println("==", status)
    }
}

func middle(c *gin.Context) {
    example := c.MustGet("example").(string)
    log.Println(example)
}

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

    //中間件，只有在中間件之后注冊的路由才會走中間件
    r.Use(middleware())
    //http://localhost:8081/middle
    r.GET("/middle", middle)

    r.Run(":8081")
}

每次執行路由函數middle()之前都會執行middleware()函數，在執行到middleware函數中c.Next()時，會執行真正的路由函數middle()，路由函數執行完成之后會繼續執行middleware()中c.Next()函數后面的代碼。