feat:完善信息
This commit is contained in:
@ -4,7 +4,10 @@ import (
|
||||
model2 "SciencesServer/app/common/model"
|
||||
"SciencesServer/app/enterprise/model"
|
||||
"SciencesServer/app/handle"
|
||||
"SciencesServer/serve/orm"
|
||||
"SciencesServer/utils"
|
||||
"errors"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type Register struct{}
|
||||
@ -12,53 +15,66 @@ type Register struct{}
|
||||
type RegisterHandle func() *Register
|
||||
|
||||
type (
|
||||
RegisterRequest struct {
|
||||
RegisterParams struct {
|
||||
Name, Mobile, Captcha, Password, RepeatPass string
|
||||
}
|
||||
)
|
||||
|
||||
func (c *RegisterRequest) checkPassword() bool {
|
||||
func (c *RegisterParams) checkPassword() bool {
|
||||
return c.Password == c.RepeatPass
|
||||
}
|
||||
|
||||
func (c *RegisterRequest) checkUserExist(mTenantUser *model.TenantUser) (bool, error) {
|
||||
func (c *RegisterParams) checkUserExist(mUserInstance *model2.UserInstance) (bool, error) {
|
||||
var count int64
|
||||
|
||||
if err := model2.Count(mTenantUser.TenantUser, &count, model2.NewWhere("mobile", c.Mobile)); err != nil {
|
||||
if err := model2.Count(mUserInstance, &count, model2.NewWhere("mobile", c.Mobile)); err != nil {
|
||||
return false, err
|
||||
}
|
||||
return count > 0, nil
|
||||
}
|
||||
func (c *RegisterRequest) checkCaptcha() (bool, error) {
|
||||
func (c *RegisterParams) checkCaptcha() (bool, error) {
|
||||
return handle.NewCaptcha().Validate(&handle.CaptchaSms{Captcha: c.Captcha, Mobile: c.Mobile})
|
||||
}
|
||||
|
||||
func (c *Register) Launch(req *RegisterRequest) (*LoginResponse, error) {
|
||||
if req.checkPassword() {
|
||||
func (c *Register) Launch(params *RegisterParams) (*InstanceLoginReturn, error) {
|
||||
if params.checkPassword() {
|
||||
return nil, errors.New("两次密码不一致")
|
||||
}
|
||||
mTenantUser := model.NewTenantUser()
|
||||
mUserInstance := model.NewUserInstance()
|
||||
|
||||
pass, err := req.checkUserExist(mTenantUser)
|
||||
pass, err := params.checkUserExist(mUserInstance.UserInstance)
|
||||
|
||||
if err != nil {
|
||||
return nil, err
|
||||
} else if pass {
|
||||
return nil, errors.New("当前手机号码已注册")
|
||||
}
|
||||
if pass, err = req.checkCaptcha(); err != nil {
|
||||
if pass, err = params.checkCaptcha(); err != nil {
|
||||
return nil, err
|
||||
} else if !pass {
|
||||
return nil, errors.New("验证码错误或已过期")
|
||||
}
|
||||
mTenantUser.Name = req.Name
|
||||
mTenantUser.Mobile = req.Mobile
|
||||
mTenantUser.Password = req.Password
|
||||
mUserInstance.Password = utils.GetRandomString(12)
|
||||
mUserInstance.Mobile = params.Mobile
|
||||
mUserInstance.Password = params.Password
|
||||
|
||||
if err = model2.Create(mTenantUser.TenantUser); err != nil {
|
||||
mUserTenant := model.NewUserTenant()
|
||||
|
||||
if err = orm.GetDB().Transaction(func(tx *gorm.DB) error {
|
||||
if err = model2.Create(mUserInstance.UserInstance, tx); err != nil {
|
||||
return err
|
||||
}
|
||||
mUserTenant.UID = mUserInstance.UUID
|
||||
mUserTenant.Name = params.Name
|
||||
mUserTenant.Selected = model2.UserTenantSelectedForYes
|
||||
return model2.Create(mUserTenant.UserTenant, tx)
|
||||
}); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return NewAccount().Login()(mTenantUser), err
|
||||
return NewInstance()().Login()(&InstanceLoginParams{
|
||||
UID: mUserTenant.UUID, Name: mUserTenant.Name, Mobile: mUserInstance.Mobile,
|
||||
Identity: mUserInstance.Identity,
|
||||
}), err
|
||||
}
|
||||
|
||||
func NewRegister() RegisterHandle {
|
||||
|
Reference in New Issue
Block a user