Эх сурвалжийг харах

233-登录验证码获取限制

gmy 7 сар өмнө
parent
commit
e1a0787449
1 өөрчлөгдсөн 10 нэмэгдсэн , 0 устгасан
  1. 10 0
      logic/user/user.go

+ 10 - 0
logic/user/user.go

@@ -1,7 +1,9 @@
 package user
 
 import (
+	"context"
 	"errors"
+	"hongze/hongze_yb/global"
 	userReq "hongze/hongze_yb/models/request/user"
 	admin2 "hongze/hongze_yb/models/tables/admin"
 	"hongze/hongze_yb/models/tables/company"
@@ -432,6 +434,13 @@ func GetUserTabBar(userInfo user.UserInfo, version string) (list []string, err e
 
 // PcSendSmsCode 发送手机短信
 func PcSendSmsCode(mobile, areaNum string) (err error, errMsg string) {
+	cacheKey := utils.HONGZEYB_ + "REPORT_SMS_CODE_PC:GetSmsCode:" + areaNum + ":" + mobile
+	smsCodeStr, _ := global.Redis.Get(context.TODO(), cacheKey).Result()
+	if smsCodeStr != "" {
+		err = errors.New("请勿频繁发送验证码")
+		return
+	}
+
 	if mobile == "" {
 		err = errors.New("请输入手机号")
 		return
@@ -456,6 +465,7 @@ func PcSendSmsCode(mobile, areaNum string) (err error, errMsg string) {
 			LastUpdatedTime: time.Time{},
 		}
 		err = item.Create()
+		global.Redis.Set(context.TODO(), cacheKey, msgCode, 60*time.Second)
 	} else {
 		err = errors.New("短信发送失败")
 	}