diff --git a/app/enterprise/api/account.go b/app/enterprise/api/account.go index a8e9da5..fe12662 100644 --- a/app/enterprise/api/account.go +++ b/app/enterprise/api/account.go @@ -22,6 +22,7 @@ type ( Captcha string `json:"captcha" form:"captcha" binding:"required"` Password string `json:"password" form:"password" binding:"required"` RepeatPass string `json:"repeat_pass" form:"repeat_pass" binding:"required"` + Identity int `json:"identity" form:"identity" binding:"required"` } ) @@ -54,7 +55,7 @@ func (a *Account) Register(c *gin.Context) { } data, err := account.NewRegister()().Launch(&account.RegisterParams{ Name: form.Name, Mobile: form.Mobile, Captcha: form.Captcha, - Password: form.Password, RepeatPass: form.RepeatPass, + Password: form.Password, RepeatPass: form.RepeatPass, Identity: form.Identity, }) api.APIResponse(err, data) } diff --git a/app/enterprise/controller/account/register.go b/app/enterprise/controller/account/register.go index c2bc9ff..5c3faea 100644 --- a/app/enterprise/controller/account/register.go +++ b/app/enterprise/controller/account/register.go @@ -17,6 +17,7 @@ type RegisterHandle func() *Register type ( RegisterParams struct { Name, Mobile, Captcha, Password, RepeatPass string + Identity int } ) @@ -57,6 +58,7 @@ func (c *Register) Launch(params *RegisterParams) (*InstanceLoginReturn, error) mUserInstance.Password = utils.GetRandomString(12) mUserInstance.Mobile = params.Mobile mUserInstance.Password = params.Password + mUserInstance.Identity = params.Identity mUserTenant := model.NewUserTenant() @@ -66,6 +68,7 @@ func (c *Register) Launch(params *RegisterParams) (*InstanceLoginReturn, error) } mUserTenant.UID = mUserInstance.UUID mUserTenant.Name = params.Name + mUserTenant.Identity = params.Identity mUserTenant.Selected = model2.UserTenantSelectedForYes return model2.Create(mUserTenant.UserTenant, tx) }); err != nil { @@ -73,7 +76,7 @@ func (c *Register) Launch(params *RegisterParams) (*InstanceLoginReturn, error) } return NewInstance()().Login()(&InstanceLoginParams{ UID: mUserTenant.UUID, Name: mUserTenant.Name, Mobile: mUserInstance.Mobile, - Identity: mUserInstance.Identity, + Identity: mUserInstance.Identity, SelectIdentity: params.Identity, }), err }