Files
lone-services/services/user/internal/logic/registerByUserLogic.go
T
2026-08-28 10:49:09 +08:00

103 lines
2.9 KiB
Go

package logic
import (
"context"
"lone-services/pkg/modelbase"
"lone-services/pkg/redis"
"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/model"
"lone-services/services/user/internal/svc"
"lone-services/services/user/validator"
"strconv"
"github.com/zeromicro/go-zero/core/logx"
)
type RegisterByUserLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewRegisterByUserLogic(ctx context.Context, svcCtx *svc.ServiceContext) *RegisterByUserLogic {
return &RegisterByUserLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
func (l *RegisterByUserLogic) RegisterByUser(in *user.RegisterByUserReq) (*user.Response, error) {
var v validator.RegisterByUserValidator
if msg := validate.ValidateFromProto(in, &v); msg != utils.StringEmpty {
return outResponse(utils.ErrorParams, msg), nil
}
codePrefix := utils.StringEmpty
if v.Type == utils.NumberTwo {
codePrefix = "1:"
}
test := utils.GetConfigBool("sms.sms_test")
isTest := test && v.Code == utils.GetConfigString("sms.sms_test_code")
codeKey := utils.CodeKey + codePrefix + v.Account
code, err := redis.Client.Get(l.ctx, codeKey).Result()
if (err != nil || code != v.Code) && !isTest {
return outResponse(utils.ErrorMissingParams, "验证码错误"), nil
}
var old dao.UserExist
err = model.UserModel{}.Init().GetOne(modelbase.Params{
Eq: map[string]string{
"account": v.Account,
"type": strconv.FormatUint(uint64(v.Type), utils.NumberTen),
},
}, &old)
if err != nil {
l.Errorf("registerByUser check account: %v", err)
return failResponse(utils.Fail), nil
}
if old.Id > utils.NumberZero {
return failResponse(utils.ErrorDataIsExist), nil
}
add := dao.UserCreateByPwd{
Type: uint8(v.Type),
Account: v.Account,
Salt: utils.GetRandString(utils.NumberFive),
AppId: v.AppId,
Status: utils.NumberOne,
IsRegistered: utils.NumberOne,
IsFaceVerified: utils.NumberTwo,
IsProfileCompleted: utils.NumberTwo,
OnTrialNum: utils.NumberOne,
}
pwd := utils.GetSaltPassword(add.Salt, v.Pwd)
add.Password, err = utils.EncryptPassword(pwd)
if err != nil {
l.Errorf("registerByUser encrypt password: %v", err)
return failResponse(utils.Fail), nil
}
if v.Type == utils.NumberOne {
encryptMobile, cErr := utils.EncryptPhone(v.Account)
if cErr != nil {
l.Errorf("registerByUser encrypt mobile: %v", cErr)
return failResponse(utils.ErrorEncryptAesError), nil
}
add.Mobile = encryptMobile
}
err = model.UserModel{}.Init().CreateByPwd(&add)
if err != nil || add.Id < utils.NumberOne {
l.Errorf("registerByUser create: %v", err)
return failResponse(utils.Fail), nil
}
redis.Client.Del(l.ctx, codeKey)
return okResponse(map[string]any{"id": add.Id}), nil
}