forked from chanxuehong/wechat
/
card_ticket_server.go
166 lines (144 loc) · 4.97 KB
/
card_ticket_server.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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
package jssdk
import (
"errors"
"math/rand"
"strconv"
"sync/atomic"
"time"
"unsafe"
"github.com/charsunny/wechat/mp/core"
)
// 卡劵 api_ticket 中控服务器接口.
type CardTicketServer interface {
Ticket() (ticket string, err error) // 请求中控服务器返回缓存的卡劵 api_ticket
RefreshTicket(currentTicket string) (ticket string, err error) // 请求中控服务器刷新卡劵 api_ticket
IIDB9BDD0A1E1DC11E5844AA4DB30FED8E1() // 接口标识, 没有实际意义
}
var _ CardTicketServer = (*DefaultCardTicketServer)(nil)
// DefaultCardTicketServer 实现了 CardTicketServer 接口.
// NOTE:
// 1. 用于单进程环境.
// 2. 因为 DefaultCardTicketServer 同时也是一个简单的中控服务器, 而不是仅仅实现 CardTicketServer 接口,
// 所以整个系统只能存在一个 DefaultCardTicketServer 实例!
type DefaultCardTicketServer struct {
coreClient *core.Client
refreshTicketRequestChan chan string // chan currentTicket
refreshTicketResponseChan chan refreshTicketResult // chan {ticket, err}
ticketCache unsafe.Pointer // *cardApiTicket
}
// NewDefaultCardTicketServer 创建一个新的 DefaultCardTicketServer.
func NewDefaultCardTicketServer(clt *core.Client) (srv *DefaultCardTicketServer) {
if clt == nil {
panic("nil core.Client")
}
srv = &DefaultCardTicketServer{
coreClient: clt,
refreshTicketRequestChan: make(chan string),
refreshTicketResponseChan: make(chan refreshTicketResult),
}
go srv.ticketUpdateDaemon(time.Hour * 24 * time.Duration(100+rand.Int63n(200)))
return
}
func (srv *DefaultCardTicketServer) IIDB9BDD0A1E1DC11E5844AA4DB30FED8E1() {}
func (srv *DefaultCardTicketServer) Ticket() (ticket string, err error) {
if p := (*cardApiTicket)(atomic.LoadPointer(&srv.ticketCache)); p != nil {
return p.Ticket, nil
}
return srv.RefreshTicket("")
}
//type refreshTicketResult struct {
// ticket string
// err error
//}
func (srv *DefaultCardTicketServer) RefreshTicket(currentTicket string) (ticket string, err error) {
srv.refreshTicketRequestChan <- currentTicket
rslt := <-srv.refreshTicketResponseChan
return rslt.ticket, rslt.err
}
func (srv *DefaultCardTicketServer) ticketUpdateDaemon(initTickDuration time.Duration) {
tickDuration := initTickDuration
NEW_TICK_DURATION:
ticker := time.NewTicker(tickDuration)
for {
select {
case currentTicket := <-srv.refreshTicketRequestChan:
cardApiTicket, cached, err := srv.updateTicket(currentTicket)
if err != nil {
srv.refreshTicketResponseChan <- refreshTicketResult{err: err}
break
}
srv.refreshTicketResponseChan <- refreshTicketResult{ticket: cardApiTicket.Ticket}
if !cached {
tickDuration = time.Duration(cardApiTicket.ExpiresIn) * time.Second
ticker.Stop()
goto NEW_TICK_DURATION
}
case <-ticker.C:
cardApiTicket, _, err := srv.updateTicket("")
if err != nil {
break
}
newTickDuration := time.Duration(cardApiTicket.ExpiresIn) * time.Second
if abs(tickDuration-newTickDuration) > time.Second*5 {
tickDuration = newTickDuration
ticker.Stop()
goto NEW_TICK_DURATION
}
}
}
}
//func abs(x time.Duration) time.Duration {
// if x >= 0 {
// return x
// }
// return -x
//}
type cardApiTicket struct {
Ticket string `json:"ticket"`
ExpiresIn int64 `json:"expires_in"`
}
// updateTicket 从微信服务器获取新的卡劵 api_ticket 并存入缓存, 同时返回该卡劵 api_ticket.
func (srv *DefaultCardTicketServer) updateTicket(currentTicket string) (ticket *cardApiTicket, cached bool, err error) {
if currentTicket != "" {
if p := (*cardApiTicket)(atomic.LoadPointer(&srv.ticketCache)); p != nil && currentTicket != p.Ticket {
return p, true, nil // 无需更改 p.ExpiresIn 参数值, cached == true 时用不到
}
}
var incompleteURL = "https://api.weixin.qq.com/cgi-bin/ticket/getticket?type=wx_card&access_token="
var result struct {
core.Error
cardApiTicket
}
if err = srv.coreClient.GetJSON(incompleteURL, &result); err != nil {
atomic.StorePointer(&srv.ticketCache, nil)
return
}
if result.ErrCode != core.ErrCodeOK {
atomic.StorePointer(&srv.ticketCache, nil)
err = &result.Error
return
}
// 由于网络的延时, 卡劵 api_ticket 过期时间留有一个缓冲区
switch {
case result.ExpiresIn > 31556952: // 60*60*24*365.2425
atomic.StorePointer(&srv.ticketCache, nil)
err = errors.New("expires_in too large: " + strconv.FormatInt(result.ExpiresIn, 10))
return
case result.ExpiresIn > 60*60:
result.ExpiresIn -= 60 * 10
case result.ExpiresIn > 60*30:
result.ExpiresIn -= 60 * 5
case result.ExpiresIn > 60*5:
result.ExpiresIn -= 60
case result.ExpiresIn > 60:
result.ExpiresIn -= 10
default:
atomic.StorePointer(&srv.ticketCache, nil)
err = errors.New("expires_in too small: " + strconv.FormatInt(result.ExpiresIn, 10))
return
}
ticketCopy := result.cardApiTicket
atomic.StorePointer(&srv.ticketCache, unsafe.Pointer(&ticketCopy))
ticket = &ticketCopy
return
}