Avoid duplicate queries in auth (#827)

Avoid identical making calls to GetUserByID(..) in SignedInUser(..)
pull/848/head
Ethan Koenig 8 years ago committed by Lunny Xiao
parent bf647ce143
commit e86d935175
  1. 105
      modules/auth/auth.go

@ -69,14 +69,7 @@ func SignedInID(ctx *macaron.Context, sess session.Store) int64 {
uid := sess.Get("uid") uid := sess.Get("uid")
if uid == nil { if uid == nil {
return 0 return 0
} } else if id, ok := uid.(int64); ok {
if id, ok := uid.(int64); ok {
if _, err := models.GetUserByID(id); err != nil {
if !models.IsErrUserNotExist(err) {
log.Error(4, "GetUserById: %v", err)
}
return 0
}
return id return id
} }
return 0 return 0
@ -89,66 +82,64 @@ func SignedInUser(ctx *macaron.Context, sess session.Store) (*models.User, bool)
return nil, false return nil, false
} }
uid := SignedInID(ctx, sess) if uid := SignedInID(ctx, sess); uid > 0 {
user, err := models.GetUserByID(uid)
if err == nil {
return user, false
} else if !models.IsErrUserNotExist(err) {
log.Error(4, "GetUserById: %v", err)
}
}
if uid <= 0 { if setting.Service.EnableReverseProxyAuth {
if setting.Service.EnableReverseProxyAuth { webAuthUser := ctx.Req.Header.Get(setting.ReverseProxyAuthUser)
webAuthUser := ctx.Req.Header.Get(setting.ReverseProxyAuthUser) if len(webAuthUser) > 0 {
if len(webAuthUser) > 0 { u, err := models.GetUserByName(webAuthUser)
u, err := models.GetUserByName(webAuthUser) if err != nil {
if err != nil { if !models.IsErrUserNotExist(err) {
if !models.IsErrUserNotExist(err) { log.Error(4, "GetUserByName: %v", err)
log.Error(4, "GetUserByName: %v", err) return nil, false
return nil, false }
}
// Check if enabled auto-registration. // Check if enabled auto-registration.
if setting.Service.EnableReverseProxyAutoRegister { if setting.Service.EnableReverseProxyAutoRegister {
u := &models.User{ u := &models.User{
Name: webAuthUser, Name: webAuthUser,
Email: gouuid.NewV4().String() + "@localhost", Email: gouuid.NewV4().String() + "@localhost",
Passwd: webAuthUser, Passwd: webAuthUser,
IsActive: true, IsActive: true,
} }
if err = models.CreateUser(u); err != nil { if err = models.CreateUser(u); err != nil {
// FIXME: should I create a system notice? // FIXME: should I create a system notice?
log.Error(4, "CreateUser: %v", err) log.Error(4, "CreateUser: %v", err)
return nil, false return nil, false
}
return u, false
} }
return u, false
} }
return u, false
} }
return u, false
} }
}
// Check with basic auth. // Check with basic auth.
baHead := ctx.Req.Header.Get("Authorization") baHead := ctx.Req.Header.Get("Authorization")
if len(baHead) > 0 { if len(baHead) > 0 {
auths := strings.Fields(baHead) auths := strings.Fields(baHead)
if len(auths) == 2 && auths[0] == "Basic" { if len(auths) == 2 && auths[0] == "Basic" {
uname, passwd, _ := base.BasicAuthDecode(auths[1]) uname, passwd, _ := base.BasicAuthDecode(auths[1])
u, err := models.UserSignIn(uname, passwd)
if err != nil {
if !models.IsErrUserNotExist(err) {
log.Error(4, "UserSignIn: %v", err)
}
return nil, false
}
return u, true u, err := models.UserSignIn(uname, passwd)
if err != nil {
if !models.IsErrUserNotExist(err) {
log.Error(4, "UserSignIn: %v", err)
}
return nil, false
} }
}
return nil, false
}
u, err := models.GetUserByID(uid) return u, true
if err != nil { }
log.Error(4, "GetUserById: %v", err)
return nil, false
} }
return u, false return nil, false
} }
// Form form binding interface // Form form binding interface

Loading…
Cancel
Save