feat:完善项目

This commit is contained in:
henry
2022-01-15 16:48:49 +08:00
parent 57daf5119d
commit 6d075dab4f
33 changed files with 196 additions and 154 deletions

View File

@ -11,10 +11,10 @@ import (
"gorm.io/gorm"
)
type Login struct{ local string }
type Login struct{ tenantID uint64 }
type (
LoginHandle func(local string) *Login
LoginHandle func(tenantID uint64) *Login
)
type (
@ -41,12 +41,12 @@ const (
LoginModeForQQ // QQ登陆
)
var loginHandle = map[LoginMode]func(*LoginParams, string) (*InstanceLoginParams, error){
var loginHandle = map[LoginMode]func(*LoginParams, uint64) (*InstanceLoginParams, error){
LoginModeForSmsCaptcha: loginForSmsCaptcha, LoginModeForPassword: loginForPassword,
}
// loginForSmsCaptcha 短信验证码登陆
func loginForSmsCaptcha(params *LoginParams, local string) (*InstanceLoginParams, error) {
func loginForSmsCaptcha(params *LoginParams, tenantID uint64) (*InstanceLoginParams, error) {
if !utils.ValidateMobile(params.Captcha.Mobile) {
return nil, errors.New("操作错误,手机号码格式异常")
}
@ -107,7 +107,7 @@ RETURNS:
}
// loginForPassword 密码登陆
func loginForPassword(params *LoginParams, local string) (*InstanceLoginParams, error) {
func loginForPassword(params *LoginParams, tenantID uint64) (*InstanceLoginParams, error) {
if !utils.ValidateMobile(params.Password.Account) {
return nil, errors.New("操作错误,手机号码格式异常")
}
@ -156,7 +156,7 @@ func (c *Login) Launch(mode LoginMode, params *LoginParams) (*InstanceLoginRetur
if !has {
return nil, errors.New("操作错误,未知的登陆模式")
}
ret, err := _handle(params, c.local)
ret, err := _handle(params, c.tenantID)
if err != nil {
return nil, err
@ -168,7 +168,7 @@ func (c *Login) Launch(mode LoginMode, params *LoginParams) (*InstanceLoginRetur
}
func NewLogin() LoginHandle {
return func(local string) *Login {
return &Login{local: local}
return func(tenantID uint64) *Login {
return &Login{tenantID: tenantID}
}
}