package utils

import (
    "errors"
    "fmt"
    "github.com/dgrijalva/jwt-go"
    "strings"
    "time"
)

// 生成Jwt Token
// @Param  secretKey 表示jwt secretKey【*******注意：發(fā)布多臺(tái)時(shí)secretKey 必須一樣*********】
// @Result  string 創(chuàng)建的Token
// @Result  error 返回驗(yàn)證錯(cuò)誤信息

func GenerateJWTToken() (string, error) {
    // secretKey 表示jwt 的secretKey【*******注意：發(fā)布多臺(tái)時(shí)secretKey 必須一樣，否則通不過(guò)驗(yàn)證*********】
    // 獲取 ini 文件配置的 Jwt SecretKey
    var secretKey = GetIniVale("jwt", "SecretKey")
    claims := jwt.MapClaims{
        "sub":  "1234567890",
        "name": "John Doe",
        "iat":  time.Now().Unix(),
        "exp":  time.Now().Add(time.Minute * 3).Unix(),
    }
    // 使用指定的簽名方法和密鑰創(chuàng)建令牌
    token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
    tokenString, err := token.SignedString([]byte(secretKey))
    if err != nil {
        return "", err
    }
    return tokenString, nil
}

// 驗(yàn)證Jwt token  并且如果Token快過(guò)期，就創(chuàng)建新的token
// @Param  tokenString 表示jwt token
// @Result  bool 驗(yàn)證是否通過(guò)
// @Result  string 快過(guò)期創(chuàng)建新的Token，沒(méi)有則為空
// @Result  error 返回驗(yàn)證錯(cuò)誤信息

func ValidateJWTToken(tokenString string) (bool, string, error) {
    // secretKey 表示jwt 的secretKey【*******注意：發(fā)布多臺(tái)時(shí)secretKey 必須一樣，否則通不過(guò)驗(yàn)證*********】
    // 獲取 ini 文件配置的 Jwt SecretKey
    var secretKey = GetIniVale("jwt", "SecretKey")
    tokenString = strings.TrimPrefix(tokenString, "Bearer ")
    token, err := jwt.Parse(tokenString, func(token *jwt.Token) (interface{}, error) {
        return []byte(secretKey), nil
    })
    if err != nil {
        return false, "", err
    }
    claims, ok := token.Claims.(jwt.MapClaims)
    // token 驗(yàn)證
    if ok && token.Valid {
        // Token是否過(guò)期
        if !claims.VerifyExpiresAt(time.Now().Unix(), false) {
            return false, "", errors.New("access token expired")
        }
        // 一分鐘過(guò)期就重新添加一個(gè)Token 返回給前端
        if !claims.VerifyExpiresAt(time.Now().Add(time.Minute).Unix(), false) {
            newToken, _ := GenerateJWTToken()
            return true, newToken, nil
        }
        return true, "", nil
    } else {
        return false, "", fmt.Errorf("Invalid token")
    }
}

package controller

import (
    "github.com/gin-gonic/gin"
    "jwt/utils"
)

type UserController struct{}

func (u UserController) Login(c *gin.Context) {
    userName := c.Query("userName")
    pwd := c.Query("password")
    if userName == "zs" && pwd == "123" {
        // 登錄成功  生成Jwt Token
        token, err := utils.GenerateJWTToken()
        if err != nil {
            c.JSON(500, gin.H{"Msg": "生成Token失敗：" + err.Error()})
        } else {
            c.JSON(200, gin.H{"Token": token})
        }
    } else {
        c.JSON(500, gin.H{"Msg": "用戶中或密碼錯(cuò)誤！"})
    }
}

package middleware

import (
    "github.com/gin-gonic/gin"
    "jwt/utils"
    "strings"
)

func ValidataToken() gin.HandlerFunc {
    return func(c *gin.Context) {    // 判斷 是不是登錄鏈接，如果是直接往下走
        if strings.Contains(strings.ToLower(c.Request.RequestURI), "/user/login") {
            c.Next()
        } else {
            token := c.GetHeader("Authorization")
            // 驗(yàn)證token  并且如果 Token快過(guò)期 就創(chuàng)建新的token
            bls, newToken, err := utils.ValidateJWTToken(token)
            if err != nil {
                c.JSON(500, gin.H{
                    "msg": err.Error(),
                })
                c.Abort()
                return
            } else {
                if bls {
                    // 是否有快過(guò)期創(chuàng)建新的Token
                    if len(newToken) > 0 {
                        c.Header("newToken", newToken) // 將新Token添加到返回的Header里，方便前端使用
                    }
                    c.Next()
                } else {
                    c.JSON(401, gin.H{"msg": "token not validated"})
                    c.Abort()
                    return
                }
            }
        }

    }
}

package main

import (
    "github.com/gin-gonic/gin"
    "jwt/middleware"
    "jwt/route"
)

func main() {
    r := gin.Default()
    r.Use(middleware.ValidataToken())
    route.AddUserInfoRoute(r)
    r.Run(":8080")
}