Use session when creating user (#2638)

This commit is contained in:
Ethan Koenig 2017-10-02 23:29:26 -07:00 committed by Lauris BH
parent b3cfa5a77e
commit 4b2c8ca533
1 changed files with 20 additions and 16 deletions

View File

@ -591,17 +591,21 @@ func (u *User) IsMailable() bool {
return u.IsActive return u.IsActive
} }
func isUserExist(e Engine, uid int64, name string) (bool, error) {
if len(name) == 0 {
return false, nil
}
return e.
Where("id!=?", uid).
Get(&User{LowerName: strings.ToLower(name)})
}
// IsUserExist checks if given user name exist, // IsUserExist checks if given user name exist,
// the user name should be noncased unique. // the user name should be noncased unique.
// If uid is presented, then check will rule out that one, // If uid is presented, then check will rule out that one,
// it is used when update a user name in settings page. // it is used when update a user name in settings page.
func IsUserExist(uid int64, name string) (bool, error) { func IsUserExist(uid int64, name string) (bool, error) {
if len(name) == 0 { return isUserExist(x, uid, name)
return false, nil
}
return x.
Where("id!=?", uid).
Get(&User{LowerName: strings.ToLower(name)})
} }
// GetUserSalt returns a random user salt token. // GetUserSalt returns a random user salt token.
@ -659,7 +663,13 @@ func CreateUser(u *User) (err error) {
return err return err
} }
isExist, err := IsUserExist(0, u.Name) sess := x.NewSession()
defer sess.Close()
if err = sess.Begin(); err != nil {
return err
}
isExist, err := isUserExist(sess, 0, u.Name)
if err != nil { if err != nil {
return err return err
} else if isExist { } else if isExist {
@ -667,16 +677,16 @@ func CreateUser(u *User) (err error) {
} }
u.Email = strings.ToLower(u.Email) u.Email = strings.ToLower(u.Email)
has, err := x. isExist, err = sess.
Where("email=?", u.Email). Where("email=?", u.Email).
Get(new(User)) Get(new(User))
if err != nil { if err != nil {
return err return err
} else if has { } else if isExist {
return ErrEmailAlreadyUsed{u.Email} return ErrEmailAlreadyUsed{u.Email}
} }
isExist, err = IsEmailUsed(u.Email) isExist, err = isEmailUsed(sess, u.Email)
if err != nil { if err != nil {
return err return err
} else if isExist { } else if isExist {
@ -698,12 +708,6 @@ func CreateUser(u *User) (err error) {
u.AllowCreateOrganization = setting.Service.DefaultAllowCreateOrganization u.AllowCreateOrganization = setting.Service.DefaultAllowCreateOrganization
u.MaxRepoCreation = -1 u.MaxRepoCreation = -1
sess := x.NewSession()
defer sess.Close()
if err = sess.Begin(); err != nil {
return err
}
if _, err = sess.Insert(u); err != nil { if _, err = sess.Insert(u); err != nil {
return err return err
} else if err = os.MkdirAll(UserPath(u.Name), os.ModePerm); err != nil { } else if err = os.MkdirAll(UserPath(u.Name), os.ModePerm); err != nil {