Skip to content

Commit

Permalink
Use session when creating user (#2638)
Browse files Browse the repository at this point in the history
  • Loading branch information
ethantkoenig authored and lafriks committed Oct 3, 2017
1 parent b3cfa5a commit 4b2c8ca
Showing 1 changed file with 20 additions and 16 deletions.
36 changes: 20 additions & 16 deletions models/user.go
Original file line number Diff line number Diff line change
Expand Up @@ -591,19 +591,23 @@ func (u *User) IsMailable() bool {
return u.IsActive
}

// IsUserExist checks if given user name exist,
// the user name should be noncased unique.
// If uid is presented, then check will rule out that one,
// it is used when update a user name in settings page.
func IsUserExist(uid int64, name string) (bool, error) {
func isUserExist(e Engine, uid int64, name string) (bool, error) {
if len(name) == 0 {
return false, nil
}
return x.
return e.
Where("id!=?", uid).
Get(&User{LowerName: strings.ToLower(name)})
}

// IsUserExist checks if given user name exist,
// the user name should be noncased unique.
// If uid is presented, then check will rule out that one,
// it is used when update a user name in settings page.
func IsUserExist(uid int64, name string) (bool, error) {
return isUserExist(x, uid, name)
}

// GetUserSalt returns a random user salt token.
func GetUserSalt() (string, error) {
return base.GetRandomString(10)
Expand Down Expand Up @@ -659,24 +663,30 @@ func CreateUser(u *User) (err error) {
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 {
return err
} else if isExist {
return ErrUserAlreadyExist{u.Name}
}

u.Email = strings.ToLower(u.Email)
has, err := x.
isExist, err = sess.
Where("email=?", u.Email).
Get(new(User))
if err != nil {
return err
} else if has {
} else if isExist {
return ErrEmailAlreadyUsed{u.Email}
}

isExist, err = IsEmailUsed(u.Email)
isExist, err = isEmailUsed(sess, u.Email)
if err != nil {
return err
} else if isExist {
Expand All @@ -698,12 +708,6 @@ func CreateUser(u *User) (err error) {
u.AllowCreateOrganization = setting.Service.DefaultAllowCreateOrganization
u.MaxRepoCreation = -1

sess := x.NewSession()
defer sess.Close()
if err = sess.Begin(); err != nil {
return err
}

if _, err = sess.Insert(u); err != nil {
return err
} else if err = os.MkdirAll(UserPath(u.Name), os.ModePerm); err != nil {
Expand Down

0 comments on commit 4b2c8ca

Please sign in to comment.