refactor: adjust project structure

service/sms/RequestHandler.go

@@ -0,0 +1,112 @@
+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/model"
+	"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 := model.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)
+}

service/sms/ValidateHandler.go

@@ -0,0 +1,52 @@
+package sms
+
+import (
+	"errors"
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"github.com/redis/go-redis/v9"
+	"health-go/config"
+	"net/http"
+)
+
+func Validate(c *gin.Context) {
+	var user struct {
+		Tel  string `json:"tel"`
+		Code string `json:"code"`
+	}
+	if err := c.ShouldBindJSON(&user); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{
+			"code": 400,
+			"msg":  "无效的请求参数",
+			"data": nil,
+		})
+		return
+	}
+
+	storageCode, err := config.RDC.Get(config.CTX, user.Tel).Result()
+	fmt.Println("storageCode:", storageCode)
+
+	if errors.Is(err, redis.Nil) {
+		c.JSON(http.StatusOK, gin.H{
+			"code": 200,
+			"msg":  "验证码已过期",
+			"data": nil,
+		})
+		return
+	}
+
+	if user.Code != storageCode {
+		c.JSON(http.StatusOK, gin.H{
+			"code": 200,
+			"msg":  "验证码错误",
+			"data": nil,
+		})
+	} else {
+		fmt.Println("验证码正确，删除redis中的验证码")
+		config.RDC.Del(config.CTX, user.Tel)
+		c.JSON(http.StatusOK, gin.H{"code": 200,
+			"msg":  "验证码正确",
+			"data": nil,
+		})
+	}
+}

service/sms/preValidate.go

@@ -0,0 +1,24 @@
+package sms
+
+import (
+	"fmt"
+	"health-go/config"
+	"health-go/model"
+)
+
+func PreValidate(userTel string) string {
+
+	var result model.User
+	// 计数查询是否存在对应的手机号
+	query := config.DB.Table("user").Where("tel = ?", userTel).First(&result)
+	if query.Error != nil {
+		return fmt.Sprint(query.Error)
+	}
+	if query.RowsAffected == 0 {
+		return "notExist"
+	}
+	if result.SMSTry > 10 {
+		return "reachLimit"
+	}
+	return "ok"
+}