user viper

pkg/jwt/auth.go

@@ -7,12 +7,12 @@
 package jwt
 
 import (
-	conf "go_blog/config"
 	"go_blog/models"
 	"time"
 
 	"github.com/gin-gonic/gin"
 	"github.com/golang-jwt/jwt"
+	"github.com/spf13/viper"
 )
 
 // 定义jwt载荷
@@ -44,14 +44,14 @@ func GenToken(id uint64, username string) (string, error) {
 		username,
 	}
 	tokenClaims := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
-	token, err := tokenClaims.SignedString([]byte(conf.JwtSecretKey.SecretKey))
+	token, err := tokenClaims.SignedString([]byte(viper.GetString("config.JwtSecretKey.secretKey")))
 	return token, err
 }
 
 // 验证token合法性
 func ValidateJwtToken(token string) (*UserClaims, error) {
 	tokenClaims, err := jwt.ParseWithClaims(token, &UserClaims{}, func(token *jwt.Token) (interface{}, error) {
-		return []byte(conf.JwtSecretKey.SecretKey), nil
+		return []byte(viper.GetString("config.JwtSecretKey.secretKey")), nil
 	})
 
 	if tokenClaims != nil {