Fix performance regression when user has many organization (#18125)

This commit is contained in:
Lunny Xiao 2021-12-30 13:03:40 +08:00 committed by GitHub
parent e0b41f8c37
commit 626b10794f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 12 deletions

View File

@ -294,13 +294,27 @@ func CanUserForkRepo(user *user_model.User, repo *repo_model.Repository) (bool,
return false, nil return false, nil
} }
// FindUserOrgForks returns the forked repositories for one user from a repository
func FindUserOrgForks(repoID, userID int64) ([]*repo_model.Repository, error) {
var cond builder.Cond = builder.And(
builder.Eq{"fork_id": repoID},
builder.In("owner_id",
builder.Select("org_id").
From("org_user").
Where(builder.Eq{"uid": userID}),
),
)
var repos []*repo_model.Repository
return repos, db.GetEngine(db.DefaultContext).Table("repository").Where(cond).Find(&repos)
}
// GetForksByUserAndOrgs return forked repos of the user and owned orgs // GetForksByUserAndOrgs return forked repos of the user and owned orgs
func GetForksByUserAndOrgs(user *user_model.User, repo *repo_model.Repository) ([]*repo_model.Repository, error) { func GetForksByUserAndOrgs(user *user_model.User, repo *repo_model.Repository) ([]*repo_model.Repository, error) {
var repoList []*repo_model.Repository var repoList []*repo_model.Repository
if user == nil { if user == nil {
return repoList, nil return repoList, nil
} }
var forkedRepo *repo_model.Repository
forkedRepo, err := repo_model.GetUserFork(repo.ID, user.ID) forkedRepo, err := repo_model.GetUserFork(repo.ID, user.ID)
if err != nil { if err != nil {
return repoList, err return repoList, err
@ -308,19 +322,11 @@ func GetForksByUserAndOrgs(user *user_model.User, repo *repo_model.Repository) (
if forkedRepo != nil { if forkedRepo != nil {
repoList = append(repoList, forkedRepo) repoList = append(repoList, forkedRepo)
} }
canCreateRepos, err := GetOrgsCanCreateRepoByUserID(user.ID) orgForks, err := FindUserOrgForks(repo.ID, user.ID)
if err != nil { if err != nil {
return repoList, err return nil, err
}
for _, org := range canCreateRepos {
forkedRepo, err := repo_model.GetUserFork(repo.ID, org.ID)
if err != nil {
return repoList, err
}
if forkedRepo != nil {
repoList = append(repoList, forkedRepo)
}
} }
repoList = append(repoList, orgForks...)
return repoList, nil return repoList, nil
} }