fix version 1
This commit is contained in:
@@ -16,9 +16,13 @@ import (
|
||||
"git.apinb.com/bsm-sdk/core/errcode"
|
||||
"git.apinb.com/bsm-sdk/core/printer"
|
||||
"git.apinb.com/bsm-sdk/core/vars"
|
||||
"github.com/redis/go-redis/v9"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
// 短信验证码在 Redis 中的 key 前缀,与 sender 模块保持一致:前缀 + 手机号
|
||||
const smsCodeKeyPrefix = "/SMS/Code/"
|
||||
|
||||
// Login 员工登录验证
|
||||
func Login(ctx context.Context, in *pb.LoginRequest) (reply *pb.LoginReply, err error) {
|
||||
var record models.MallStaff
|
||||
@@ -47,7 +51,27 @@ func Login(ctx context.Context, in *pb.LoginRequest) (reply *pb.LoginReply, err
|
||||
return nil, errcode.ErrInvalidArgument
|
||||
}
|
||||
|
||||
err := impl.DBService.Where("phone = ?", in.Phone).First(&record).Error
|
||||
// 校验短信验证码,key 规则与 sender 模块一致:/SMS/Code/ + 手机号
|
||||
smsKey := smsCodeKeyPrefix + in.Phone
|
||||
storedCode, codeErr := impl.RedisService.Client.Get(impl.RedisService.Ctx, smsKey).Result()
|
||||
if codeErr != nil {
|
||||
if errors.Is(codeErr, redis.Nil) {
|
||||
printer.Error("验证码已过期或不存在: phone=%s", in.Phone)
|
||||
return nil, errcode.ErrInvalidArgument
|
||||
}
|
||||
printer.Error(codeErr.Error())
|
||||
return nil, errcode.ErrInternal
|
||||
}
|
||||
if storedCode != in.VerifyCode {
|
||||
printer.Error("验证码不正确: phone=%s", in.Phone)
|
||||
return nil, errcode.ErrInvalidArgument
|
||||
}
|
||||
// 校验成功后立即删除,避免同一验证码被重放
|
||||
if err := impl.RedisService.Client.Del(impl.RedisService.Ctx, smsKey).Err(); err != nil {
|
||||
printer.Error("清除验证码缓存异常: %v", err)
|
||||
}
|
||||
|
||||
err = impl.DBService.Where("phone = ?", in.Phone).First(&record).Error
|
||||
if err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil, errcode.ErrAccountNotFound
|
||||
@@ -55,7 +79,6 @@ func Login(ctx context.Context, in *pb.LoginRequest) (reply *pb.LoginReply, err
|
||||
printer.Error(err.Error())
|
||||
return nil, errcode.ErrDB
|
||||
}
|
||||
// TODO: 添加验证码校验逻辑
|
||||
|
||||
default:
|
||||
return nil, errcode.ErrUnimplemented
|
||||
|
||||
Reference in New Issue
Block a user