wqy
/
FuShouXian-Backend
1
0
Fork 0
Code Issues Pull Requests 2 Packages Projects Releases Wiki Activity

update

Browse Source
zhangzeliang
liangliangit 2 years ago
parent e4a276eea3
commit 335b79f482
2 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File

14
module/usercenter/SMSVerificationCode.go
Unescape View File

@ -8,26 +8,26 @@ import (
"time" "time"
) )
var regSMSVerificationCode sync.Map var loginOrregSMSVerificationCode sync.Map
// 登记注册用验证码 // 登记注册用验证码
func StoreRegSMSVerificationCode(mobile string) (string, error) { func StoreLoginOrloginOrRegSMSVerificationCode(mobile string) (string, error) {
codeInterface, ok := regSMSVerificationCode.Load(mobile) codeInterface, ok := loginOrregSMSVerificationCode.Load(mobile)
if ok { if ok {
return codeInterface.(string), nil return codeInterface.(string), nil
} }
verificationCode := randCharNumber(6) verificationCode := randCharNumber(6)
regSMSVerificationCode.Store(mobile, verificationCode) loginOrregSMSVerificationCode.Store(mobile, verificationCode)
go func(mobile string) { go func(mobile string) {
time.Sleep(time.Second * 5 * 60) time.Sleep(time.Second * 5 * 60)
regSMSVerificationCode.Delete(mobile) loginOrregSMSVerificationCode.Delete(mobile)
}(mobile) }(mobile)
return verificationCode, nil return verificationCode, nil
} }
// 验证注册验证码 // 验证注册验证码
func RegSMSVerification(mobile, verificationCode string) bool { func LoginOrRegSMSVerification(mobile, verificationCode string) bool {
codeInterface, ok := regSMSVerificationCode.LoadAndDelete(mobile) codeInterface, ok := loginOrregSMSVerificationCode.LoadAndDelete(mobile)
if !ok { if !ok {
return false return false
} }

10
module/usercenter/api.user.go
Unescape View File

@ -30,7 +30,7 @@ func InitManageApi() {
towgo.SetFunc(_methodHead+"/user/regByMobile", userRegByMobile) towgo.SetFunc(_methodHead+"/user/regByMobile", userRegByMobile)
//获取注册短信验证码 //获取注册短信验证码
towgo.SetFunc(_methodHead+"/user/getRegSMSVerificationCode", getRegSMSVerificationCode) towgo.SetFunc(_methodHead+"/user/getLoginOrRegSMSVerificationCode", getLoginOrRegSMSVerificationCode)
//修改密码 F //修改密码 F
towgo.SetFunc(_methodHead+"/user/changepassword", userChangepassword) towgo.SetFunc(_methodHead+"/user/changepassword", userChangepassword)
@ -45,7 +45,7 @@ func isPhoneNumber(input string) bool {
return reg.MatchString(input) return reg.MatchString(input)
} }
func getRegSMSVerificationCode(rpcConn towgo.JsonRpcConnection) { func getLoginOrRegSMSVerificationCode(rpcConn towgo.JsonRpcConnection) {
var params struct { var params struct {
Mobile string `json:"mobile"` Mobile string `json:"mobile"`
} }
@ -56,7 +56,7 @@ func getRegSMSVerificationCode(rpcConn towgo.JsonRpcConnection) {
return return
} }
code, err := StoreRegSMSVerificationCode(params.Mobile) code, err := StoreLoginOrloginOrRegSMSVerificationCode(params.Mobile)
if err != nil { if err != nil {
rpcConn.WriteError(500, err.Error()) rpcConn.WriteError(500, err.Error())
return return
@ -92,7 +92,7 @@ func userRegByMobile(rpcConn towgo.JsonRpcConnection) {
return return
} }
if !RegSMSVerification(params.Mobile, params.VerificationCode) { if !LoginOrRegSMSVerification(params.Mobile, params.VerificationCode) {
rpcConn.WriteError(500, "验证码错误") rpcConn.WriteError(500, "验证码错误")
return return
} }
@ -186,7 +186,7 @@ func userLoginOrRegByMobile(rpcConn towgo.JsonRpcConnection) {
return return
} }
if !RegSMSVerification(params.Mobile, params.VerificationCode) { if !LoginOrRegSMSVerification(params.Mobile, params.VerificationCode) {
rpcConn.WriteError(500, "验证码错误") rpcConn.WriteError(500, "验证码错误")
return return
} }

Write Preview
Loading…
Cancel
Save