/
access_token.go
78 lines (61 loc) · 1.93 KB
/
access_token.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
package wxwork
import (
"encoding/json"
"fmt"
"sync"
"time"
)
// AccessToken 定义了获取 access_token 时的响应
type AccessToken struct {
AccessToken string `json:"access_token"` // 获取到的凭证,最长为512字节
ExpiresIn int64 `json:"expires_in,omitempty"` // 凭证的有效时间(秒),通常为2小时(7200秒)
ExpireAt time.Time `json:"expire_at,omitempty"` // 过期时间,超过时重新获取
baseCaller
mu sync.Mutex
}
// IsExpire 验证 access_token 是否过期
func (token *AccessToken) IsExpire() bool {
return token.ExpireAt.Before(time.Now())
}
// RefreshAccessToken 用于刷新 access_token
func (a *Agent) RefreshAccessToken() error {
a.accessToken.mu.Lock()
defer a.accessToken.mu.Unlock()
var token AccessToken
path := fmt.Sprintf("%sgettoken?corpid=%s&corpsecret=%s", BaseURL, a.corpID, a.secret)
err := a.Execute("GET", path, nil, &token)
if err != nil {
return err
}
token.ExpireAt = time.Now().Add(time.Duration(token.ExpiresIn) * time.Second)
a.accessToken = &token
if a.cache != nil {
bt, _ := json.Marshal(&token)
a.cache.Set("access_token", bt)
}
return nil
}
// getAccessTokenFromCache 从缓存中获取 access_token
func (a *Agent) getAccessTokenFromCache() (string, error) {
if a.cache == nil {
return "", fmt.Errorf("client cache processor not found")
}
accessToken := a.cache.Get("access_token")
err := json.Unmarshal(accessToken, &a.accessToken)
if a.accessToken.IsExpire() || a.accessToken.AccessToken == "" {
err = a.RefreshAccessToken()
}
return a.accessToken.AccessToken, err
}
// GetAccessToken 获取access_token
func (a *Agent) GetAccessToken() (string, error) {
// 如果设置了 缓存器,从缓存器中获取 token,防止频繁刷新
if a.cache != nil {
return a.getAccessTokenFromCache()
}
var err error
if a.accessToken.IsExpire() {
err = a.RefreshAccessToken()
}
return a.accessToken.AccessToken, err
}