From a951b982b36307e96fff2919327a18cd0fcd2153 Mon Sep 17 00:00:00 2001
From: lejianwen <84855512@qq.com>
Date: Wed, 15 Jan 2025 20:26:26 +0800
Subject: [PATCH] fix: Jwt

---
 http/middleware/rustauth.go | 2 +-
 lib/jwt/jwt.go              | 6 +++++-
 service/user.go             | 2 +-
 3 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/http/middleware/rustauth.go b/http/middleware/rustauth.go
index 391d2ab..0f8ae5a 100644
--- a/http/middleware/rustauth.go
+++ b/http/middleware/rustauth.go
@@ -32,7 +32,7 @@ func RustAuth() gin.HandlerFunc {
 		//验证token
 
 		//检查是否设置了jwt key
-		if global.Config.Jwt.Key != "" {
+		if len(global.Jwt.Key) > 0 {
 			uid, _ := service.AllService.UserService.VerifyJWT(token)
 			if uid == 0 {
 				c.JSON(401, gin.H{
diff --git a/lib/jwt/jwt.go b/lib/jwt/jwt.go
index 85eb0ab..e522be2 100644
--- a/lib/jwt/jwt.go
+++ b/lib/jwt/jwt.go
@@ -24,6 +24,10 @@ func NewJwt(key string, tokenExpireDuration time.Duration) *Jwt {
 }
 
 func (s *Jwt) GenerateToken(userId uint) string {
+	if len(s.Key) == 0 {
+		fmt.Println("jwt key is nil")
+		return ""
+	}
 	t := jwt.NewWithClaims(jwt.SigningMethodHS256,
 		UserClaims{
 			UserId: userId,
@@ -33,7 +37,7 @@ func (s *Jwt) GenerateToken(userId uint) string {
 		})
 	token, err := t.SignedString(s.Key)
 	if err != nil {
-		fmt.Println(err)
+		fmt.Printf("jwt token generate error: %v", err)
 		return ""
 	}
 	return token
diff --git a/service/user.go b/service/user.go
index 71e8b52..63e1564 100644
--- a/service/user.go
+++ b/service/user.go
@@ -68,7 +68,7 @@ func (us *UserService) InfoByAccessToken(token string) (*model.User, *model.User
 
 // GenerateToken 生成token
 func (us *UserService) GenerateToken(u *model.User) string {
-	if global.Config.Jwt.Key != "" {
+	if len(global.Jwt.Key) > 0 {
 		return global.Jwt.GenerateToken(u.Id)
 	}
 	return utils.Md5(u.Username + time.Now().String())