From 807c9712efd4ae43052c971e92bda5724a9c3724 Mon Sep 17 00:00:00 2001
From: Lunny Xiao <xiaolunwen@gmail.com>
Date: Mon, 3 Jul 2023 09:53:05 +0800
Subject: Get latest commit statuses from database instead of git data on
 dashboard for repositories (#25605)

related #24638
---
 models/git/branch_list.go | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

(limited to 'models')

diff --git a/models/git/branch_list.go b/models/git/branch_list.go
index da78248c0b..836d4ffc41 100644
--- a/models/git/branch_list.go
+++ b/models/git/branch_list.go
@@ -130,3 +130,20 @@ func FindBranchNames(ctx context.Context, opts FindBranchOptions) ([]string, err
 	}
 	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
+}
-- 
cgit v1.2.3