magic_rb/forgejo
Template
1
0
Fork 0
mirror of https://codeberg.org/forgejo/forgejo synced 2024-11-28 12:46:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

Optimization for user.GetRepositoryAccesses to reduce db query times (#495)

Browse source 
* optimization for user.GetRepositoryAccesses to reduce db query times

* fix missing cache
This commit is contained in:
Lunny Xiao 2016-12-28 09:34:35 +08:00 committed by GitHub
parent 9fae9f0dc2
commit c463b1b8cb
1 changed files with 29 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

47
models/access.go
View file

@ -4,11 +4,7 @@
package models package models
import ( import "fmt"
"fmt"
"code.gitea.io/gitea/modules/log"
)
// AccessMode specifies the users access mode // AccessMode specifies the users access mode
type AccessMode int type AccessMode int
@ -103,26 +99,39 @@ func HasAccess(user *User, repo *Repository, testMode AccessMode) (bool, error)
// GetRepositoryAccesses finds all repositories with their access mode where a user has access but does not own. // GetRepositoryAccesses finds all repositories with their access mode where a user has access but does not own.
func (user *User) GetRepositoryAccesses() (map[*Repository]AccessMode, error) { func (user *User) GetRepositoryAccesses() (map[*Repository]AccessMode, error) {
accesses := make([]*Access, 0, 10) accesses := make([]*Access, 0, 10)
if err := x.Find(&accesses, &Access{UserID: user.ID}); err != nil { type RepoAccess struct {
Access `xorm:"extends"`
Repository `xorm:"extends"`
}
rows, err := x.
Join("INNER", "repository", "respository.id = access.repo_id").
Where("access.user_id = ?", user.ID).
And("repository.owner_id <> ?", user.ID).
Rows(new(RepoAccess))
if err != nil {
return nil, err
}
defer rows.Close()
var repos = make(map[*Repository]AccessMode, len(accesses))
var ownerCache = make(map[int64]*User, len(accesses))
for rows.Next() {
var repo RepoAccess
err = rows.Scan(&repo)
if err != nil {
return nil, err return nil, err
} }
repos := make(map[*Repository]AccessMode, len(accesses)) var ok bool
for _, access := range accesses { if repo.Owner, ok = ownerCache[repo.OwnerID]; !ok {
repo, err := GetRepositoryByID(access.RepoID)
if err != nil {
if IsErrRepoNotExist(err) {
log.Error(4, "GetRepositoryByID: %v", err)
continue
}
return nil, err
}
if err = repo.GetOwner(); err != nil { if err = repo.GetOwner(); err != nil {
return nil, err return nil, err
} else if repo.OwnerID == user.ID {
continue
} }
repos[repo] = access.Mode ownerCache[repo.OwnerID] = repo.Owner
}
repos[&repo.Repository] = repo.Access.Mode
} }
return repos, nil return repos, nil
} }