huangzhenpc
1f22d05270
UniSMS 通用模板要求 code 为纯数字，原来用 UUID 截取会包含字母。 新增 GenerateNumericCode 函数，SMS 验证码专用。 Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
89 lines
1.9 KiB
Go
89 lines
1.9 KiB
Go
package common
|
|
|
|
import (
|
|
"math/rand"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
type verificationValue struct {
|
|
code string
|
|
time time.Time
|
|
}
|
|
|
|
const (
|
|
EmailVerificationPurpose = "v"
|
|
PasswordResetPurpose = "r"
|
|
SmsVerificationPurpose = "s"
|
|
)
|
|
|
|
var verificationMutex sync.Mutex
|
|
var verificationMap map[string]verificationValue
|
|
var verificationMapMaxSize = 10
|
|
var VerificationValidMinutes = 10
|
|
|
|
func GenerateVerificationCode(length int) string {
|
|
code := uuid.New().String()
|
|
code = strings.Replace(code, "-", "", -1)
|
|
if length == 0 {
|
|
return code
|
|
}
|
|
return code[:length]
|
|
}
|
|
|
|
func GenerateNumericCode(length int) string {
|
|
digits := make([]byte, length)
|
|
for i := range digits {
|
|
digits[i] = '0' + byte(rand.Intn(10))
|
|
}
|
|
return string(digits)
|
|
}
|
|
|
|
func RegisterVerificationCodeWithKey(key string, code string, purpose string) {
|
|
verificationMutex.Lock()
|
|
defer verificationMutex.Unlock()
|
|
verificationMap[purpose+key] = verificationValue{
|
|
code: code,
|
|
time: time.Now(),
|
|
}
|
|
if len(verificationMap) > verificationMapMaxSize {
|
|
removeExpiredPairs()
|
|
}
|
|
}
|
|
|
|
func VerifyCodeWithKey(key string, code string, purpose string) bool {
|
|
verificationMutex.Lock()
|
|
defer verificationMutex.Unlock()
|
|
value, okay := verificationMap[purpose+key]
|
|
now := time.Now()
|
|
if !okay || int(now.Sub(value.time).Seconds()) >= VerificationValidMinutes*60 {
|
|
return false
|
|
}
|
|
return code == value.code
|
|
}
|
|
|
|
func DeleteKey(key string, purpose string) {
|
|
verificationMutex.Lock()
|
|
defer verificationMutex.Unlock()
|
|
delete(verificationMap, purpose+key)
|
|
}
|
|
|
|
// no lock inside, so the caller must lock the verificationMap before calling!
|
|
func removeExpiredPairs() {
|
|
now := time.Now()
|
|
for key := range verificationMap {
|
|
if int(now.Sub(verificationMap[key].time).Seconds()) >= VerificationValidMinutes*60 {
|
|
delete(verificationMap, key)
|
|
}
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
verificationMutex.Lock()
|
|
defer verificationMutex.Unlock()
|
|
verificationMap = make(map[string]verificationValue)
|
|
}
|