64 lines
1.2 KiB
Go
64 lines
1.2 KiB
Go
package models
|
|
|
|
import (
|
|
"fmt"
|
|
"main/configs"
|
|
"math/rand"
|
|
"time"
|
|
)
|
|
|
|
type Code struct {
|
|
Email string `json:"email"`
|
|
Mobile string `json:"mobile"`
|
|
Code string `json:"code"`
|
|
Expire time.Time `json:"expire"`
|
|
}
|
|
|
|
func init() {
|
|
configs.ORMDB().AutoMigrate(&Code{})
|
|
}
|
|
|
|
func CodeCreate(email, mobile string) (err error) {
|
|
code := fmt.Sprintf("%06v", rand.New(rand.NewSource(time.Now().UnixNano())).Int31n(1000000))
|
|
// 如果是邮箱，发送邮件
|
|
if email != "" {
|
|
go func() {
|
|
// 发送邮件
|
|
// ...
|
|
}()
|
|
}
|
|
// 如果是手机，发送短信
|
|
if mobile != "" {
|
|
go func() {
|
|
// 发送短信
|
|
// ...
|
|
}()
|
|
}
|
|
// 保存验证码
|
|
configs.ORMDB().Create(&Code{
|
|
Email: email,
|
|
Mobile: mobile,
|
|
Code: code,
|
|
Expire: time.Now().Add(time.Minute * 5),
|
|
})
|
|
return nil
|
|
}
|
|
|
|
func EmailCheck(email, code string) (err error) {
|
|
var data Code
|
|
configs.ORMDB().Where("email = ?", email).First(&data)
|
|
if data.Code == code && data.Expire.After(time.Now()) {
|
|
return nil
|
|
}
|
|
return fmt.Errorf("验证码错误")
|
|
}
|
|
|
|
func MobileCheck(mobile, code string) (err error) {
|
|
var data Code
|
|
configs.ORMDB().Where("mobile = ?", mobile).First(&data)
|
|
if data.Code == code && data.Expire.After(time.Now()) {
|
|
return nil
|
|
}
|
|
return fmt.Errorf("验证码错误")
|
|
}
|