1. Cookie练习

  • 模拟实现权限验证中间件

    • 有2个路由，login和home
    • login用于设置cookie
    • home是访问查看信息的请求
    • 在请求home之前，先跑中间件代码，检验是否存在cookie
  • 访问home，会显示错误，因为权限校验未通过

Cookie练习 - 图1

  • 然后访问登录的请求，登录并设置cookie

Cookie练习 - 图2

  • 再次访问home，访问成功

Cookie练习 - 图3

  1. package main
  3. import (
  4.    "github.com/gin-gonic/gin"
  5.    "net/http"
  6. )
  8. func AuthMiddleWare() gin.HandlerFunc {
  9.    return func(c *gin.Context) {
  10.       // 获取客户端cookie并校验
  11.       if cookie, err := c.Cookie("abc"); err == nil {
  12.          if cookie == "123" {
  13.             c.Next()
  14.             return
  15.          }
  16.       }
  17.       // 返回错误
  18.       c.JSON(http.StatusUnauthorized, gin.H{"error": "err"})
  19.       // 若验证不通过，不再调用后续的函数处理
  20.       c.Abort()
  21.       return
  22.    }
  23. }
  25. func main() {
  26.    // 1.创建路由
  27.    r := gin.Default()
  28.    r.GET("/login", func(c *gin.Context) {
  29.       // 设置cookie
  30.       c.SetCookie("abc", "123", 60, "/",
  31.          "localhost", false, true)
  32.       // 返回信息
  33.       c.String(200, "Login success!")
  34.    })
  35.    r.GET("/home", AuthMiddleWare(), func(c *gin.Context) {
  36.       c.JSON(200, gin.H{"data": "home"})
  37.    })
  38.    r.Run(":8000")
  39. }

访问 /home 和 /login进行测试