lejianwen 1 год назад
Родитель
Сommit
a951b982b3
3 измененных файлов с 7 добавлено и 3 удалено
  1. 1 1
      http/middleware/rustauth.go
  2. 5 1
      lib/jwt/jwt.go
  3. 1 1
      service/user.go

+ 1 - 1
http/middleware/rustauth.go

@@ -32,7 +32,7 @@ func RustAuth() gin.HandlerFunc {
32 32
 		//验证token
33 33
 
34 34
 		//检查是否设置了jwt key
35
-		if global.Config.Jwt.Key != "" {
35
+		if len(global.Jwt.Key) > 0 {
36 36
 			uid, _ := service.AllService.UserService.VerifyJWT(token)
37 37
 			if uid == 0 {
38 38
 				c.JSON(401, gin.H{

+ 5 - 1
lib/jwt/jwt.go

@@ -24,6 +24,10 @@ func NewJwt(key string, tokenExpireDuration time.Duration) *Jwt {
24 24
 }
25 25
 
26 26
 func (s *Jwt) GenerateToken(userId uint) string {
27
+	if len(s.Key) == 0 {
28
+		fmt.Println("jwt key is nil")
29
+		return ""
30
+	}
27 31
 	t := jwt.NewWithClaims(jwt.SigningMethodHS256,
28 32
 		UserClaims{
29 33
 			UserId: userId,
@@ -33,7 +37,7 @@ func (s *Jwt) GenerateToken(userId uint) string {
33 37
 		})
34 38
 	token, err := t.SignedString(s.Key)
35 39
 	if err != nil {
36
-		fmt.Println(err)
40
+		fmt.Printf("jwt token generate error: %v", err)
37 41
 		return ""
38 42
 	}
39 43
 	return token

+ 1 - 1
service/user.go

@@ -68,7 +68,7 @@ func (us *UserService) InfoByAccessToken(token string) (*model.User, *model.User
68 68
 
69 69
 // GenerateToken 生成token
70 70
 func (us *UserService) GenerateToken(u *model.User) string {
71
-	if global.Config.Jwt.Key != "" {
71
+	if len(global.Jwt.Key) > 0 {
72 72
 		return global.Jwt.GenerateToken(u.Id)
73 73
 	}
74 74
 	return utils.Md5(u.Username + time.Now().String())