package jwts
import (
"fmt"
"github.com/dgrijalva/jwt-go"
"github.com/kataras/iris/v12"
"strings"
"time"
)
// 生成token
func (config *ClaimsConfigJson) TokenObtain() (string, error) {
config.IssuedAt = time.Now().Unix()
config.ExpiresAt = time.Now().Add(ExpireTime).Unix()
token := jwt.NewWithClaims(jwt.SigningMethodHS256, config)
signedToken, err := token.SignedString([]byte(SignKey))
if err != nil {
return "", err
}
return signedToken, nil
}
// =======================
// 刷新token
func (config *ClaimsConfigJson) tokenRefresh(ctx iris.Context) {
jwtObj := ctx.GetHeader("Authorization")
if jwtObj == "" {
_, _ = ctx.JSON("未登录或已失效")
return
}
StrToken := strings.Split(jwtObj, " ")[1]
token, err := jwt.ParseWithClaims(StrToken, &ClaimsConfigJson{}, func(token *jwt.Token) (interface{}, error) {
return []byte(SignKey), nil
})
if err != nil {
println(err)
return
}
claims, ok := token.Claims.(*ClaimsConfigJson)
if !ok {
println("test")
return
}
if err := token.Claims.Valid(); err != nil {
println(err)
return
}
claims.UserId = config.UserId
claims.ExpiresAt = time.Now().Unix() + (claims.ExpiresAt - claims.IssuedAt)
newToken := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
signedToken, err := newToken.SignedString([]byte(SignKey))
if err != nil {
println(err)
return
}
fmt.Println(signedToken)
}
// =====================
// 路由中间件
func JwtHandle(ctx iris.Context) {
jwtObj := ctx.GetHeader("Authorization")
if jwtObj == "" {
_, _ = ctx.JSON("未登录或已失效")
return
}
jwtStr := strings.Split(jwtObj, " ")[1]
token, err := jwt.ParseWithClaims(jwtStr, &ClaimsConfigJson{}, func(token *jwt.Token) (interface{}, error) {
return []byte(SignKey), nil
})
if err != nil {
_, _ = ctx.JSON("令牌已过期")
return
}
claims, ok := token.Claims.(*ClaimsConfigJson)
if !ok {
_, _ = ctx.JSON("解析用户数据错误")
return
}
if err := token.Claims.Valid(); err != nil {
_, _ = ctx.JSON("令牌已失效")
return
}
ctx.Values().Set(DataKey, claims)
ctx.Next()
}