-
Notifications
You must be signed in to change notification settings - Fork 0
/
jwt.go
88 lines (73 loc) · 1.5 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
package jwt
import (
"errors"
"strings"
"time"
jwt "github.com/dgrijalva/jwt-go"
)
var (
issuer = "jd"
)
// 错误
var (
ErrBadIssuer = errors.New("Token Bad Issuer")
ErrExpired = errors.New("Token Expired")
ErrNotExist = errors.New("Token Not Exist")
)
// Token 令牌
type Token struct {
secret []byte
}
// New 新建
func New(secret []byte) *Token {
return &Token{
secret: secret,
}
}
// CustomClaims 自定义
type CustomClaims struct {
UserID int `json:"userID"`
jwt.StandardClaims
}
// Sign 签名
func (t *Token) Sign(userID int) (string, error) {
// Create the Claims
claims := &CustomClaims{
UserID: userID,
StandardClaims: jwt.StandardClaims{
ExpiresAt: time.Now().AddDate(0, 0, 3).Unix(),
Issuer: issuer,
},
}
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
ss, err := token.SignedString(t.secret)
if err != nil {
return ss, err
}
return ss, nil
}
// Verify 校验
func (t *Token) Verify(tokenString string) (int, error) {
if strings.TrimSpace(tokenString) == "" {
return 0, nil
}
token, err := jwt.ParseWithClaims(tokenString, &CustomClaims{}, func(token *jwt.Token) (interface{}, error) {
return t.secret, nil
})
if err != nil {
return 0, err
}
var userID int
if claims, ok := token.Claims.(*CustomClaims); ok && token.Valid {
if claims.Issuer != issuer {
return 0, ErrBadIssuer
}
if claims.ExpiresAt <= time.Now().Unix() {
return 0, ErrExpired
}
userID = claims.UserID
} else {
return 0, ErrNotExist
}
return userID, nil
}