-
Notifications
You must be signed in to change notification settings - Fork 2
/
jwt.go
96 lines (80 loc) · 2.24 KB
/
jwt.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
package multi
import (
"fmt"
"github.com/golang-jwt/jwt"
)
var hmacSampleSecret = []byte("updPA0L2uQ56LwHZoyUX")
// JwtAuth
type JwtAuth struct {
HmacSecret []byte
}
// NewJwtAuth
func NewJwtAuth(hmacSecret []byte) *JwtAuth {
ja := &JwtAuth{
HmacSecret: hmacSecret,
}
if ja.HmacSecret == nil {
ja.HmacSecret = hmacSampleSecret
}
return ja
}
// GenerateToken
func (ra *JwtAuth) GenerateToken(claims *MultiClaims) (string, int64, error) {
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
// Sign and get the complete encoded token as a string using the secret
tokenString, err := token.SignedString(ra.HmacSecret)
if err != nil {
return "", 0, err
}
return tokenString, 0, nil
}
// GetTokenByClaims 获取用户信息
func (ra *JwtAuth) GetTokenByClaims(cla *MultiClaims) (string, error) {
return "", nil
}
// GetMultiClaims 获取用户信息
func (ra *JwtAuth) GetMultiClaims(tokenString string) (*MultiClaims, error) {
mc := &MultiClaims{}
token, err := jwt.ParseWithClaims(tokenString, mc, func(token *jwt.Token) (interface{}, error) {
// Don't forget to validate the alg is what you expect:
if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok {
return nil, fmt.Errorf("不支持的签名方法: %v", token.Header["alg"])
}
return ra.HmacSecret, nil
})
if err != nil {
return nil, err
}
if _, ok := token.Claims.(*MultiClaims); ok && token.Valid {
return mc, nil
} else {
return nil, ErrTokenInvalid
}
}
// SetUserTokenMaxCount 最大登录限制
func (ra *JwtAuth) SetUserTokenMaxCount(tokenMaxCount int64) error {
return nil
}
//UpdateUserTokenCacheExpire 更新过期时间
func (ra *JwtAuth) UpdateUserTokenCacheExpire(token string) error {
return nil
}
// DelUserTokenCache 删除token缓存
func (ra *JwtAuth) DelUserTokenCache(token string) error {
return nil
}
// CleanUserTokenCache 清空token缓存
func (ra *JwtAuth) CleanUserTokenCache(authorityType int, userId string) error {
return nil
}
// IsRole
func (ra *JwtAuth) IsRole(token string, authorityType int) (bool, error) {
rcc, err := ra.GetMultiClaims(token)
if err != nil {
return false, fmt.Errorf("get User's infomation return error: %w", err)
}
return rcc.AuthorityType == authorityType, nil
}
// Close
func (ra *JwtAuth) Close() {
}