u

code/allapp/allapp-go/utils/jwtx/index.go

@@ -3,13 +3,13 @@ package jwtx
 import (
 	"allapp/config"
 	"context"
-	"errors"
 	"sync"
 	"time"
 
 	"github.com/golang-jwt/jwt/v5"
 )
 
+// CustomClaims 定义 JWT 自定义 Claims
 type CustomClaims struct {
 	UserID int64 `json:"user_id"`
 	jwt.RegisteredClaims
@@ -26,8 +26,8 @@ var (
 	once    sync.Once
 )
 
-// InitJwt 初始化 JWT 单例
-func InitJwt() {
+// getManager 自动初始化单例
+func getManager() *jwtManager {
 	once.Do(func() {
 		cfg := config.GetConf().JWT
 		manager = &jwtManager{
@@ -35,20 +35,19 @@ func InitJwt() {
 			tokenExpiry: cfg.AccessExpiry,
 		}
 	})
+	return manager
 }
 
 // CreateToken 生成 JWT
 func CreateToken(ctx context.Context, userID int64) (string, error) {
-	if manager == nil {
-		return "", errors.New("jwt manager not initialized")
-	}
-	return manager.createToken(userID)
+	m := getManager()
+	return m.createToken(userID)
 }
 
 // VerifyToken 校验 token 并返回 claims
-func VerifyToken(ctx context.Context, tokenString string) (*CustomClaims, error) {
+func VerifyToken(ctx context.Context, tokenString string) (*CustomClaims, bool) {
 	if tokenString == "" || manager == nil {
-		return nil, errors.New("token empty or manager not initialized")
+		return nil, false
 	}
 
 	token, err := jwt.ParseWithClaims(tokenString, &CustomClaims{}, func(token *jwt.Token) (interface{}, error) {
@@ -56,15 +55,15 @@ func VerifyToken(ctx context.Context, tokenString string) (*CustomClaims, error)
 	})
 
 	if err != nil || !token.Valid {
-		return nil, errors.New("invalid token")
+		return nil, false
 	}
 
 	claims, ok := token.Claims.(*CustomClaims)
 	if !ok {
-		return nil, errors.New("invalid claims")
+		return nil, false
 	}
 
-	return claims, nil
+	return claims, true
 }
 
 // createToken 私有方法