}
return branches, nil
}
+
+func FindBranchesByRepoAndBranchName(ctx context.Context, repoBranches map[int64]string) (map[int64]string, error) {
+ cond := builder.NewCond()
+ for repoID, branchName := range repoBranches {
+ cond = cond.Or(builder.And(builder.Eq{"repo_id": repoID}, builder.Eq{"name": branchName}))
+ }
+ var branches []*Branch
+ if err := db.GetEngine(ctx).
+ Where(cond).Find(&branches); err != nil {
+ return nil, err
+ }
+ branchMap := make(map[int64]string, len(branches))
+ for _, branch := range branches {
+ branchMap[branch.RepoID] = branch.CommitID
+ }
+ return branchMap, nil
+}
// collect the latest commit of each repo
// at most there are dozens of repos (limited by MaxResponseItems), so it's not a big problem at the moment
- repoIDsToLatestCommitSHAs := make(map[int64]string, len(repos))
+ repoBranchNames := make(map[int64]string, len(repos))
for _, repo := range repos {
- commitID, err := repo_service.GetBranchCommitID(ctx, repo, repo.DefaultBranch)
- if err != nil {
- continue
- }
- repoIDsToLatestCommitSHAs[repo.ID] = commitID
+ repoBranchNames[repo.ID] = repo.DefaultBranch
+ }
+
+ repoIDsToLatestCommitSHAs, err := git_model.FindBranchesByRepoAndBranchName(ctx, repoBranchNames)
+ if err != nil {
+ log.Error("FindBranchesByRepoAndBranchName: %v", err)
+ return
}
// call the database O(1) times to get the commit statuses for all repos