72 lines
1.7 KiB
Go
72 lines
1.7 KiB
Go
package fadada
|
||
|
||
import (
|
||
"fmt"
|
||
"strconv"
|
||
"sync"
|
||
"time"
|
||
)
|
||
|
||
// tokenCache accessToken 内存缓存（官方有效期约 2 小时）
|
||
type tokenCache struct {
|
||
mu sync.Mutex
|
||
token string
|
||
expiresAt time.Time
|
||
}
|
||
|
||
func (c *tokenCache) get() (string, bool) {
|
||
c.mu.Lock()
|
||
defer c.mu.Unlock()
|
||
if c.token == "" || time.Now().After(c.expiresAt) {
|
||
return "", false
|
||
}
|
||
return c.token, true
|
||
}
|
||
|
||
func (c *tokenCache) set(token string, ttl time.Duration) {
|
||
c.mu.Lock()
|
||
defer c.mu.Unlock()
|
||
c.token = token
|
||
c.expiresAt = time.Now().Add(ttl)
|
||
}
|
||
|
||
// ensureAccessToken 获取可用 accessToken，过期前自动刷新
|
||
func (c *Client) ensureAccessToken() (string, error) {
|
||
if token, ok := c.tokenCache.get(); ok {
|
||
return token, nil
|
||
}
|
||
|
||
c.tokenMu.Lock()
|
||
defer c.tokenMu.Unlock()
|
||
|
||
if token, ok := c.tokenCache.get(); ok {
|
||
return token, nil
|
||
}
|
||
|
||
res, err := c.http.PostToken(pathGetAccessToken)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
if res.Code != "" && res.Code != successCode {
|
||
return "", fmt.Errorf("获取法大大 accessToken 失败: code=%s msg=%s requestId=%s", res.Code, res.Msg, res.RequestID)
|
||
}
|
||
|
||
var data accessTokenData
|
||
if err := decodeData(res.Data, &data); err != nil {
|
||
return "", fmt.Errorf("解析 accessToken 失败: %w", err)
|
||
}
|
||
if data.AccessToken == "" {
|
||
return "", fmt.Errorf("获取法大大 accessToken 失败: 响应为空 code=%s msg=%s", res.Code, res.Msg)
|
||
}
|
||
|
||
ttl := 2*time.Hour - 5*time.Minute
|
||
if expiresIn, err := strconv.Atoi(data.ExpiresIn); err == nil && expiresIn > 0 {
|
||
ttl = time.Duration(expiresIn)*time.Second - 5*time.Minute
|
||
if ttl < time.Minute {
|
||
ttl = time.Minute
|
||
}
|
||
}
|
||
c.tokenCache.set(data.AccessToken, ttl)
|
||
return data.AccessToken, nil
|
||
}
|