55 lines
1.3 KiB
Go
55 lines
1.3 KiB
Go
package common
|
||
|
||
import (
|
||
"github.com/dgrijalva/jwt-go"
|
||
"ssh_manage/config"
|
||
"time"
|
||
)
|
||
|
||
func init() {
|
||
if config.Config.Jwt.Key == "ss_jwt_token" {
|
||
panic("请先修改Jwt key！")
|
||
}
|
||
}
|
||
|
||
var jwt_ket = []byte(config.Config.Jwt.Key)
|
||
|
||
type Claims struct {
|
||
Userid uint
|
||
jwt.StandardClaims
|
||
}
|
||
|
||
func ReleaseToken(id uint) (token string, err error) {
|
||
expire_time := time.Now().Add(3 * 24 * time.Hour)
|
||
claims := &Claims{
|
||
Userid: id,
|
||
StandardClaims: jwt.StandardClaims{
|
||
ExpiresAt: expire_time.Unix(),
|
||
IssuedAt: time.Now().Unix(),
|
||
Issuer: "admin",
|
||
Subject: "user",
|
||
},
|
||
}
|
||
token_obj := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
|
||
token, err = token_obj.SignedString(jwt_ket)
|
||
return
|
||
}
|
||
|
||
func ParseToken(token string) (*Claims, error) {
|
||
|
||
//用于解析鉴权的声明，方法内部主要是具体的解码和校验的过程，最终返回*Token
|
||
tokenClaims, err := jwt.ParseWithClaims(token, &Claims{}, func(token *jwt.Token) (interface{}, error) {
|
||
return jwt_ket, nil
|
||
})
|
||
|
||
if tokenClaims != nil {
|
||
// 从tokenClaims中获取到Claims对象，并使用断言，将该对象转换为我们自己定义的Claims
|
||
// 要传入指针，项目中结构体都是用指针传递，节省空间。
|
||
if claims, ok := tokenClaims.Claims.(*Claims); ok && tokenClaims.Valid {
|
||
return claims, nil
|
||
}
|
||
}
|
||
return nil, err
|
||
|
||
}
|