- remove legacy

business/jxutils/tasks/configrefresh.go

@@ -0,0 +1,124 @@
+package tasks
+
+import (
+	"fmt"
+	"time"
+
+	"git.rosy.net.cn/jx-callback/business/model/legacymodel"
+	"git.rosy.net.cn/jx-callback/globals/api"
+
+	"git.rosy.net.cn/baseapi/utils"
+	"git.rosy.net.cn/jx-callback/globals"
+
+	"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)) error {
+	sleepGap := expiresTime / 10
+	needRefreshGap := expiresTime * 8 / 10
+	if sleepGap > maxRefreshGap {
+		sleepGap = maxRefreshGap
+	}
+	refreshFunc := func() (string, error) {
+		curConfig := &legacymodel.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.Errorf("db error:%v, curConfig:%v", err, curConfig)
+				return "", err
+			}
+			handleType = 2
+		} else if curConfig.Date <= utils.Time2Str(time.Now().Add(-needRefreshGap)) {
+			handleType = 1
+		}
+		if handleType != 0 {
+			if curConfig.Token = configGetter(); curConfig.Token == "" {
+				return "", fmt.Errorf("can not get token for %s", configKey)
+			}
+			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, curConfig:%v", err, num, curConfig)
+				return "", err
+			} else {
+				configSetter(curConfig.Token)
+			}
+		}
+		return curConfig.Token, nil
+	}
+	token, err := refreshFunc() // 这样写的目的是强制第一次调用时要刷新一次
+	if err == nil {
+		configSetter(token)
+		utils.CallFuncAsync(func() {
+			for {
+				time.Sleep(sleepGap)
+				refreshFunc()
+			}
+		})
+	}
+	return err
+}
+
+func RefreshWeixinToken() error {
+	return RefreshConfig("wechat", weixinTokenExpires, func() string {
+		if tokenInfo, err := api.WeixinAPI.RefreshToken(); err == nil {
+			return tokenInfo.AccessToken
+		}
+		return ""
+	}, func(value string) {
+		api.WeixinAPI.SetToken(value)
+	})
+}
+
+func RefreshElmToken() error {
+	return RefreshConfig("eleme", elmTokenExpires, func() string {
+		if tokenInfo, err := api.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) {
+		var tokenInfo ElmTokenForCompatible
+		err := utils.UnmarshalUseNumber([]byte(value), &tokenInfo)
+		if err == nil {
+			api.ElmAPI.SetToken(tokenInfo.AccessToken)
+		}
+	})
+}