diff --git a/utils/jwt.go b/utils/jwt.go index 4d724bd..f2453f1 100644 --- a/utils/jwt.go +++ b/utils/jwt.go @@ -152,8 +152,9 @@ func (j *JWT) ValidateToken(tokenString string) bool { // secretKey: 密钥 // userID: 用户 ID // userName: 用户名 +// role: 用户角色 // expires: 过期时间(小时) -func CreateToken(secretKey, userID, userName string, expires int) (string, error) { +func CreateToken(secretKey, userID, userName string, role, expires int) (string, error) { jwtUtil := NewJWT(JWTConfig{ SecretKey: secretKey, ExpiresTime: time.Duration(expires) * time.Hour, @@ -162,6 +163,7 @@ func CreateToken(secretKey, userID, userName string, expires int) (string, error claims := &CustomClaims{ UserID: userID, UserName: userName, + Role: role, } return jwtUtil.GenerateToken(claims)