From e4d4598e206c04f05c48f82f68ffe75281240811 Mon Sep 17 00:00:00 2001 From: JustSong Date: Sat, 5 Nov 2022 21:42:55 +0800 Subject: [PATCH] Delete code after verification --- common/verification.go | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/common/verification.go b/common/verification.go index 68be561..f01d179 100644 --- a/common/verification.go +++ b/common/verification.go @@ -54,6 +54,9 @@ func GetWeChatIDByCode(code string) string { verificationMutex.Lock() defer verificationMutex.Unlock() value, okay := verificationMap[WeChatVerificationPurpose+code] + if okay { + delete(verificationMap, WeChatVerificationPurpose+code) + } now := time.Now() if !okay || int(now.Sub(value.time).Seconds()) >= VerificationValidMinutes*60 { return "" @@ -78,6 +81,9 @@ func VerifyCodeWithKey(key string, code string, purpose string) bool { defer verificationMutex.Unlock() value, okay := verificationMap[purpose+key] now := time.Now() + if okay { + delete(verificationMap, purpose+key) + } if !okay || int(now.Sub(value.time).Seconds()) >= VerificationValidMinutes*60 { return false }