- refactor verify code

business/auth2/authprovider/defauther.go

@@ -2,11 +2,25 @@ package authprovider
 
 import (
 	"errors"
+	"fmt"
+	"math/rand"
+	"strings"
+	"time"
 
 	"git.rosy.net.cn/baseapi/utils"
 	"git.rosy.net.cn/jx-callback/business/auth2"
 	"git.rosy.net.cn/jx-callback/business/model"
 	"git.rosy.net.cn/jx-callback/business/model/dao"
+	"git.rosy.net.cn/jx-callback/globals"
+	"git.rosy.net.cn/jx-callback/globals/api"
+)
+
+const (
+	DefVerifyCodeDuration = 5 * time.Minute
+
+	VerifyCodeHeader = "VC"
+	VerifyCodeVer    = "V2"
+	TokenTypeSep     = "."
 )
 
 type DefAuther struct {
@@ -85,3 +99,47 @@ func (a *DefAuther) UnionFindAuthBind(curAuthType string, unionAuthTypeList []st
 	}
 	return authBindEx, err
 }
+
+// cache相关
+
+func (a *DefAuther) SaveVerifyCode(keyID, verifyCode string) {
+	api.Cacher.Set(a.buildCacheKey(keyID), verifyCode, DefVerifyCodeDuration)
+}
+
+func (a *DefAuther) LoadVerifyCode(keyID string) (verifyCode string) {
+	if value := api.Cacher.Get(a.buildCacheKey(keyID)); value != nil {
+		return value.(string)
+	}
+	return ""
+}
+
+func (a *DefAuther) DeleteVerifyCode(keyID string) {
+	api.Cacher.Del(a.buildCacheKey(keyID))
+}
+
+// cache相关
+
+func (a *DefAuther) GenerateVerifyCode(keyID string) (verifyCode string) {
+	verifyCode = a.LoadVerifyCode(keyID)
+	if verifyCode == "" {
+		verifyCode = fmt.Sprintf("%06d", rand.Intn(1000000))
+	}
+	globals.SugarLogger.Debugf("GenerateVerifyCode:%s", verifyCode)
+	return verifyCode
+}
+
+func (a *DefAuther) VerifyCode(keyID, verifyCode string) (isSame bool) {
+	savedVerifyCode := a.LoadVerifyCode(keyID)
+	if isSame = (verifyCode != "" && savedVerifyCode != "" && verifyCode == savedVerifyCode); isSame {
+		a.DeleteVerifyCode(keyID)
+	}
+	return isSame
+}
+
+func (a *DefAuther) buildCacheKey(keyID string) string {
+	return strings.Join([]string{
+		VerifyCodeHeader,
+		VerifyCodeVer,
+		keyID,
+	}, TokenTypeSep)
+}