125 lines
3.3 KiB
Go
125 lines
3.3 KiB
Go
package tasks
|
|
|
|
import (
|
|
"time"
|
|
|
|
"git.rosy.net.cn/jx-callback/globals/globals2"
|
|
|
|
"git.rosy.net.cn/baseapi/utils"
|
|
"git.rosy.net.cn/jx-callback/globals"
|
|
|
|
"git.rosy.net.cn/jx-callback/business/models"
|
|
"github.com/astaxie/beego/orm"
|
|
)
|
|
|
|
const (
|
|
weixinTokenExpires = 7200 * time.Second
|
|
elmTokenExpires = 20 * 24 * 3600 * time.Second
|
|
maxRefreshGap = 5 * 60 * time.Second
|
|
)
|
|
|
|
type ElmTokenForCompatible struct {
|
|
Error string `json:"error"`
|
|
ErrorDescription string `json:"error_description"`
|
|
AccessToken string `json:"accessToken"`
|
|
TokenType string `json:"tokenType"`
|
|
Expires int `json:"expires"`
|
|
RefreshToken string `json:"refreshToken"`
|
|
Success bool `json:"success"`
|
|
}
|
|
|
|
func RefreshConfig(configKey string, expiresTime time.Duration, configGetter func() string, configSetter func(value string) bool) {
|
|
go func() {
|
|
sleepGap := expiresTime / 10
|
|
needRefreshGap := expiresTime * 8 / 10
|
|
if sleepGap > maxRefreshGap {
|
|
sleepGap = maxRefreshGap
|
|
}
|
|
|
|
for {
|
|
curConfig := &models.Config{
|
|
Thirdparty: configKey,
|
|
}
|
|
|
|
db := orm.NewOrm()
|
|
// 0: don't refresh, 1 update, 2 insert
|
|
handleType := 0
|
|
if err := db.Read(curConfig, "Thirdparty"); err != nil {
|
|
if err != orm.ErrNoRows {
|
|
globals.SugarLogger.Error("db error:%v", err)
|
|
} else {
|
|
curConfig.Token = configGetter()
|
|
handleType = 2
|
|
}
|
|
} else {
|
|
latestTimeStr := utils.Time2Str(time.Now().Add(-needRefreshGap))
|
|
if curConfig.Date <= latestTimeStr {
|
|
curConfig.Token = configGetter()
|
|
handleType = 1
|
|
} else {
|
|
configSetter(curConfig.Token)
|
|
}
|
|
}
|
|
|
|
if handleType != 0 {
|
|
globals.SugarLogger.Debugf("refresh %s, value:%s", configKey, curConfig.Token)
|
|
curConfig.Date = utils.GetCurTimeStr()
|
|
var num int64
|
|
var err error
|
|
|
|
if handleType == 1 {
|
|
num, err = db.Update(curConfig, "Token", "Date")
|
|
} else {
|
|
num, err = db.Insert(curConfig)
|
|
}
|
|
if err != nil || num == 0 {
|
|
globals.SugarLogger.Errorf("db error:%v, num:%d", err, num)
|
|
} else {
|
|
if configKey == "wechat" {
|
|
globals2.FreshFoodAPI.AccessFreshFood("weixin/accesstoken/update", nil)
|
|
} else if configKey == "eleme" {
|
|
globals2.FreshFoodAPI.AccessFreshFood("eleme/token/update", nil)
|
|
}
|
|
}
|
|
}
|
|
time.Sleep(sleepGap)
|
|
}
|
|
}()
|
|
}
|
|
|
|
func RefreshWeixinToken() {
|
|
RefreshConfig("wechat", weixinTokenExpires, func() string {
|
|
if tokenInfo, err := globals2.WeixinAPI.RefreshToken(); err == nil {
|
|
return tokenInfo.AccessToken
|
|
}
|
|
return ""
|
|
}, func(value string) bool {
|
|
return globals2.WeixinAPI.SetToken(value)
|
|
})
|
|
}
|
|
|
|
func RefreshElmToken() {
|
|
RefreshConfig("eleme", elmTokenExpires, func() string {
|
|
if tokenInfo, err := globals2.ElmAPI.RefreshTokenIndividual(); err == nil {
|
|
tokenInfo2 := &ElmTokenForCompatible{
|
|
Error: "",
|
|
ErrorDescription: "",
|
|
AccessToken: tokenInfo.AccessToken,
|
|
TokenType: tokenInfo.TokenType,
|
|
Expires: tokenInfo.ExpiresIn,
|
|
RefreshToken: "",
|
|
Success: true,
|
|
}
|
|
return string(utils.MustMarshal(tokenInfo2))
|
|
}
|
|
return ""
|
|
}, func(value string) bool {
|
|
var tokenInfo ElmTokenForCompatible
|
|
err := utils.UnmarshalUseNumber([]byte(value), &tokenInfo)
|
|
if err == nil {
|
|
return globals2.ElmAPI.SetToken(tokenInfo.AccessToken)
|
|
}
|
|
return false
|
|
})
|
|
}
|