@@ -12,6 +12,7 @@ import (
1212 repo_model "code.gitea.io/gitea/models/repo"
1313 "code.gitea.io/gitea/models/unit"
1414 user_model "code.gitea.io/gitea/models/user"
15+ "code.gitea.io/gitea/modules/container"
1516 "code.gitea.io/gitea/modules/git"
1617 "code.gitea.io/gitea/modules/log"
1718 "code.gitea.io/gitea/modules/optional"
@@ -169,16 +170,22 @@ func GetBranch(ctx context.Context, repoID int64, branchName string) (*Branch, e
169170 return & branch , nil
170171}
171172
172- func GetBranches (ctx context.Context , repoID int64 , branchNames []string ) ([]* Branch , error ) {
173+ func GetBranches (ctx context.Context , repoID int64 , branchNames []string , includeDeleted bool ) ([]* Branch , error ) {
173174 branches := make ([]* Branch , 0 , len (branchNames ))
174- return branches , db .GetEngine (ctx ).Where ("repo_id=?" , repoID ).In ("name" , branchNames ).Find (& branches )
175+
176+ sess := db .GetEngine (ctx ).Where ("repo_id=?" , repoID ).In ("name" , branchNames )
177+ if ! includeDeleted {
178+ sess .And ("is_deleted=?" , false )
179+ }
180+ return branches , sess .Find (& branches )
175181}
176182
177- func GetExistBranches (ctx context.Context , repoID int64 , branchNames []string ) ([]* Branch , error ) {
178- branches := make ([]* Branch , 0 , len (branchNames ))
179- return branches , db .GetEngine (ctx ).Where ("repo_id=?" , repoID ).
180- And ("is_deleted=?" , false ).
181- In ("name" , branchNames ).Find (& branches )
183+ func BranchesToNamesSet (branches []* Branch ) container.Set [string ] {
184+ names := make (container.Set [string ], len (branches ))
185+ for _ , branch := range branches {
186+ names .Add (branch .Name )
187+ }
188+ return names
182189}
183190
184191func AddBranches (ctx context.Context , branches []* Branch ) error {
0 commit comments