Golang实现JWT身份验证的示例详解

import (
	"fmt"
	"time"
	jwt "github.com/golang-jwt/jwt/v5"
)

type JWTInstance struct {
	SecretKey []byte
}

func InitJwt(SecretKey []byte) JWTInstance {
	return JWTInstance{SecretKey}
}

// GenerateJWT 生成JWT
func (that JWTInstance) GenerateJWT(username string, count time.Duration) string {
	// 创建一个新的JWT token
	jwtToken := jwt.New(jwt.SigningMethodHS256)
	// 设置一些声明
	claims := jwtToken.Claims.(jwt.MapClaims)
	claims["username"] = username
	claims["exp"] = time.Now().Add(time.Hour * count).Unix()

	// 设置签名并获取token字符串
	token, err := jwtToken.SignedString(that.SecretKey)
	if err != nil {
		return ""
	}
	return token
}

// ParseJWT 解析JWT
func (that JWTInstance) ParseJWT(tokenString string) jwt.MapClaims {
	// 解析JWT字符串
	token, err := jwt.Parse(tokenString, func(token *jwt.Token) (interface{}, error) {
		return that.SecretKey, nil
	})

	if err != nil {
		return nil
	}

	// 验证token
	if claims, ok := token.Claims.(jwt.MapClaims); ok && token.Valid {
		return claims
	}

	return nil
}

func TestJwt() {
	jwt := InitJwt([]byte("2024/3/23"))
	token := jwt.GenerateJWT("example_user", 24) // 生成有效期为24小时的 JWT
	fmt.Println(token)

	claims := jwt.ParseJWT(token)
	fmt.Println(claims)
}