- elm and weixin token refresh added.

business/tasks/configrefresh.go

@@ -0,0 +1,90 @@
+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
+)
+
+func RefreshConfig(configKey string, expiresTime time.Duration, configGetter func() string) {
+	go func() {
+		sleepGap := expiresTime / 10
+		needRefreshGap := expiresTime * 8 / 10
+
+		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
+				}
+			}
+
+			if handleType != 0 {
+				globals.SugarLogger.Debug(configKey)
+				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 RefreshElmToken() {
+	RefreshConfig("eleme", elmTokenExpires, func() string {
+		if tokenInfo, err := globals2.ElmAPI.RefreshToken(); err == nil {
+			return string(utils.MustMarshal(tokenInfo))
+		}
+		return ""
+	})
+}