fix: jwt添加role信息

This commit is contained in:
2026-03-06 12:03:14 +08:00
parent 22d44aa5ec
commit 96b407ef89

View File

@@ -152,8 +152,9 @@ func (j *JWT) ValidateToken(tokenString string) bool {
// secretKey: 密钥 // secretKey: 密钥
// userID: 用户 ID // userID: 用户 ID
// userName: 用户名 // userName: 用户名
// role: 用户角色
// expires: 过期时间(小时) // 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{ jwtUtil := NewJWT(JWTConfig{
SecretKey: secretKey, SecretKey: secretKey,
ExpiresTime: time.Duration(expires) * time.Hour, ExpiresTime: time.Duration(expires) * time.Hour,
@@ -162,6 +163,7 @@ func CreateToken(secretKey, userID, userName string, expires int) (string, error
claims := &CustomClaims{ claims := &CustomClaims{
UserID: userID, UserID: userID,
UserName: userName, UserName: userName,
Role: role,
} }
return jwtUtil.GenerateToken(claims) return jwtUtil.GenerateToken(claims)