[Golang]基于Gin框架快速实现中间件——http拦截器

Go是比较容易基于net/http来实现中间件的，比如现在要加一个http访问的拦截器，可以对那些需要授权才能访问的接口进行验证。

比如：

func HTTPInterceptor(h http.HandlerFunc) http.HandlerFunc { return http.HandlerFunc( func(w http.ResponseWriter, r *http.Request) { r.ParseForm() // TODO: 进行身份验证，比如校验cookie或token
 h(w, r) }) } // 然后在创建路由时类似这么调用: // http.HandleFunc("/test", HTTPInterceptor(YourCustomHandler))

而基于Gin框架来实现这个功能的话，也是非常简单，只要注册一个中间件函数即可。
只有注册中间件之后的路由会被修饰，而注册中间件之前的路由不会受影响；
具体示例如下:

 

package main

import (
"github.com/gin-gonic/gin"
"net/http"
"crypto/md5"
"encoding/hex"
"strings"
)

const (
    // 可自定义盐值
    TokenSalt = "default_salt"
)

func MD5(data []byte) string {
    _md5 := md5.New()
    _md5.Write(data)
    return hex.EncodeToString(_md5.Sum([]byte("")))
}

func Authorize() gin.HandlerFunc{
  return func(c *gin.Context){
        username := c.Query("username") // 用户名
        ts := c.Query("ts") // 时间戳
        token := c.Query("token") // 访问令牌
        
        if strings.ToLower(MD5([]byte(username+ts+TokenSalt))) == strings.ToLower(token) {
            // 验证通过，会继续访问下一个中间件
            c.Next()
        } else {
            // 验证不通过，不再调用后续的函数处理
            c.Abort()
            c.JSON(http.StatusUnauthorized,gin.H{"message":"访问未授权"})
            // return可省略, 只要前面执行Abort()就可以让后面的handler函数不再执行
            return
        }
  }
}

func ServiceWithoutAuth(c *gin.Context){
    c.JSON(http.StatusOK, gin.H{"message":"这是一个不用经过认证就能访问的接口"})
}

func ServiceWithAuth(c *gin.Context){
    c.JSON(http.StatusOK, gin.H{"message":"这是一个需要经过认证才能访问的接口，看到此信息说明验证已通过"})
}


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

    // Use(Authorize())之前的接口，都不用经过身份验证
    router.GET("/service_without_auth", ServiceWithoutAuth)

    //以下的接口，都使用Authorize()中间件身份验证
    router.Use(Authorize())
    router.GET("/service_with_auth", ServiceWithAuth)

  router.Run(":9999")
}

 

 

 

curl测试用例：

 

$ curl 'http://localhost:9999/service_with_auth?username=admin'
{"message":"访问未授权"}

$ curl 'http://localhost:9999/service_without_auth?username=admin'
{"message":"这是一个不用经过认证就能访问的接口"}

$ curl 'http://localhost:9999/service_with_auth?username=admin&ts=1552440940&token=142f0bee2b4a088fc93b5aee341b7c56'
{"message":"这是一个需要经过认证才能访问的接口，看到此信息说明验证已通过"}