Gin框架下的JWT

什么是JWT

我们都知道，HTTP协议是无状态的，click here,那么服务端怎么知道用户状态的呢（比如是否登录呢），这里就需要用到中间件来进行用户认证。

header（头部）
payload（数据）
signature（签名）

const encodedString = base64UrlEncode(header) + "." + base64UrlEncode(payload); 
HMACSHA256(encodedString, 'secret');

废话不多说，直接上代码，原理上面讲过，详情请见注释！

package main

import (
	"errors"
	"fmt"
	"net/http"
	"time"

	"github.com/gin-gonic/gin"
	"github.com/golang-jwt/jwt/v4"
)

var mySecret = []byte("haomiao000") // 直接将密钥作为字符串存储（这里只是个例子，通常会非常复杂保证安全性）

// 定义自定义声明结构体并内嵌 jwt.RegisteredClaims
type MyClaims struct {
	UserID int64 `json:"user_id"`
	jwt.RegisteredClaims
}

// 生成JWT
func GenToken(userID int64) (string, error) {
	// 创建自定义声明
	claims := MyClaims{
		UserID: userID,
		RegisteredClaims: jwt.RegisteredClaims{
			ExpiresAt: jwt.NewNumericDate(time.Now().Add(1 * time.Hour)), // 过期时间设置为1小时
			Issuer:    "jwt",                                             // 签发人
		},
	}
	// 使用指定的签名方法创建签名对象
	token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
	// 使用指定的secret签名并获得完整的编码后的字符串token
	return token.SignedString(mySecret)
}

// 解析JWT
func ParseToken(tokenString string) (*MyClaims, error) {
	var claims MyClaims
	// 解析token
	token, err := jwt.ParseWithClaims(tokenString, &claims, keyFunc)
	if err != nil {
		return nil, fmt.Errorf("parsing token: %w", err)
	}
	// 校验token
	if !token.Valid {
		return nil, errors.New("invalid token")
	}
	return &claims, nil
}

// keyFunc 定义获取密钥的函数
func keyFunc(_ *jwt.Token) (interface{}, error) {
	return mySecret, nil
}

func main() {
	// 初始化Gin引擎
	r := gin.Default()

	// 设置路由
	r.POST("/login", loginHandler)
	authGroup := r.Group("/api")
	authGroup.Use(authMiddleware())
	authGroup.GET("/user", getUserHandler)

	// 启动服务器
	r.Run(":8080")
}

// authMiddleware 定义JWT身份验证中间件
func authMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		tokenString := c.GetHeader("Authorization")
		if tokenString == "" {
			c.JSON(http.StatusUnauthorized, gin.H{"error": "Unauthorized"})
			c.Abort()
			return
		}

		claims, err := ParseToken(tokenString)
		if err != nil {
			c.JSON(http.StatusUnauthorized, gin.H{"error": "Unauthorized"})
			c.Abort()
			return
		}

		// 在上下文中设置用户ID
		c.Set("userID", claims.UserID)

		// 继续处理请求
		c.Next()
	}
}

// loginHandler 登录处理程序
func loginHandler(c *gin.Context) {
	// 这里可以编写身份验证逻辑
	// 例如从请求中获取用户名和密码，然后验证用户信息

	// 假设身份验证成功后，获取用户ID
	userID := int64(123)

	// 生成JWT
	token, err := GenToken(userID)
	if err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"error": "Internal Server Error"})
		return
	}

	// 返回JWT给客户端
	c.JSON(http.StatusOK, gin.H{"token": token})
}

// getUserHandler 示例处理程序，需要JWT验证
func getUserHandler(c *gin.Context) {
	userID, exists := c.Get("userID")
	if !exists {
		c.JSON(http.StatusInternalServerError, gin.H{"error": "Internal Server Error"})
		return
	}

	// 在这里可以使用userID进行进一步的处理，例如从数据库中检索用户信息等

	c.JSON(http.StatusOK, gin.H{"userID": userID})
}

• 登录生成token
  

• 验证token