99 lines
3.4 KiB
Go
99 lines
3.4 KiB
Go
package auth
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"github.com/pkg/errors"
|
||
"math/rand"
|
||
"qnc-server/common/xerr"
|
||
"time"
|
||
|
||
"qnc-server/app/user/cmd/api/internal/svc"
|
||
"qnc-server/app/user/cmd/api/internal/types"
|
||
|
||
openapi "github.com/alibabacloud-go/darabonba-openapi/v2/client"
|
||
dysmsapi "github.com/alibabacloud-go/dysmsapi-20170525/v3/client"
|
||
"github.com/alibabacloud-go/tea-utils/v2/service"
|
||
"github.com/alibabacloud-go/tea/tea"
|
||
"github.com/zeromicro/go-zero/core/logx"
|
||
)
|
||
|
||
type SendSmsLogic struct {
|
||
logx.Logger
|
||
ctx context.Context
|
||
svcCtx *svc.ServiceContext
|
||
}
|
||
|
||
func NewSendSmsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *SendSmsLogic {
|
||
return &SendSmsLogic{
|
||
Logger: logx.WithContext(ctx),
|
||
ctx: ctx,
|
||
svcCtx: svcCtx,
|
||
}
|
||
}
|
||
|
||
func (l *SendSmsLogic) SendSms(req *types.SendSmsReq) error {
|
||
// 检查手机号是否在一分钟内已发送过验证码
|
||
redisKey := fmt.Sprintf("%s:%s", req.ActionType, req.Mobile)
|
||
exists, err := l.svcCtx.Redis.Exists(redisKey)
|
||
if err != nil {
|
||
return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "短信发送, 读取redis缓存失败: %s", req.Mobile)
|
||
}
|
||
|
||
if exists {
|
||
// 如果 Redis 中已经存在标记，说明在 1 分钟内请求过，返回错误
|
||
return errors.Wrapf(xerr.NewErrMsg("一分钟内不能重复发送验证码"), "短信发送, 手机号1分钟内重复请求发送二维码: %s", req.Mobile)
|
||
}
|
||
|
||
code := fmt.Sprintf("%06d", rand.New(rand.NewSource(time.Now().UnixNano())).Intn(1000000))
|
||
|
||
// 发送短信
|
||
smsResp, err := l.sendSmsRequest(req.Mobile, code)
|
||
if err != nil {
|
||
return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "短信发送, 调用阿里客户端失败: %+v", err)
|
||
}
|
||
if *smsResp.Body.Code != "OK" {
|
||
return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "短信发送, 阿里客户端响应失败: %s", *smsResp.Body.Message)
|
||
}
|
||
|
||
// 将验证码保存到 Redis，设置过期时间
|
||
err = l.svcCtx.Redis.Setex(req.Mobile, code, l.svcCtx.Config.VerifyCode.ValidTime) // 验证码有效期5分钟
|
||
if err != nil {
|
||
return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "短信发送, 验证码设置过期时间失败: %+v", err)
|
||
}
|
||
// 在 Redis 中设置 1 分钟的标记，限制重复请求
|
||
err = l.svcCtx.Redis.Setex(redisKey, code, 60) // 标记 1 分钟内不能重复请求
|
||
if err != nil {
|
||
return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "短信发送, 验证码设置限制重复请求失败: %+v", err)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// CreateClient 创建阿里云短信客户端
|
||
func (l *SendSmsLogic) CreateClient() (*dysmsapi.Client, error) {
|
||
config := &openapi.Config{
|
||
AccessKeyId: &l.svcCtx.Config.VerifyCode.AccessKeyID,
|
||
AccessKeySecret: &l.svcCtx.Config.VerifyCode.AccessKeySecret,
|
||
}
|
||
config.Endpoint = tea.String(l.svcCtx.Config.VerifyCode.EndpointURL)
|
||
return dysmsapi.NewClient(config)
|
||
}
|
||
|
||
// sendSmsRequest 发送短信请求
|
||
func (l *SendSmsLogic) sendSmsRequest(mobile, code string) (*dysmsapi.SendSmsResponse, error) {
|
||
// 初始化阿里云短信客户端
|
||
cli, err := l.CreateClient()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
request := &dysmsapi.SendSmsRequest{
|
||
SignName: tea.String(l.svcCtx.Config.VerifyCode.SignName),
|
||
TemplateCode: tea.String(l.svcCtx.Config.VerifyCode.TemplateCode),
|
||
PhoneNumbers: tea.String(mobile),
|
||
TemplateParam: tea.String(fmt.Sprintf("{\"code\":\"%s\"}", code)),
|
||
}
|
||
runtime := &service.RuntimeOptions{}
|
||
return cli.SendSmsWithOptions(request, runtime)
|
||
}
|