113 lines
2.6 KiB
Go
113 lines
2.6 KiB
Go
package sms
|
||
|
||
import (
|
||
"fmt"
|
||
"github.com/aliyun/alibaba-cloud-sdk-go/services/dysmsapi"
|
||
"github.com/gin-gonic/gin"
|
||
"gorm.io/gorm"
|
||
"health-go/config"
|
||
"health-go/models"
|
||
"math/rand"
|
||
"net/http"
|
||
"strconv"
|
||
"time"
|
||
)
|
||
|
||
func Request(c *gin.Context) {
|
||
var user struct {
|
||
Tel string `json:"tel"`
|
||
}
|
||
if err := c.ShouldBindJSON(&user); err != nil {
|
||
c.JSON(http.StatusBadRequest, gin.H{
|
||
"code": 400,
|
||
"msg": "无效的请求参数",
|
||
"data": nil,
|
||
})
|
||
return
|
||
}
|
||
if PreValidate(user.Tel) == "notExist" {
|
||
c.JSON(http.StatusOK, gin.H{
|
||
"code": 200,
|
||
"data": nil,
|
||
"msg": "用户不存在, 正在注册"})
|
||
register(user.Tel)
|
||
return
|
||
}
|
||
|
||
if PreValidate(user.Tel) == "reachLimit" {
|
||
c.JSON(http.StatusOK, gin.H{
|
||
"code": 200,
|
||
"data": nil,
|
||
"msg": "已达到发送上限"})
|
||
return
|
||
}
|
||
|
||
if PreValidate(user.Tel) == "ok" {
|
||
config.DB.Table("user").Where("tel = ?", user.Tel).Update("SMS_try",
|
||
gorm.Expr("SMS_try + ?", 1))
|
||
code := GenerateCode()
|
||
callBack := SendMsg(user.Tel, code)
|
||
if callBack == "success" {
|
||
config.RDC.Set(config.CTX, user.Tel, code, 5*time.Minute)
|
||
c.JSON(http.StatusOK, gin.H{
|
||
"code": 200,
|
||
"data": nil,
|
||
"msg": "验证码已发送"})
|
||
return
|
||
} else if callBack == "frequency_limit" {
|
||
c.JSON(http.StatusOK, gin.H{
|
||
"code": 200,
|
||
"data": nil,
|
||
"msg": "服务端发送频率限制",
|
||
})
|
||
return
|
||
} else {
|
||
|
||
}
|
||
}
|
||
|
||
}
|
||
|
||
func SendMsg(tel string, code string) string {
|
||
client, err := dysmsapi.NewClientWithAccessKey(
|
||
"cn-hangzhou",
|
||
"LTAI5tAYdQcmbGAAiMhGs8sT",
|
||
"PU5WmUbAOsSQ8lQNTMPMD170ONxBLl")
|
||
request := dysmsapi.CreateSendSmsRequest()
|
||
request.Scheme = "https"
|
||
request.PhoneNumbers = tel //手机号变量值
|
||
request.SignName = "Biter健康助手" //签名
|
||
request.TemplateCode = "SMS_472460337" //模板编码
|
||
request.TemplateParam = "{\"code\":\"" + code + "\"}"
|
||
response, err := client.SendSms(request)
|
||
// Aliyun SMS API callback response code
|
||
fmt.Println(response.Code)
|
||
if response.Code == "isv.BUSINESS_LIMIT_CONTROL" {
|
||
return "frequency_limit"
|
||
}
|
||
if err != nil {
|
||
fmt.Print(err.Error())
|
||
return "send failed: " + err.Error()
|
||
}
|
||
return "success"
|
||
}
|
||
|
||
// GenerateCode 生成6位随机验证码
|
||
func GenerateCode() string {
|
||
rand.NewSource(time.Now().UnixNano())
|
||
code := rand.Intn(899999) + 100000
|
||
res := strconv.Itoa(code)
|
||
return res
|
||
}
|
||
|
||
// 注册
|
||
func register(userTel string) {
|
||
user := models.User{Tel: userTel}
|
||
// 写入数据库，用户ID自增
|
||
if err := config.DB.Select("tel").Create(&user).Error; err != nil {
|
||
fmt.Println("register error:", err)
|
||
return
|
||
}
|
||
fmt.Println("register success for", userTel)
|
||
}
|