mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-04 16:30:25 +08:00
48 lines
1.2 KiB
Go
48 lines
1.2 KiB
Go
package req
|
||
|
||
import (
|
||
"errors"
|
||
"mayfly-go/pkg/config"
|
||
"time"
|
||
|
||
"github.com/golang-jwt/jwt/v5"
|
||
)
|
||
|
||
// 创建用户token
|
||
func CreateToken(userId uint64, username string) (string, error) {
|
||
// 带权限创建令牌
|
||
// 设置有效期，过期需要重新登录获取token
|
||
token := jwt.NewWithClaims(jwt.SigningMethodHS256, jwt.MapClaims{
|
||
"id": userId,
|
||
"username": username,
|
||
"exp": time.Now().Add(time.Minute * time.Duration(config.Conf.Jwt.ExpireTime)).Unix(),
|
||
})
|
||
|
||
// 使用自定义字符串加密 and get the complete encoded token as a string
|
||
tokenString, err := token.SignedString([]byte(config.Conf.Jwt.Key))
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
return tokenString, nil
|
||
}
|
||
|
||
// 解析token，并返回登录者账号信息
|
||
func ParseToken(tokenStr string) (uint64, string, error) {
|
||
if tokenStr == "" {
|
||
return 0, "", errors.New("token error")
|
||
}
|
||
|
||
// Parse token
|
||
token, err := jwt.Parse(tokenStr, func(token *jwt.Token) (any, error) {
|
||
return []byte(config.Conf.Jwt.Key), nil
|
||
})
|
||
if err != nil || token == nil {
|
||
return 0, "", err
|
||
}
|
||
if !token.Valid {
|
||
return 0, "", errors.New("token invalid")
|
||
}
|
||
i := token.Claims.(jwt.MapClaims)
|
||
return uint64(i["id"].(float64)), i["username"].(string), nil
|
||
}
|