:recycle:重构了部分代码

component/jwt.go

@@ -29,14 +29,18 @@ func JWT() JwtClaims {
 // @receiver Jwt
 // @return token
 // @return err
-func (j JwtClaims) GenerateToken(userId string) (token string, err error) {
+func (j JwtClaims) GenerateToken(userId string) (token string, expireTime *jwt.NumericDate, err error) {
+	timeNow := time.Now().Local()
+	expireTime = jwt.NewNumericDate(timeNow.Add(7 * time.Hour))
+	notBefore := jwt.NewNumericDate(timeNow)
+	issuedAt := jwt.NewNumericDate(timeNow)
 	claims := JwtClaims{
 		ID: userId,
 		RegisteredClaims: jwt.RegisteredClaims{
 			Subject:   "wireguard-dashboard",
-			ExpiresAt: jwt.NewNumericDate(time.Now().Local().Add(7 * time.Hour)),
-			NotBefore: jwt.NewNumericDate(time.Now().Local()),
-			IssuedAt:  jwt.NewNumericDate(time.Now().Local()),
+			ExpiresAt: expireTime,
+			NotBefore: notBefore,
+			IssuedAt:  issuedAt,
 		},
 	}
 
@@ -44,7 +48,7 @@ func (j JwtClaims) GenerateToken(userId string) (token string, err error) {
 	token, err = t.SignedString([]byte(Secret))
 	if err != nil {
 		log.Errorf("生成token失败: %v", err.Error())
-		return "", errors.New("生成token失败")
+		return "", nil, errors.New("生成token失败")
 	}
 
 	client.Redis.Set(context.Background(), fmt.Sprintf("%s:%s", constant.Token, userId), token, 7*time.Hour)