gin框架JWT验证实践(原理介绍，代码实践)

HMACSHA256(base64UrlEncode(header)+"."+base64UrlEncode(payload),secret)

func main() {
  //从配置文件中读取数据库的配置信息并连接数据库
  if dbErr := opdb.InitMySqlConn(); dbErr != nil {
    log.Panicln(dbErr)
  }
  defer opdb.DB.Close()
  //初始化表结构
  opdb.InitModel()
  router := gin.Default()
  v1 := router.Group("apis/v1")
  {   //注册
    v1.POST("/register", controller.RegisterUser)
    //登陆，返回token
    v1.POST("/login", controller.Login)
  }
  sv1 := v1.Group("/auth")
  //检验token
  sv1.Use(middleware.JWTAuth)
  {   //测试
    sv1.GET("/sayHello", controller.SayHello)
    //更新token
    sv1.GET("/refresh", controller.Refresh)
  }
  router.Run(":8080")
}

  //验证账号密码是否正确，即是否在数据库中存在，注册过
  pass, dbErr := opdb.LoginPass(loginReq)
  if !pass {
    c.JSON(http.StatusOK, gin.H{
      "status": -1,
      "msg":    "账号或密码错误" + dbErr.Error(),
      "data":   nil,
    })
    return
  }
  //创建一个token
  token := generateToken(c, loginReq)

//创建一个token
func generateToken(c *gin.Context, loginReq dfst.LoginReq) (token string) {
  // 构造SignKey: 签名和解签名需要使用一个值
  jwt := middleware.NewJWT()
  // 构造用户claims信息(负荷)
  claims := middleware.CustomClaims{
    Name: loginReq.Name,
    StandardClaims: jwt2.StandardClaims{
      NotBefore: time.Now().Unix() - 1000, // 签名生效时间
      ExpiresAt: time.Now().Unix() + 3600, // 签名过期时间
      Issuer:    "wxf.top",                // 签名颁发者
    },
  }
  // 根据claims生成token对象
  token, err := jwt.CreateToken(claims)
  if err != nil {
    c.JSON(http.StatusOK, gin.H{
      "status": -1,
      "msg":    err.Error(),
      "data":   nil,
    })
  }
  log.Println("create token", token)
  return
}

//创建token
func (j *JWT) CreateToken(claims CustomClaims) (string, error) {
  token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
  //获取完整的签名令牌
  return token.SignedString(j.SigningKey)
}

func JWTAuth(c *gin.Context) {
  //从header的头部的token字段获取token
  token := c.Request.Header.Get("token")
  if token == "" {
    c.JSON(http.StatusOK, gin.H{
      "status": -1,
      "msg":    "请求未携带token，无权限访问",
      "data":   nil,
    })
    c.Abort()
    return
  }
  log.Println("recv tokens:", token)
  j := NewJWT()
  //解析token，并将PAYLOAD负载提取出来
  claims, err := j.ParserToken(token)
  if err != nil {
    // token过期
    if err == TokenExpired {
      c.JSON(http.StatusOK, gin.H{
        "status": -1,
        "msg":    "token授权已过期，请重新申请授权",
        "data":   nil,
      })
      //中断调用链
      c.Abort()
      return
    }
    // 其他错误
    c.JSON(http.StatusOK, gin.H{
      "status": -1,
      "msg":    err.Error(),
      "data":   nil,
    })
    c.Abort()
    return
  }
  //将负载添加到context上下文中供调用链中的函数使用
  c.Set("claims", claims)
}

//解析token，并将PAYLOAD负载提取出来
func (j *JWT) ParserToken(tokenString string) (*CustomClaims, error) {
  token, err := jwt.ParseWithClaims(tokenString, &CustomClaims{}, func(token *jwt.Token) (interface{}, error) {
    return j.SigningKey, nil
  })
  if err != nil {
    // jwt.ValidationError 是一个无效token的错误结构
    if ve, ok := err.(*jwt.ValidationError); ok {
      // ValidationErrorMalformed是一个uint常量，表示token不可用
      if ve.Errors&jwt.ValidationErrorMalformed != 0 {
        return nil, TokenMalformed
        // ValidationErrorExpired表示Token过期
      } else if ve.Errors&jwt.ValidationErrorExpired != 0 {
        return nil, TokenExpired
        // ValidationErrorNotValidYet表示无效token
      } else if ve.Errors&jwt.ValidationErrorNotValidYet != 0 {
        return nil, TokenNotValidYet
      } else {
        return nil, TokenInvalid
      }
    }
  }
  if claims, ok := token.Claims.(*CustomClaims); ok && token.Valid {
    return claims, nil
  }
  return nil, TokenInvalid
}