45 lines
1.3 KiB
Go
45 lines
1.3 KiB
Go
package utils
|
||
|
||
import (
|
||
"github.com/golang-jwt/jwt/v5"
|
||
"knowledge/config"
|
||
"time"
|
||
)
|
||
|
||
type Token struct {
|
||
UserId string `json:"user_id"` // 用户id
|
||
Client int `json:"client"` // 客户端（1:前台 2:后台）
|
||
QaId string `json:"qa_id"` // 知识问答id
|
||
jwt.RegisteredClaims // v5版本新加的方法
|
||
}
|
||
|
||
// NewJWT GenerateJWT 生成JWT
|
||
func (t Token) NewJWT(ttl time.Duration) (string, error) {
|
||
if ttl == 0 {
|
||
ttl = time.Duration(config.C.Jwt.Ttl)
|
||
}
|
||
|
||
t.RegisteredClaims.ExpiresAt = jwt.NewNumericDate(time.Now().Add(ttl * time.Hour)) // 过期时间24小时
|
||
t.RegisteredClaims.IssuedAt = jwt.NewNumericDate(time.Now()) // 签发时间
|
||
t.RegisteredClaims.NotBefore = jwt.NewNumericDate(time.Now()) // 生效时间
|
||
|
||
// 使用HS256签名算法
|
||
token := jwt.NewWithClaims(jwt.SigningMethodHS256, t)
|
||
s, err := token.SignedString([]byte(config.C.Jwt.SignKey))
|
||
|
||
return s, err
|
||
}
|
||
|
||
// ParseJwt 解析JWT
|
||
func ParseJwt(authorization string) (*Token, error) {
|
||
t, err := jwt.ParseWithClaims(authorization, &Token{}, func(token *jwt.Token) (interface{}, error) {
|
||
return []byte(config.C.Jwt.SignKey), nil
|
||
})
|
||
|
||
if claims, ok := t.Claims.(*Token); ok && t.Valid {
|
||
return claims, nil
|
||
} else {
|
||
return nil, err
|
||
}
|
||
}
|