zzw/lone-services
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: aliyun sms

Browse Source
This commit is contained in:
zzw
2026-09-03 17:11:34 +08:00
parent 493416551b
commit 73a5fa41ff
17 changed files with 488 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files
services/user/internal/config/sms.go
+19
View File
@@ -0,0 +1,19 @@
package config
import (
"lone-services/pkg/sms"
"lone-services/pkg/utils"
)
func NewSMS() (sms.SMS, error) {
return sms.New(sms.Config{
AccessKeyId: utils.GetConfigString("sms.accessKeyId"),
AccessKeySecret: utils.GetConfigString("sms.accessKeySecret"),
SignName: utils.GetConfigString("sms.signName"),
Endpoint: utils.GetConfigString("sms.endpoint"),
TemplateCode: utils.GetConfigString("sms.template.code"),
TemplatePwd: utils.GetConfigString("sms.template.password"),
Timeout: utils.GetConfigInt("sms.timeout"),
})
}
services/user/internal/dao/const.go
+9 -5
View File
@@ -13,15 +13,19 @@ const (
GrantTypePassword = "password"
GrantTypeSms = "sms"
// Type* 登录身份（写入 session.type / X-User-Type），与订单侧约定一致
TypeSale uint8 = 1 // 销售
TypeStore uint8 = 2 // 门店
TypeUser uint8 = 3 // 普通用户
SmsSceneLogin int32 = 1
SmsSceneResetPassword int32 = 2
SmsRateLimitSeconds = 60
SmsCodeLength = 6
TypeSale uint8 = 1
TypeStore uint8 = 2
TypeUser uint8 = 3
SubjectTypeSale = "sale"
SubjectTypeStore = "store"
SubjectTypeUser = "user"
// ExtraJsonEmpty MySQL JSON 列不能写空串，无扩展字段时用 {}
ExtraJsonEmpty = "{}"
)
services/user/internal/logic/sendSmsLogic.go
+120
View File
@@ -0,0 +1,120 @@
package logic
import (
"context"
"crypto/rand"
"fmt"
"math/big"
"strconv"
"time"
"lone-services/pkg/redis"
"lone-services/pkg/sms"
"lone-services/pkg/utils"
"lone-services/pkg/validate"
user "lone-services/rpc/user/pb"
"lone-services/services/user/internal/dao"
"lone-services/services/user/internal/svc"
"lone-services/services/user/validator"
"github.com/zeromicro/go-zero/core/logx"
)
type SendSmsLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewSendSmsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *SendSmsLogic {
return &SendSmsLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
func (l *SendSmsLogic) SendSms(in *user.SendSmsReq) (*user.Response, error) {
var req validator.SendSmsValidator
if msg := validate.ValidateFromProto(in, &req); msg != utils.StringEmpty {
return outResponse(utils.ErrorParams, msg), nil
}
if l.svcCtx.SMS == nil {
return outResponse(utils.Fail, "短信服务未配置"), nil
}
kind, ok := smsSceneKind(req.Scene)
if !ok {
return outResponse(utils.ErrorParams, "短信场景不正确"), nil
}
freqKey := smsFreqRedisKey(req.Scene, req.Mobile)
okSet, err := redis.Client.SetNX(l.ctx, freqKey, "1", time.Duration(dao.SmsRateLimitSeconds)*time.Second).Result()
if err != nil {
l.Errorf("sms freq setnx: %v", err)
return failResponse(utils.Fail), nil
}
if !okSet {
return outResponse(utils.ErrorParams, "发送过于频繁，请稍后再试"), nil
}
code, err := genDigitCode(dao.SmsCodeLength)
if err != nil {
_ = redis.Client.Del(l.ctx, freqKey).Err()
l.Errorf("sms gen code: %v", err)
return failResponse(utils.Fail), nil
}
if err := l.svcCtx.SMS.SendByKind(req.Mobile, kind, map[string]string{"code": code}); err != nil {
_ = redis.Client.Del(l.ctx, freqKey).Err()
l.Errorf("sms send: %v", err)
return outResponse(utils.Fail, "短信发送失败"), nil
}
ttl := l.svcCtx.SMS.Timeout()
if ttl <= utils.NumberZero {
ttl = 600
}
codeKey := smsCodeRedisKey(req.Scene, req.Mobile)
if err := redis.Client.Set(l.ctx, codeKey, code, time.Duration(ttl)*time.Second).Err(); err != nil {
l.Errorf("sms cache code: %v", err)
return failResponse(utils.Fail), nil
}
return okResponse(nil), nil
}
func smsSceneKind(scene int32) (sms.TemplateKind, bool) {
switch scene {
case dao.SmsSceneLogin:
return sms.TemplateCode, true
case dao.SmsSceneResetPassword:
return sms.TemplatePassword, true
default:
return "", false
}
}
func smsCodeRedisKey(scene int32, mobile string) string {
return utils.CodeKey + strconv.Itoa(int(scene)) + ":" + mobile
}
func smsFreqRedisKey(scene int32, mobile string) string {
return "sms:freq:" + strconv.Itoa(int(scene)) + ":" + mobile
}
func genDigitCode(length int) (string, error) {
if length <= utils.NumberZero {
return utils.StringEmpty, fmt.Errorf("invalid code length")
}
out := make([]byte, length)
for i := range out {
n, err := rand.Int(rand.Reader, big.NewInt(10))
if err != nil {
return utils.StringEmpty, err
}
out[i] = byte('0' + n.Int64())
}
return string(out), nil
}
services/user/internal/server/userserver.go
+5
View File
@@ -23,6 +23,11 @@ func NewUserServer(svcCtx *svc.ServiceContext) *UserServer {
}
}
func (s *UserServer) SendSms(ctx context.Context, in *user.SendSmsReq) (*user.Response, error) {
l := logic.NewSendSmsLogic(ctx, s.svcCtx)
return l.SendSms(in)
}
func (s *UserServer) Login(ctx context.Context, in *user.LoginReq) (*user.Response, error) {
l := logic.NewLoginLogic(ctx, s.svcCtx)
return l.Login(in)
services/user/internal/svc/servicecontext.go
+9
View File
@@ -1,6 +1,7 @@
package svc
import (
"lone-services/pkg/sms"
"lone-services/pkg/utils"
"lone-services/services/user/internal/config"
@@ -14,6 +15,7 @@ type ServiceContext struct {
Prefix string
JWT *config.JWTAuth
SaleSvcName string
SMS sms.SMS
}
func NewServiceContext(c config.Config, db *gorm.DB, jwtAuth *config.JWTAuth) *ServiceContext {
@@ -21,11 +23,18 @@ func NewServiceContext(c config.Config, db *gorm.DB, jwtAuth *config.JWTAuth) *S
if saleSvc == utils.StringEmpty {
logx.Error("config services.sale empty")
}
smsClient, err := config.NewSMS()
if err != nil {
logx.Errorf("sms init: %v", err)
}
return &ServiceContext{
Config: c,
DB: db,
Prefix: utils.GetConfigString("mysql.prefix"),
JWT: jwtAuth,
SaleSvcName: saleSvc,
SMS: smsClient,
}
}
services/user/userClient/user.go
+7
View File
@@ -21,6 +21,7 @@ type (
Response = user.Response
RevokeBizClientData = user.RevokeBizClientData
RevokeBizClientReq = user.RevokeBizClientReq
SendSmsReq = user.SendSmsReq
UpdateMobileData = user.UpdateMobileData
UpdateMobileReq = user.UpdateMobileReq
UserBriefItem = user.UserBriefItem
@@ -30,6 +31,7 @@ type (
UsersByIdsReq = user.UsersByIdsReq
User interface {
SendSms(ctx context.Context, in *SendSmsReq, opts ...grpc.CallOption) (*Response, error)
Login(ctx context.Context, in *LoginReq, opts ...grpc.CallOption) (*Response, error)
Info(ctx context.Context, in *InfoReq, opts ...grpc.CallOption) (*Response, error)
UserItems(ctx context.Context, in *UserItemsReq, opts ...grpc.CallOption) (*Response, error)
@@ -51,6 +53,11 @@ func NewUser(cli zrpc.Client) User {
}
}
func (m *defaultUser) SendSms(ctx context.Context, in *SendSmsReq, opts ...grpc.CallOption) (*Response, error) {
client := user.NewUserClient(m.cli.Conn())
return client.SendSms(ctx, in, opts...)
}
func (m *defaultUser) Login(ctx context.Context, in *LoginReq, opts ...grpc.CallOption) (*Response, error) {
client := user.NewUserClient(m.cli.Conn())
return client.Login(ctx, in, opts...)
services/user/validator/user.go
+14
View File
@@ -11,6 +11,20 @@ type LoginValidator struct {
SmsCode string `validate:"required_if=GrantType sms,omitempty,max=8"`
}
type SendSmsValidator struct {
Mobile string `validate:"required,len=11"`
Scene int32 `validate:"required,oneof=1 2"`
}
func (p SendSmsValidator) GetMessage() validate.ValidatorMessages {
return validate.ValidatorMessages{
"Mobile.required": "手机号不能为空",
"Mobile.len": "手机号格式错误",
"Scene.required": "短信场景不能为空",
"Scene.oneof": "短信场景不正确",
}
}
func (p LoginValidator) GetMessage() validate.ValidatorMessages {
return validate.ValidatorMessages{
"ClientCode.required": "端编码不能为空",