From 221968c6cbf0c74bf2186292aa301acf5172989a Mon Sep 17 00:00:00 2001
From: howell <2827207845@qq.com>
Date: Tue, 25 Jan 2022 11:40:25 +0800
Subject: [PATCH] =?UTF-8?q?fix:=E9=AA=8C=E8=AF=81=E7=A0=81=E9=97=AE?=
 =?UTF-8?q?=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/cache/sms.go                         | 8 ++++----
 internal/v2/controller/app/upgrade/upgrade.go | 2 +-
 internal/v2/logic/app/upgrade/upgrade.go      | 2 +-
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/internal/cache/sms.go b/internal/cache/sms.go
index 0f4a52c..1995edd 100755
--- a/internal/cache/sms.go
+++ b/internal/cache/sms.go
@@ -67,16 +67,16 @@ func GetUserDestroySMSCode(user uint) string {
 	return v
 }
 
-func SetUpgradeSMSCode(user uint, code string) {
-	k := fmt.Sprintf("sms:upgrade:%v", user)
+func SetUpgradeSMSCode(user uint, mobile, code string) {
+	k := fmt.Sprintf("sms:upgrade:%s:%v", mobile, user)
 	e := dbc.Rds.Set(k, code, time.Minute*10).Err()
 	if e != nil {
 		panic(e)
 	}
 }
 
-func GetUpgradeSMSCode(user uint) string {
-	k := fmt.Sprintf("sms:upgrade:%v", user)
+func GetUpgradeSMSCode(user uint, mobile string) string {
+	k := fmt.Sprintf("sms:upgrade:%s:%v", mobile, user)
 	v, err := dbc.Rds.Get(k).Result()
 	if err == redis.Nil {
 		return ""
diff --git a/internal/v2/controller/app/upgrade/upgrade.go b/internal/v2/controller/app/upgrade/upgrade.go
index c81ffa8..d503712 100644
--- a/internal/v2/controller/app/upgrade/upgrade.go
+++ b/internal/v2/controller/app/upgrade/upgrade.go
@@ -91,7 +91,7 @@ func (o Proxy) Send(c *gin.Context) {
 	}
 
 	if response.IsSuccess() {
-		cache.SetUpgradeSMSCode(u.Id, randString)
+		cache.SetUpgradeSMSCode(u.Id, args.Mobile, randString)
 
 		var p message.AliResp
 		if err = json.Unmarshal(response.GetHttpContentBytes(), &p); err != nil {
diff --git a/internal/v2/logic/app/upgrade/upgrade.go b/internal/v2/logic/app/upgrade/upgrade.go
index bb3c509..4dc003d 100644
--- a/internal/v2/logic/app/upgrade/upgrade.go
+++ b/internal/v2/logic/app/upgrade/upgrade.go
@@ -87,7 +87,7 @@ func (o logic) Create(args ApplyCreate) error {
 	// if args.Kind == 1 {
 	// 	state = upgrade.Suc
 	// }
-	if args.Code != cache.GetUpgradeSMSCode(u1.Id) {
+	if args.Code != cache.GetUpgradeSMSCode(u1.Id, args.Mobile) {
 		return fmt.Errorf("验证码不正确")
 	}
 	if err := mysql.Db.Transaction(func(tx *gorm.DB) error {